def testIsPanicModeNoPackages(self): """Test IsPanicModeNoPackages().""" self.mox.StubOutWithMock(common, 'IsPanicMode') common.IsPanicMode(common.PANIC_MODE_NO_PACKAGES).AndReturn(123) self.mox.ReplayAll() self.assertEqual(123, common.IsPanicModeNoPackages()) self.mox.VerifyAll()
def testIsPanicMode(self): """Tests IsPanicMode().""" mode = common.PANIC_MODES[0] k = '%s%s' % (common.PANIC_MODE_PREFIX, mode) self.mox.StubOutWithMock( common.models.KeyValueCache, 'MemcacheWrappedGet') common.models.KeyValueCache.MemcacheWrappedGet(k).AndReturn(1) common.models.KeyValueCache.MemcacheWrappedGet(k).AndReturn(None) self.mox.ReplayAll() self.assertTrue(common.IsPanicMode(mode)) self.assertFalse(common.IsPanicMode(mode)) self.assertRaises(ValueError, common.IsPanicMode, 'never a mode') self.mox.VerifyAll()
def testPostWhenVerifyEnable(self, *_): mode = 'no_packages' resp = self.testapp.post( '/admin/panic', {'mode': mode, 'enabled': 'enable', 'verify': 1}, status=httplib.FOUND) self.assertTrue(resp.headers['Location'].endswith('/admin/panic')) self.assertTrue(common.IsPanicMode(mode))
def get(self): """GET handler.""" if not self.IsAdminUser(): return modes = [] for mode in common.PANIC_MODES: d = { 'name': mode, 'enabled': common.IsPanicMode(mode), } modes.append(d) self.Render('panic.html', {'modes': modes, 'report_type': 'panic'})