def setUpClass(cls): cls.records = [ RepeatRecord(domain='a'), RepeatRecord(domain='b'), RepeatRecord(domain='c'), ] RepeatRecord.bulk_save(cls.records)
def test_get_all_repeat_records_by_domain_since(self): new_records = [ # FAIL RepeatRecord( domain=self.domain, repeater_id=self.repeater_id, last_checked=datetime(2017, 5, 24, 0, 0, 0), failure_reason='some error', ), # CANCELLED RepeatRecord( domain=self.domain, repeater_id=self.repeater_id, last_checked=datetime(2017, 5, 10, 0, 0, 0), cancelled=True, ), # CANCELLED RepeatRecord( domain=self.domain, repeater_id=self.repeater_id, last_checked=datetime(2017, 5, 24, 0, 0, 0), cancelled=True, ), ] RepeatRecord.bulk_save(new_records) self.addCleanup(RepeatRecord.bulk_delete, new_records) records = list( iter_repeat_records_by_domain(self.domain, state=RECORD_CANCELLED_STATE, since=datetime(2017, 5, 20))) self.assertEqual(len(records), 1) record, = records self.assertEqual(record.to_json(), new_records[-1].to_json())