update vioscreen test

biocore · Sep 10, 2024 · b781035 · b781035
1 parent 6084b95
commit b781035
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git 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
@@ -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):