def test_core_invalid_argument(self): self.__reset_state() io.write_file(settings.MAINTENANCE_MODE_STATE_FILE_PATH, 'not bool') self.assertRaises(ValueError, core.get_maintenance_mode) self.assertRaises(TypeError, core.set_maintenance_mode, 'not bool')
def test_core_invalid_argument(self): self.__reset_state() io.write_file(settings.MAINTENANCE_MODE_STATE_FILE_PATH, 'not bool') self.assertRaises(ValueError, core.get_maintenance_mode) self.assertRaises(TypeError, core.set_maintenance_mode, 'not bool')
def set_maintenance_mode(value): if not isinstance(value, bool): raise TypeError('value argument type is not boolean') value = str(int(value)) io.write_file(settings.MAINTENANCE_MODE_STATE_FILE_PATH, value)
def test_backend_local_file_invalid_values(self): self.__reset_state() file_path = settings.MAINTENANCE_MODE_STATE_FILE_PATH io.write_file(file_path, 'test') backend = core.get_maintenance_mode_backend() self.assertRaises(ValueError, backend.get_value) self.assertRaises(ValueError, backend.set_value, 2) self.assertRaises(ValueError, backend.set_value, 'test')
def set_maintenance_mode(value): # If maintenance mode is defined in settings, it can't be changed. if settings.MAINTENANCE_MODE is not None: raise ImproperlyConfigured('Maintenance mode cannot be set dynamically if defined in settings.') if not isinstance(value, bool): raise TypeError('value argument type is not boolean') value = str(int(value)) io.write_file(settings.MAINTENANCE_MODE_STATE_FILE_PATH, value)
def test_backend_local_file_invalid_values(self): self.__reset_state() file_path = settings.MAINTENANCE_MODE_STATE_FILE_PATH io.write_file(file_path, 'test') backend = core.get_maintenance_mode_backend() self.assertRaises(ValueError, backend.get_value) self.assertRaises(ValueError, backend.set_value, 2) self.assertRaises(ValueError, backend.set_value, 'test')
def test_io(self): self.__reset_state() file_path = settings.MAINTENANCE_MODE_STATE_FILE_PATH val = io.read_file(file_path) self.assertEqual(val, '') val = io.write_file(file_path, 'test') self.assertTrue(val) #ensure overwrite instead of append val = io.write_file(file_path, 'test') val = io.write_file(file_path, 'test') val = io.read_file(file_path) self.assertEqual(val, 'test')
def test_io_invalid_file_path(self): self.__reset_state() file_path = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ:/maintenance_mode_state.txt' val = io.write_file(file_path, 'test') self.assertFalse(val) val = io.read_file(file_path) self.assertEqual(val, '')
def test_io(self): self.__reset_state() file_path = settings.MAINTENANCE_MODE_STATE_FILE_PATH val = io.read_file(file_path) self.assertEqual(val, '') # ensure overwrite instead of append io.write_file(file_path, 'test') io.write_file(file_path, 'test') io.write_file(file_path, 'test') val = io.read_file(file_path) self.assertEqual(val, 'test')
def set_value(self, value): value = str(int(value)) if value.strip() not in ['0', '1']: raise ValueError('state file content value is not 0|1') write_file(settings.MAINTENANCE_MODE_STATE_FILE_PATH, value)
def set_value(self, value): value = str(int(value)) if value not in ['0', '1']: raise ValueError('state file content value is not 0|1') write_file(settings.MAINTENANCE_MODE_STATE_FILE_PATH, value)