20 """Set up and clean up fixtures for the Unity SettingsWizard tests."""
25 from os.path
import expanduser
28 class SettingsWizard(fixtures.Fixture):
29 WIZARD_FILE = expanduser(
"~") + \
30 '/.config/ubuntu-system-settings/wizard-has-run'
32 def __init__(self, enable):
38 original_state = self.is_settings_wizard_enabled()
39 self.addCleanup(self.set_settings_wizard, original_state)
40 if self.enable != original_state:
41 self.set_settings_wizard(self.enable)
43 def is_settings_wizard_enabled(self):
44 return not os.path.exists(self.WIZARD_FILE)
46 def set_settings_wizard(self, enabled):
48 self.enable_settings_wizard()
50 self.disable_settings_wizard()
52 def enable_settings_wizard(self):
53 if os.path.exists(self.WIZARD_FILE):
54 os.remove(self.WIZARD_FILE)
56 def disable_settings_wizard(self):
57 if not os.path.exists(self.WIZARD_FILE):
58 open(self.WIZARD_FILE,
'a').close()