def mock_event(mock_task_config, is_terminal=False):
return Event(kind='task',
timestamp=1234.5678,
terminal=is_terminal,
success=False,
task_id=mock_task_config.task_id,
platform_type='mesos',
message='mock_message',
task_config=mock_task_config,
raw='raw_event')
def test_event_creation():
x = object()
e = Event(kind='task',
task_id="123",
raw=x,
terminal=True,
platform_type='killed')
assert e.raw == x
assert e.terminal
assert e.platform_type == 'killed'
def mock_event(mock_task_config):
return Event(
kind='task',
timestamp=1234.5678,
terminal=True,
task_id=mock_task_config.task_id,
platform_type='mesos',
message='mock_message',
task_config=mock_task_config,
raw='raw_event',
)
def test_event_type_checks(event):
with pytest.raises(InvariantException) as e:
Event()
assert 'missing_fields' in str(e.value)
with pytest.raises(PTypeError) as e:
event.set(terminal="hello")
with pytest.raises(PTypeError) as e:
event.set(platform_type=123)
with pytest.raises(PTypeError) as e:
event.set(task_id=123)
def test_retry_loop_filters_out_non_task(mock_retrying_executor):
mock_event = Event(kind='control',
raw='some message',
message='stop',
terminal=True)
mock_retrying_executor.stopping = True
mock_retrying_executor._is_current_attempt = mock.Mock(return_value=True)
mock_retrying_executor.event_with_retries = mock.Mock()
mock_retrying_executor.src_queue.put(mock_event)
mock_retrying_executor.retry_loop()
assert mock_retrying_executor.dest_queue.qsize() == 1
def test_mesos_update_to_event(mock_time):
mock_time.time.return_value = 12345678.0
for key, val in MESOS_STATUS_MAP.items():
mesos_status = mock.MagicMock()
mesos_status.state = key
assert mesos_update_to_event(mesos_status,
addict.Dict(task_id='123')) == Event(
kind='task',
raw=mesos_status,
task_id='123',
task_config={'task_id': '123'},
timestamp=12345678.0,
terminal=val.terminal,
platform_type=val.platform_type,
success=val.get('success', None),
)
def event():
return Event(kind='task')
def event():
return Event(kind='task', task_id="123")