def test_oneof_all_canceled(self):
request_states = ['CANCELED', 'CANCELED', 'CANCELED']
ur = dmixer.blend(UserRequest, operator='ONEOF')
dmixer.cycle(3).blend(Request, state=(state for state in request_states), user_request=ur)
aggregate_state = aggregate_request_states(ur)
self.assertEqual(aggregate_state, 'CANCELED')
def test_oneof_pending_and_expired(self):
request_states = ['WINDOW_EXPIRED', 'PENDING', 'PENDING']
ur = dmixer.blend(UserRequest, operator='ONEOF')
dmixer.cycle(3).blend(Request, state=(state for state in request_states), user_request=ur)
aggregate_state = aggregate_request_states(ur)
self.assertEqual(aggregate_state, 'PENDING')
def test_many_canceled_and_complete(self):
request_states = ['CANCELED', 'COMPLETED', 'CANCELED']
ur = dmixer.blend(UserRequest, operator='MANY')
dmixer.cycle(3).blend(Request, state=(state for state in request_states), user_request=ur)
aggregate_state = aggregate_request_states(ur)
self.assertEqual(aggregate_state, 'COMPLETED')
def test_many_all_expired(self):
request_states = ['WINDOW_EXPIRED', 'WINDOW_EXPIRED', 'WINDOW_EXPIRED']
ur = dmixer.blend(UserRequest, operator='MANY')
dmixer.cycle(3).blend(Request, state=(state for state in request_states), user_request=ur)
aggregate_state = aggregate_request_states(ur)
self.assertEqual(aggregate_state, 'WINDOW_EXPIRED')
def test_many_any_pending(self):
request_states = ['COMPLETED', 'CANCELED', 'PENDING']
ur = dmixer.blend(UserRequest, operator='MANY')
dmixer.cycle(3).blend(Request, state=(state for state in request_states), user_request=ur)
aggregate_state = aggregate_request_states(ur)
self.assertEqual(aggregate_state, 'PENDING')
