def test_find_should_return_records_with_date_and_status_in_audit_period(self): end_max = datetime.utcnow() status = 'pending' unordered_results = self.mox.CreateMockAnything() expected_results = [1, 2] related_results = self.mox.CreateMockAnything() self.mox.StubOutWithMock(InstanceExists.objects, 'select_related') InstanceExists.objects.select_related().AndReturn(related_results) related_results.filter(audit_period_ending__lte=dt.dt_to_decimal( end_max), status=status).AndReturn(unordered_results) unordered_results.order_by('id').AndReturn(expected_results) self.mox.ReplayAll() results = InstanceExists.find(end_max, status) self.mox.VerifyAll() self.assertEqual(results, [1, 2])
def test_mark_exists_as_sent_unverified_return_absent_exists(self): message_ids = ["0708cb0b-6169-4d7c-9f58-3cf3d5bf694b", "9156b83e-f684-4ec3-8f94-7e41902f27aa"] exist1 = self.mox.CreateMockAnything() exist1.status = "pending" exist1.save() self.mox.StubOutWithMock(InstanceExists.objects, 'get') InstanceExists.objects.get(message_id=message_ids[0]).AndReturn(exist1) InstanceExists.objects.get(message_id=message_ids[1]).AndRaise( Exception) self.mox.ReplayAll() results = InstanceExists.mark_exists_as_sent_unverified(message_ids) self.assertEqual(results, (['9156b83e-f684-4ec3-8f94-7e41902f27aa'], [])) self.mox.VerifyAll()