def test_dcos_log(mocked_dcos_log_enabked, mocked_http_get, mocked_get_config_val): mocked_dcos_log_enabked.return_value = True m = mock.MagicMock() m.status_code = 200 mocked_http_get.return_value = m mocked_get_config_val.return_value = 'http://127.0.0.1' main._dcos_log(False, 10, True, '', None, []) mocked_http_get.assert_called_with( 'http://127.0.0.1/system/v1/logs/v1/range/?skip_prev=10', headers={'Accept': 'text/plain'})
def test_dcos_log_stream(mocked_dcos_log_enabked, mocked_http_get, mocked_get_config_val, mocked_follow_logs): mocked_dcos_log_enabked.return_value = True m = mock.MagicMock() m.status_code = 200 mocked_http_get.return_value = m mocked_get_config_val.return_value = 'http://127.0.0.1' main._dcos_log(True, 20, False, 'mesos-id', None, []) mocked_follow_logs.assert_called_with( 'http://127.0.0.1/system/v1/agent/mesos-id/logs' '/v1/stream/?skip_prev=20')
def test_dcos_log_leader_mesos(mocked_dcos_log_enabked, mocked_http_get, mocked_get_config_val): mocked_dcos_log_enabked.return_value = True m = mock.MagicMock() m.status_code = 200 mocked_http_get.return_value = m mocked_get_config_val.return_value = 'http://127.0.0.1' main._dcos_log(False, 10, True, '', None, []) mocked_http_get.assert_called_with( 'http://127.0.0.1/system/v1/leader/mesos/logs/v1/range/?skip_prev=10', headers={'Accept': 'text/plain'})
def test_dcos_log_leader_marathon(mocked_dcos_log_enabled, mocked_http_get, mocked_get_config_val): mocked_dcos_log_enabled.return_value = True m = mock.MagicMock() m.status_code = 200 mocked_http_get.return_value = m mocked_get_config_val.return_value = 'http://127.0.0.1' main._dcos_log(False, 10, True, '', 'dcos-marathon', []) mocked_http_get.assert_called_with( 'http://127.0.0.1/system/v1/leader/marathon/logs/v1/range/' '?skip_prev=10&filter=_SYSTEMD_UNIT:dcos-marathon.service', headers={'Accept': 'text/plain'})
def test_dcos_log_leader_marathon(mocked_dcos_log_enabked, mocked_http_get, mocked_get_config_val): mocked_dcos_log_enabked.return_value = True m = mock.MagicMock() m.status_code = 200 mocked_http_get.return_value = m mocked_get_config_val.return_value = 'http://127.0.0.1' main._dcos_log(False, 10, True, '', 'dcos-marathon', []) mocked_http_get.assert_called_with( 'http://127.0.0.1/system/v1/leader/marathon/logs/v1/range/' '?skip_prev=10&filter=_SYSTEMD_UNIT:dcos-marathon.service', headers={'Accept': 'text/plain'})
def test_dcos_log_filters(mocked_dcos_log_enabked, mocked_http_get, mocked_get_config_val, mocked_follow_logs): mocked_dcos_log_enabked.return_value = True m = mock.MagicMock() m.status_code = 200 mocked_http_get.return_value = m mocked_get_config_val.return_value = 'http://127.0.0.1' main._dcos_log(True, 20, False, 'mesos-id', 'dcos-mesos-master', ['key1:value1', 'key2:value2']) mocked_follow_logs.assert_called_with( 'http://127.0.0.1/system/v1/agent/mesos-id/logs/v1/stream/' '?skip_prev=20&filter=key1:value1&filter=key2:value2&' 'filter=_SYSTEMD_UNIT:dcos-mesos-master.service')