def test_status_ui_url(self): app_id = "test_app" mock_scheduler = MagicMock() resp = DescribeAppResponse() resp.ui_url = "https://foobar" mock_scheduler.submit.return_value = app_id mock_scheduler.describe.return_value = resp session = StandaloneSession(name="test_ui_url_session", schedulers={"default": mock_scheduler}) role = Role("ignored").runs("/bin/echo").on(self.test_container) app_handle = session.run(Application(app_id).of(role)) status = session.status(app_handle) self.assertEquals(resp.ui_url, status.ui_url)
def test_status_structured_msg(self): app_id = "test_app" mock_scheduler = MagicMock() resp = DescribeAppResponse() resp.structured_error_msg = '{"message": "test error"}' mock_scheduler.submit.return_value = app_id mock_scheduler.describe.return_value = resp session = StandaloneSession(name="test_structured_msg", schedulers={"default": mock_scheduler}) role = Role("ignored").runs("/bin/echo").on(self.test_container) app_handle = session.run(Application(app_id).of(role)) status = session.status(app_handle) self.assertEquals(resp.structured_error_msg, status.structured_error_msg)
def test_status_unknown_app(self): session = StandaloneSession(name="test_session", scheduler=self.scheduler, wait_interval=1) with self.assertRaises(UnknownAppException): session.status("unknown_app_id")
def test_status_unknown_app(self): session = StandaloneSession(name=SESSION_NAME, schedulers={"default": self.scheduler}, wait_interval=1) self.assertIsNone( session.status("default://test_session/unknown_app_id"))