def test1(self): session_key = temp.NamedTemporaryFile('rw', delete=True).name api.start(True) self.assertRaises(MaintenanceModeError, CommandTask, session_key, timeout=2, check_maintenance=True)
def handle_label(self, cmd, **options): verbosity = options.get("verbosity") timeout = options.get("timeout") ignore_session = options.get("ignore_session") ret, msg = 0, "Unknow error" if cmd not in Command.args: raise CommandError("Allowed options are: %s" % self.args) if cmd in ("check", "status"): ret, msg = api.check() print msg elif cmd in ("on", "activate"): ret, msg = api.start(ignore_session, timeout, verbosity) if verbosity >= 1: print msg elif cmd in ("off", "deactivate"): ret, msg = api.stop() if verbosity >= 1: print msg elif cmd in ("list",): now = datetime.datetime.now() for s in Session.objects.filter(expire_date__gte=now): offset = time.mktime(s.expire_date.timetuple()) - time.mktime(now.timetuple()) print s.pk, s.expire_date, offset if ret: raise CommandError(msg)
def handle_label(self, cmd, **options): verbosity = options.get('verbosity') timeout = options.get('timeout') ignore_session = options.get('ignore_session') ret, msg = 0,'Unknow error' if cmd not in Command.args: raise CommandError('Allowed options are: %s' % self.args) if cmd in ('check', 'status'): ret, msg = api.check() print msg elif cmd in ('on', 'activate'): ret, msg = api.start(ignore_session, timeout, verbosity) if verbosity >= 1: print msg elif cmd in ('off', 'deactivate'): ret, msg = api.stop() if verbosity >= 1: print msg elif cmd in ('list',): now = datetime.datetime.now() for s in Session.objects.filter(expire_date__gte=now): offset = (time.mktime(s.expire_date.timetuple())-time.mktime(now.timetuple())) print s.pk, s.expire_date, offset if ret: raise CommandError(msg)
def test_is_online(self): ret, msg = api.start() self.assertEqual(ret, SUCCESS) self.assertFalse(api.is_online()) api.stop() self.assertTrue(api.is_online())
def test_start(self): ret, msg = api.start() self.assertEqual(ret, SUCCESS) self.assertTrue(os.path.exists(api.MAINTENANCE_FILE)) self.assertFalse(os.path.exists(api.PENDING_MAINTENANCE_FILE))
def test1(self): session_key = temp.NamedTemporaryFile("rw", delete=True).name api.start(True) self.assertRaises(MaintenanceModeError, CommandTask, session_key, timeout=2, check_maintenance=True)