From b60a06f00ed0efdfb5866221ac8f15f78d770cfc Mon Sep 17 00:00:00 2001 From: Jorrit Schaap <schaap@astron.nl> Date: Fri, 21 Feb 2020 15:47:44 +0100 Subject: [PATCH] TMSS-153: fixed test --- SAS/TMSS/test/t_tmss_test_database.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/SAS/TMSS/test/t_tmss_test_database.py b/SAS/TMSS/test/t_tmss_test_database.py index 418ad854483..3e99b742d72 100755 --- a/SAS/TMSS/test/t_tmss_test_database.py +++ b/SAS/TMSS/test/t_tmss_test_database.py @@ -46,14 +46,15 @@ class TMSSPostgresTestMixinTestCase(TMSSPostgresTestMixin, unittest.TestCase): def test_db_basics(self): '''Can we do some simple plain sql queries?''' with PostgresDatabaseConnection(self.dbcreds) as db: - self.assertEqual(0, db.executeQuery("SELECT COUNT(*) FROM tmssapp_cycle;", fetch=FETCH_ONE)['count']) + cycle_count = db.executeQuery("SELECT COUNT(*) FROM tmssapp_cycle;", fetch=FETCH_ONE)['count'] + self.assertGreaterEqual(cycle_count, 0) now = datetime.utcnow() db.executeQuery('''INSERT INTO tmssapp_cycle VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s);''', qargs=([], now, now, "my_description", "my_name", now, now, 0, 1, 2, 3)) - self.assertEqual(1, db.executeQuery("SELECT COUNT(*) FROM tmssapp_cycle;", fetch=FETCH_ONE)['count']) + self.assertEqual(cycle_count+1, db.executeQuery("SELECT COUNT(*) FROM tmssapp_cycle;", fetch=FETCH_ONE)['count']) if __name__ == "__main__": -- GitLab