def test_update_uploaded_row_by_uid(self): h = DBHandler(inmemory=True) records = [Record(1, datetime(2020,12,24,0,0,0), 88, 99), Record(1, datetime(2020,12,24,0,0,0), 88, 99), Record(1, datetime(2020,12,24,0,0,0), 88, 99),] [h.insert_record(r) for r in records] h.update_uploaded_row_by_uid(2) rows = h.read_unuploaded_rows() uids = [r[0] for r in rows] self.assertEqual(uids, [1,3])