def test_mark_exists_as_sent_unverified_and_return_exist_not_pending(self): message_ids = ["0708cb0b-6169-4d7c-9f58-3cf3d5bf694b", "9156b83e-f684-4ec3-8f94-7e41902f27aa"] exist1 = self.mox.CreateMockAnything() exist1.status = "pending" exist1.save() exist2 = self.mox.CreateMockAnything() exist2.status = "verified" exist3 = self.mox.CreateMockAnything() exist3.status = "pending" exist3.save() self.mox.StubOutWithMock(ImageExists.objects, 'filter') ImageExists.objects.filter(message_id=message_ids[0]).AndReturn( [exist1, exist2]) ImageExists.objects.filter(message_id=message_ids[1]).AndReturn( [exist3]) self.mox.ReplayAll() results = ImageExists.mark_exists_as_sent_unverified(message_ids) self.assertEqual(results, ([], ["0708cb0b-6169-4d7c-9f58-3cf3d5bf694b"])) self.mox.VerifyAll()
def test_group_exists_with_date_status_in_audit_period_by_owner_rawid(self): end_max = datetime.utcnow() status = 'pending' exist1 = self.mox.CreateMockAnything() exist1.owner = "owner1" exist1.raw_id = "1" exist2 = self.mox.CreateMockAnything() exist2.owner = "owner2" exist2.raw_id = "2" exist3 = self.mox.CreateMockAnything() exist3.owner = "owner1" exist3.raw_id = "1" exist4 = self.mox.CreateMockAnything() exist4.owner = "owner1" exist4.raw_id = "3" ordered_results = [exist1, exist3, exist4, exist2] unordered_results = self.mox.CreateMockAnything() related_results = self.mox.CreateMockAnything() self.mox.StubOutWithMock(ImageExists.objects, 'select_related') ImageExists.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('owner').AndReturn(ordered_results) self.mox.ReplayAll() results = ImageExists.find_and_group_by_owner_and_raw_id(end_max, status) self.mox.VerifyAll() self.assertEqual(results, {'owner1-1': [exist1, exist3], 'owner1-3': [exist4], 'owner2-2': [exist2]})
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(ImageExists.objects, 'select_related') ImageExists.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 = ImageExists.find(end_max, status) self.mox.VerifyAll() self.assertEqual(results, [1, 2])