def test_stop(self, _): session = StandaloneSession( name=SESSION_NAME, schedulers={"default": self.scheduler}, wait_interval=1 ) self.assertIsNone(session.stop("default://test_session/unknown_app_id")) with self.assertRaises(SessionMismatchException): session.stop("default://another_session/some_app_id")
def test_status(self, _): session = StandaloneSession( name=SESSION_NAME, schedulers={"default": self.scheduler}, wait_interval=1 ) role = Role(name="sleep").runs("sleep.sh", "60").on(self.test_container) app = Application("sleeper").of(role) app_handle = session.run(app, cfg=self.cfg) self.assertEqual(AppState.RUNNING, session.status(app_handle).state) session.stop(app_handle) self.assertEqual(AppState.CANCELLED, session.status(app_handle).state)
def test_status(self): session = StandaloneSession(name="test_session", scheduler=self.scheduler, wait_interval=1) role = session.role(name="sleep").runs("sleep.sh", "60").on(self.test_container) app = session.app("sleeper").of(role) app_id = session.run(app) self.assertEqual(AppState.RUNNING, session.status(app_id).state) session.stop(app_id) self.assertEqual(AppState.CANCELLED, session.status(app_id).state)
def test_attach(self): session1 = StandaloneSession(name="test_session1", scheduler=self.scheduler) role = (session1.role(name="sleep").runs("sleep.sh", "60").on(self.test_container)) app = session1.app("sleeper").of(role) app_id = session1.run(app) session2 = StandaloneSession(name="test_session2", scheduler=self.scheduler) session2.attach(app_id) self.assertEqual(AppState.RUNNING, session2.status(app_id).state) session2.stop(app_id) self.assertEqual(AppState.CANCELLED, session2.status(app_id).state)
def test_stop_unknown_app(self): session = StandaloneSession(name="test_session", scheduler=self.scheduler, wait_interval=1) with self.assertRaises(UnknownAppException): session.stop("unknown_app_id")