Add basic integration tests to test out the API

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")