def test_context_processor(self): with self.settings(MAINTENANCE_MODE_STATE_FILE_PATH=self.tmp_dir): for environ, override, result in self.override_cases: core.set_maintenance_mode(environ) with core.override_maintenance_mode(override): self.assertEqual(core.get_maintenance_mode(), result)
def test_context_manager_override(self): with self.settings(MAINTENANCE_MODE_STATE_FILE_PATH=self.tmp_dir): for environ, override, result in self.override_cases: core.set_maintenance_mode(environ) with core.override_maintenance_mode(override): self.assertEqual(core.get_maintenance_mode(), result) self.assertEqual(core.get_maintenance_mode(), environ)