diff options
Diffstat (limited to 'tests/integration')
-rw-r--r-- | tests/integration/__init__.py | 0 | ||||
-rw-r--r-- | tests/integration/test_gemma.py | 15 |
2 files changed, 15 insertions, 0 deletions
diff --git a/tests/integration/__init__.py b/tests/integration/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/tests/integration/__init__.py diff --git a/tests/integration/test_gemma.py b/tests/integration/test_gemma.py new file mode 100644 index 0000000..3afbc0c --- /dev/null +++ b/tests/integration/test_gemma.py @@ -0,0 +1,15 @@ +"""Integration tests for gemma API endpoints""" +import unittest + +from gn3.app import create_app + + +class GemmaAPITest(unittest.TestCase): + """Test cases for the Gemma API""" + def setUp(self): + self.app = create_app().test_client() + + def test_gemma_index(self): + """Test that the correct response is returned""" + response = self.app.get("/gemma", follow_redirects=True) + self.assertEqual(response.get_json().get("result"), "hello world") |