aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/integration/__init__.py0
-rw-r--r--tests/integration/test_gemma.py15
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")