From ee851978b616079537c783a7de912aa2589328b8 Mon Sep 17 00:00:00 2001 From: gengchc2 Date: Mon, 12 Nov 2018 23:56:57 -0800 Subject: [PATCH] Add session delete unit test for sqlalchemy Change-Id: I6d9909b13c1378aea3ce24f1c89ffca0b4ea17b0 --- .../tests/unit/sqlalchemy/test_session.py | 24 +++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/freezer_api/tests/unit/sqlalchemy/test_session.py b/freezer_api/tests/unit/sqlalchemy/test_session.py index 1c236fc9..a7b47dcd 100644 --- a/freezer_api/tests/unit/sqlalchemy/test_session.py +++ b/freezer_api/tests/unit/sqlalchemy/test_session.py @@ -60,3 +60,27 @@ class DbSessionTestCase(base.DbTestCase): self.fake_session_0.get('project_id')) self.assertEqual(result.get('schedule'), self.fake_session_0.get('schedule')) + + def test_add_and_delete_session(self): + session_doc = copy.deepcopy(self.fake_session_0) + session_id = self.dbapi.add_session(project_id=self.fake_session_0. + get('project_id'), + user_id=self.fake_session_0. + get('user_id'), + doc=session_doc) + self.assertIsNotNone(session_id) + + result = self.dbapi.delete_session(project_id=self.fake_session_0. + get('project_id'), + user_id=self.fake_session_0. + get('user_id'), + session_id=session_id) + + self.assertIsNotNone(result) + self.assertEqual(result, session_id) + result = self.dbapi.get_session(project_id=self.fake_session_0. + get('project_id'), + user_id=self.fake_session_0. + get('user_id'), + session_id=session_id) + self.assertEqual(len(result), 0)