def test_close_context(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
widget.storeSpecificSettings = Mock()
handler.initialize(widget)
widget.schema_only_setting = 0xD06F00D
widget.current_context = handler.new_context()
handler.close_context(widget)
self.assertEqual(widget.schema_only_setting, 0xD06F00D)
def test_close_context(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
widget.storeSpecificSettings = Mock()
handler.initialize(widget)
widget.schema_only_setting = 0xD06F00D
widget.current_context = handler.new_context()
handler.close_context(widget)
self.assertEqual(widget.schema_only_setting, 0xD06F00D)
def test_fast_save(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
handler.initialize(widget)
context = widget.current_context = handler.new_context()
handler.fast_save(widget, 'context_setting', 55)
self.assertEqual(context.values['context_setting'], 55)
def test_about_pack_settings_signal(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
handler.initialize(widget)
fn = Mock()
widget.settingsAboutToBePacked.connect(fn)
handler.pack_data(widget)
self.assertEqual(1, fn.call_count)
handler.update_defaults(widget)
self.assertEqual(2, fn.call_count)
def test_about_pack_settings_signal(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
handler.initialize(widget)
fn = Mock()
widget.settingsAboutToBePacked.connect(fn)
handler.pack_data(widget)
self.assertEqual(1, fn.call_count)
handler.update_defaults(widget)
self.assertEqual(2, fn.call_count)
def test_pack_settings_stores_version(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
handler.initialize(widget)
widget.context_setting = [DummyContext() for _ in range(3)]
settings = handler.pack_data(widget)
self.assertIn("context_settings", settings)
for c in settings["context_settings"]:
self.assertIn(VERSION_KEY, c.values)
def test_fast_save(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
handler.initialize(widget)
context = widget.current_context = handler.new_context()
handler.fast_save(widget, 'context_setting', 55)
self.assertEqual(context.values['context_setting'], 55)
self.assertEqual(handler.known_settings['context_setting'].default,
SimpleWidget.context_setting.default)
def test_pack_settings_stores_version(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
handler.initialize(widget)
widget.context_setting = [DummyContext() for _ in range(3)]
settings = handler.pack_data(widget)
self.assertIn("context_settings", settings)
for c in settings["context_settings"]:
self.assertIn(VERSION_KEY, c.values)
def test_fast_save(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
handler.initialize(widget)
context = widget.current_context = handler.new_context()
handler.fast_save(widget, "context_setting", 55)
self.assertEqual(context.values["context_setting"], 55)
self.assertEqual(
handler.known_settings["context_setting"].default,
SimpleWidget.context_setting.default,
)
def test_write_defaults_stores_version(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
widget.current_context = None
widget.context_settings = [DummyContext() for _ in range(3)]
handler.update_defaults(widget)
f = BytesIO()
f.close = lambda: None
with patch("builtins.open", Mock(return_value=f)):
handler.write_defaults()
f.seek(0)
pickle.load(f) # settings
contexts = pickle.load(f)
for c in contexts:
self.assertEqual(c.values.get("__version__", 0xBAD), 1)
def test_write_defaults_stores_version(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
widget.current_context = None
widget.context_settings = [DummyContext() for _ in range(3)]
handler.update_defaults(widget)
f = BytesIO()
f.close = lambda: None
with patch("builtins.open", Mock(return_value=f)):
handler.write_defaults()
f.seek(0)
pickle.load(f) # settings
contexts = pickle.load(f)
for c in contexts:
self.assertEqual(c.values.get("__version__", 0xBAD), 1)
def test_migrates_settings_removes_incompatible(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
contexts = [Context(foo=i) for i in (13, 13, 0, 1, 13, 2, 13)]
def migrate_context(context, _):
if context.foo == 13:
raise IncompatibleContext()
with patch.object(SimpleWidget, "migrate_context", migrate_context):
handler.initialize(widget, dict(context_settings=contexts))
contexts = widget.context_settings
self.assertEqual(len(contexts), 3)
self.assertTrue(
all(context.foo == i for i, context in enumerate(contexts)))
def test_initialize_migrates_contexts(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
# Old settings without version
contexts = [DummyContext() for _ in range(3)]
migrate_context = Mock()
with patch.object(SimpleWidget, "migrate_context", migrate_context):
handler.initialize(widget, dict(context_settings=contexts))
migrate_context.assert_has_calls([call(c, 0) for c in contexts])
# Settings with version
contexts = [DummyContext(version=i) for i in range(1, 4)]
migrate_context = Mock()
with patch.object(SimpleWidget, "migrate_context", migrate_context):
handler.initialize(widget, dict(context_settings=deepcopy(contexts)))
migrate_context.assert_has_calls([call(c, c.values[VERSION_KEY]) for c in contexts])
def test_migrates_settings_removes_incompatible(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
contexts = [Context(foo=i) for i in (13, 13, 0, 1, 13, 2, 13)]
def migrate_context(context, _):
if context.foo == 13:
raise IncompatibleContext()
with patch.object(SimpleWidget, "migrate_context", migrate_context):
handler.initialize(widget, dict(context_settings=contexts))
contexts = widget.context_settings
self.assertEqual(len(contexts), 3)
self.assertTrue(
all(context.foo == i
for i, context in enumerate(contexts)))
def test_initialize_migrates_contexts(self):
handler = ContextHandler()
handler.bind(SimpleWidget)
widget = SimpleWidget()
# Old settings without version
contexts = [DummyContext() for _ in range(3)]
migrate_context = Mock()
with patch.object(SimpleWidget, "migrate_context", migrate_context):
handler.initialize(widget, dict(context_settings=contexts))
migrate_context.assert_has_calls([call(c, 0) for c in contexts])
# Settings with version
contexts = [DummyContext(version=i) for i in range(1, 4)]
migrate_context = Mock()
with patch.object(SimpleWidget, "migrate_context", migrate_context):
handler.initialize(widget,
dict(context_settings=deepcopy(contexts)))
migrate_context.assert_has_calls(
[call(c, c.values[VERSION_KEY]) for c in contexts])
