def test_instance_task(mock_get_task, mock_instance_status, mock_add_slave_info, mock_add_executor_info): mock_request = mock.Mock(swagger_data={ "task_id": "123", "slave_hostname": "host1" }) mock_instance_status.return_value = {"marathon": {"app_id": "app1"}} mock_task_1 = mock.Mock() mock_get_task.return_value = mock_task_1 ret = instance.instance_task(mock_request) assert not mock_add_slave_info.called assert not mock_add_executor_info.called assert ret == mock_task_1._Task__items mock_request = mock.Mock(swagger_data={ "task_id": "123", "slave_hostname": "host1", "verbose": True }) ret = instance.instance_task(mock_request) mock_add_slave_info.assert_called_with(mock_task_1) mock_add_executor_info.assert_called_with(mock_add_slave_info.return_value) expected = mock_add_executor_info.return_value._Task__items assert ret == expected
def test_instance_task(mock_get_task, mock_instance_status, mock_add_slave_info, mock_add_executor_info): mock_request = mock.Mock(swagger_data={ 'task_id': '123', 'slave_hostname': 'host1' }) mock_instance_status.return_value = {'marathon': {'app_id': 'app1'}} mock_task_1 = mock.Mock() mock_get_task.return_value = mock_task_1 ret = instance.instance_task(mock_request) assert not mock_add_slave_info.called assert not mock_add_executor_info.called assert ret == mock_task_1._Task__items mock_request = mock.Mock(swagger_data={ 'task_id': '123', 'slave_hostname': 'host1', 'verbose': True }) ret = instance.instance_task(mock_request) mock_add_slave_info.assert_called_with(mock_task_1) mock_add_executor_info.assert_called_with(mock_add_slave_info.return_value) expected = mock_add_executor_info.return_value._Task__items assert ret == expected mock_instance_status.return_value = {'chronos': {}} with raises(ApiFailure): ret = instance.instance_task(mock_request)
def test_instance_task(mock_get_task, mock_instance_status, mock_add_slave_info, mock_add_executor_info): mock_request = mock.Mock(swagger_data={'task_id': '123', 'slave_hostname': 'host1'}) mock_instance_status.return_value = {'marathon': {'app_id': 'app1'}} mock_task_1 = mock.Mock() mock_get_task.return_value = mock_task_1 ret = instance.instance_task(mock_request) assert not mock_add_slave_info.called assert not mock_add_executor_info.called assert ret == mock_task_1._Task__items mock_request = mock.Mock(swagger_data={'task_id': '123', 'slave_hostname': 'host1', 'verbose': True}) ret = instance.instance_task(mock_request) mock_add_slave_info.assert_called_with(mock_task_1) mock_add_executor_info.assert_called_with(mock_add_slave_info.return_value) expected = mock_add_executor_info.return_value._Task__items assert ret == expected mock_instance_status.return_value = {'chronos': {}} with raises(ApiFailure): ret = instance.instance_task(mock_request)