def test_mixer_cycle():
mixer = Mixer()
test = mixer.cycle(3).blend(Test)
assert len(test) == 3
assert test[0].__class__ == Test
test = mixer.cycle(3).blend(Test, name=mixer.sequence('lama{0}'))
assert test[2].name == 'lama2'
def test_mixer_cycle(self):
mixer = Mixer()
test = mixer.cycle(3).blend(Test)
self.assertEqual(len(test), 3)
self.assertTrue(type(test[0]), Test)
test = mixer.cycle(3).blend(Test,
name=mixer.sequence('lama{0}'))
self.assertEqual(test[2].name, 'lama2')
def test_oneof_requests_canceled_to_completed(self, modify_mock):
now = timezone.now()
self.ur.operator = 'ONEOF'
self.ur.save()
for req in self.requests:
req.state = 'CANCELED'
req.save()
self.ur.state = 'CANCELED'
self.ur.save()
dmixer.cycle(3).blend(Window, request=(r for r in self.requests), start=now - timedelta(days=2),
end=(e for e in [now - timedelta(days=1), now - timedelta(days=1), now + timedelta(days=1)]))
molecules3 = mixer.cycle(3).blend(PondMolecule, completed=True, failed=False, request_num=self.requests[2].id, tracking_num=self.ur.id)
pond_blocks = [mixer.blend(PondBlock, molecules=molecules3, start=now - timedelta(minutes=30), end=now - timedelta(minutes=20))._to_dict()]
update_request_states_from_pond_blocks(pond_blocks)
request_states = ['CANCELED', 'CANCELED', 'COMPLETED']
for i, req in enumerate(self.requests):
req.refresh_from_db()
self.assertEqual(req.state, request_states[i])
self.ur.refresh_from_db()
self.assertEqual(self.ur.state, 'COMPLETED')
def test_request_state_pond_state_initial_state_pending(self):
request = dmixer.blend(Request, state='PENDING')
fail_count = request.fail_count
now = timezone.now()
molecules = mixer.cycle(4).blend(PondMolecule,
completed=False,
failed=False,
events=[])
pond_blocks = [
mixer.blend(PondBlock,
molecules=molecules,
canceled=False,
start=now - timedelta(minutes=30),
end=now + timedelta(minutes=30))._to_dict(),
mixer.blend(PondBlock, molecules=molecules)._to_dict()
]
state_changed = update_request_state(request, pond_blocks, False)
request.refresh_from_db()
self.assertFalse(state_changed)
self.assertEqual(request.state, 'PENDING')
self.assertEqual(fail_count, request.fail_count)
