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")