aboutsummaryrefslogtreecommitdiff
path: root/tests/integration/test_gemma.py
blob: 3afbc0cd7a3dd3e966842a261fad7c639400add6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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")