def test_dcos_log(mocked_get_config_val, mocked_http_get): mocked_get_config_val.return_value = 'http://127.0.0.1' m = MagicMock() m.status_code = 200 mocked_http_get.return_value = m executor_info = { 'tasks': [{ 'container': 'container1', 'state': 'TASK_RUNNING', 'statuses': [{ 'state': 'TASK_RUNNING', 'container_status': { 'container_id': { 'value': 'child-123', 'parent': { 'value': 'parent-456' } } } }], 'slave_id': 'slave-123', 'framework_id': 'framework-123', 'id': 'id-123' }] } task = MagicMock task.executor = lambda: executor_info _dcos_log(False, [task], 10, 'stdout', False) mocked_http_get.assert_called_with( 'http://127.0.0.1/system/v1/agent/slave-123/logs/v1/range/framework/' 'framework-123/executor/id-123/container/parent-456.child-123' '?skip_prev=10&filter=STREAM:STDOUT', headers={'Accept': 'text/plain'}, is_success=is_success)
def test_dcos_log(mocked_get_config_val, mocked_http_get): mocked_get_config_val.return_value = 'http://127.0.0.1' m = MagicMock() m.status_code = 200 mocked_http_get.return_value = m executor_info = { 'tasks': [ { 'container': 'container1', 'state': 'TASK_RUNNING', 'statuses': [ { 'state': 'TASK_RUNNING', 'container_status': { 'container_id': { 'value': 'child-123', 'parent': { 'value': 'parent-456' } } } } ], 'slave_id': 'slave-123', 'framework_id': 'framework-123', 'id': 'id-123' } ] } task = MagicMock task.executor = lambda: executor_info _dcos_log(False, [task], 10, 'stdout', False) mocked_http_get.assert_called_with( 'http://127.0.0.1/system/v1/agent/slave-123/logs/v1/range/framework/' 'framework-123/executor/id-123/container/parent-456.child-123' '?skip_prev=10&filter=STREAM:STDOUT', headers={'Accept': 'text/plain'})
def test_dcos_log_stream(mocked_get_config_val, mocked_http_get, mocked_follow_logs): mocked_get_config_val.return_value = 'http://127.0.0.1' m = MagicMock() m.status_code = 200 mocked_http_get.return_value = m executor_info = { 'tasks': [{ 'container': 'container1', 'state': 'TASK_RUNNING', 'statuses': [{ 'state': 'TASK_RUNNING', 'container_status': { 'container_id': { 'value': 'child-123', } } }], 'slave_id': 'slave-123', 'framework_id': 'framework-123', 'id': 'id-123' }] } task = MagicMock task.executor = lambda: executor_info _dcos_log(True, [task], 10, 'stderr', False) mocked_follow_logs.assert_called_with( 'http://127.0.0.1/system/v1/agent/slave-123/logs/v1/' 'stream/framework/framework-123/executor/id-123/container/' 'child-123?skip_prev=10&filter=STREAM:STDERR')
def test_dcos_log_stream(mocked_get_config_val, mocked_http_get, mocked_follow_logs): mocked_get_config_val.return_value = 'http://127.0.0.1' m = MagicMock() m.status_code = 200 mocked_http_get.return_value = m executor_info = { 'tasks': [ { 'container': 'container1', 'state': 'TASK_RUNNING', 'statuses': [ { 'state': 'TASK_RUNNING', 'container_status': { 'container_id': { 'value': 'child-123', } } } ], 'slave_id': 'slave-123', 'framework_id': 'framework-123', 'id': 'id-123' } ] } task = MagicMock task.executor = lambda: executor_info _dcos_log(True, [task], 10, 'stderr', False) mocked_follow_logs.assert_called_with( 'http://127.0.0.1/system/v1/agent/slave-123/logs/v1/' 'stream/framework/framework-123/executor/id-123/container/' 'child-123?skip_prev=10&filter=STREAM:STDERR')