def test_log_lines(self):
app_id = "mock_app"
scheduler_mock = MagicMock()
scheduler_mock.describe.return_value = DescribeAppResponse(
app_id, AppState.RUNNING)
scheduler_mock.log_iter.return_value = iter(["hello", "world"])
session = StandaloneSession(name=SESSION_NAME,
schedulers={"default": scheduler_mock},
wait_interval=1)
role_name = "trainer"
replica_id = 2
regex = "QPS.*"
since = datetime.datetime.now()
until = datetime.datetime.now()
lines = list(
session.log_lines(
f"default://test_session/{app_id}",
role_name,
replica_id,
regex,
since,
until,
))
self.assertEqual(["hello", "world"], lines)
scheduler_mock.log_iter.assert_called_once_with(
app_id, role_name, replica_id, regex, since, until)
def test_log_lines_unknown_app(self):
session = StandaloneSession(name=SESSION_NAME,
schedulers={"default": self.scheduler},
wait_interval=1)
with self.assertRaises(UnknownAppException):
session.log_lines("default://test_session/unknown", "trainer")
