Create test_session_tracking.py

test session tracking

Signed-off-by: Josef Edwards <joed6834@colorado.edu>

test_session_tracking.py

@@ -0,0 +1,39 @@
+import unittest
+from session_tracking import get_db_connection, create_sessions_table, insert_session_data, retrieve_session_data
+
+class TestSessionTracking(unittest.TestCase):
+
+    def setUp(self):
+        """Shared setup method to initialize the database and table."""
+        self.conn = get_db_connection()
+        self.cursor = self.conn.cursor()
+        create_sessions_table(self.cursor)
+
+    def tearDown(self):
+        """Clean up method to delete data after each test."""
+        self.cursor.execute('DELETE FROM sessions')
+        self.conn.commit()
+
+    def test_create_sessions_table(self):
+        """Test if the sessions table is created."""
+        self.assertTrue(self.cursor.execute('SELECT * FROM sessions').fetchone() is None)
+
+    def test_insert_session_data(self):
+        """Test if session data is inserted correctly."""
+        insert_session_data(self.cursor, 1, 'reset_context=False, check_flags=False', 'Started the conversation with some context.', '2023-03-01 12:00:00')
+        self.assertTrue(self.cursor.execute('SELECT * FROM sessions').fetchone() is not None)
+
+    def test_retrieve_session_data(self):
+        """Test if session data is retrieved correctly."""
+        insert_session_data(self.cursor, 1, 'reset_context=False, check_flags=False', 'Started the conversation with some context.', '2023-03-01 12:00:00')
+        session_data = retrieve_session_data(self.cursor, 1)
+        self.assertEqual(session_data[0], 'reset_context=False, check_flags=False')
+        self.assertEqual(session_data[1], 'Started the conversation with some context.')
+
+    def test_retrieve_non_existent_session(self):
+        """Test retrieving a session that doesn't exist."""
+        session_data = retrieve_session_data(self.cursor, 999)  # assuming 999 is an invalid ID
+        self.assertIsNone(session_data)
+
+if __name__ == '__main__':
+    unittest.main()