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)
