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