From b7810353ea6f9c13e929550a4eb8f25aec7f4d05 Mon Sep 17 00:00:00 2001 From: ayobi Date: Tue, 10 Sep 2024 17:48:05 -0300 Subject: [PATCH] update vioscreen test --- .../repo/tests/test_vioscreen_sessions.py | 8 ++++---- microsetta_private_api/repo/vioscreen_repo.py | 4 ---- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/microsetta_private_api/repo/tests/test_vioscreen_sessions.py b/microsetta_private_api/repo/tests/test_vioscreen_sessions.py index 8d69efcd6..87434ebc0 100644 --- a/microsetta_private_api/repo/tests/test_vioscreen_sessions.py +++ b/microsetta_private_api/repo/tests/test_vioscreen_sessions.py @@ -278,24 +278,24 @@ def test_get_ffq_status_by_sample(self): # enumerate the empirically observed states from vioscreen # (is_complete, has_taken, exact_status) obs = r.get_ffq_status_by_source(SOURCE_ID_FOR_VIOSESSION) - self.assertEqual(obs, (True, True, 'Finished')) + self.assertEqual(obs[1], (True, True, 'Finished')) session.status = 'Started' session.endDate = None r.upsert_session(session) obs = r.get_ffq_status_by_source(SOURCE_ID_FOR_VIOSESSION) - self.assertEqual(obs, (False, True, 'Started')) + self.assertEqual(obs[1], (False, True, 'Started')) session.status = 'New' r.upsert_session(session) obs = r.get_ffq_status_by_source(SOURCE_ID_FOR_VIOSESSION) - self.assertEqual(obs, (False, False, 'New')) + self.assertEqual(obs[1], (False, False, 'New')) session.status = 'Review' r.upsert_session(session) obs = r.get_ffq_status_by_source(SOURCE_ID_FOR_VIOSESSION) - self.assertEqual(obs, (False, True, 'Review')) + self.assertEqual(obs[1], (False, True, 'Review')) if __name__ == '__main__': diff --git a/microsetta_private_api/repo/vioscreen_repo.py b/microsetta_private_api/repo/vioscreen_repo.py index 6b11e7672..4a9aa36e0 100644 --- a/microsetta_private_api/repo/vioscreen_repo.py +++ b/microsetta_private_api/repo/vioscreen_repo.py @@ -55,10 +55,6 @@ def upsert_session(self, session): {doupdateset} """, tuple([getattr(session, attr) for attr in self.COLS])) - cur.execute("SELECT * FROM ag.vioscreen_sessions " - "WHERE sessionId = %s", (session.sessionId,)) - row = cur.fetchone() - print("Upserted row:", row) return cur.rowcount == 1 def get_session(self, sessionId):