def test_get(self):
states = WidgetStates()
_create_widget("trigger", states).trigger_value = True
_create_widget("bool", states).bool_value = True
_create_widget("float", states).double_value = 0.5
_create_widget("int", states).int_value = 123
_create_widget("string", states).string_value = "howdy!"
session_state = SessionState()
session_state.set_widgets_from_proto(states)
session_state._set_widget_metadata(
create_metadata("trigger", "trigger_value"))
session_state._set_widget_metadata(
create_metadata("bool", "bool_value"))
session_state._set_widget_metadata(
create_metadata("float", "double_value"))
session_state._set_widget_metadata(create_metadata("int", "int_value"))
session_state._set_widget_metadata(
create_metadata("string", "string_value"))
self.assertEqual(True, session_state.get("trigger"))
self.assertEqual(True, session_state.get("bool"))
self.assertAlmostEqual(0.5, session_state.get("float"))
self.assertEqual(123, session_state.get("int"))
self.assertEqual("howdy!", session_state.get("string"))
def test_coalesce_widget_states(self):
session_state = SessionState()
old_states = WidgetStates()
_create_widget("old_set_trigger", old_states).trigger_value = True
_create_widget("old_unset_trigger", old_states).trigger_value = False
_create_widget("missing_in_new", old_states).int_value = 123
_create_widget("shape_changing_trigger", old_states).trigger_value = True
session_state._set_widget_metadata(
create_metadata("old_set_trigger", "trigger_value")
)
session_state._set_widget_metadata(
create_metadata("old_unset_trigger", "trigger_value")
)
session_state._set_widget_metadata(
create_metadata("missing_in_new", "int_value")
)
session_state._set_widget_metadata(
create_metadata("shape changing trigger", "trigger_value")
)
new_states = WidgetStates()
_create_widget("old_set_trigger", new_states).trigger_value = False
_create_widget("new_set_trigger", new_states).trigger_value = True
_create_widget("added_in_new", new_states).int_value = 456
_create_widget("shape_changing_trigger", new_states).int_value = 3
session_state._set_widget_metadata(
create_metadata("new_set_trigger", "trigger_value")
)
session_state._set_widget_metadata(create_metadata("added_in_new", "int_value"))
session_state._set_widget_metadata(
create_metadata("shape_changing_trigger", "int_value")
)
session_state.set_widgets_from_proto(
coalesce_widget_states(old_states, new_states)
)
self.assertIsNone(session_state.get("old_unset_trigger"))
self.assertIsNone(session_state.get("missing_in_new"))
self.assertEqual(True, session_state.get("old_set_trigger"))
self.assertEqual(True, session_state.get("new_set_trigger"))
self.assertEqual(456, session_state.get("added_in_new"))
# Widgets that were triggers before, but no longer are, will *not*
# be coalesced
self.assertEqual(3, session_state.get("shape_changing_trigger"))
def test_reset_triggers(self):
states = WidgetStates()
session_state = SessionState()
_create_widget("trigger", states).trigger_value = True
_create_widget("int", states).int_value = 123
session_state.set_widgets_from_proto(states)
session_state.set_metadata(
WidgetMetadata("trigger", lambda x, s: x, None, "trigger_value"))
session_state.set_metadata(
WidgetMetadata("int", lambda x, s: x, None, "int_value"))
self.assertTrue(session_state.get("trigger"))
self.assertEqual(123, session_state.get("int"))
session_state.reset_triggers()
self.assertFalse(session_state.get("trigger"))
self.assertEqual(123, session_state.get("int"))
def test_rerun_data_coalescing(self):
"""Test that multiple RERUN requests get coalesced with
expected values.
(This is similar to widgets_test.test_coalesce_widget_states -
it's testing the same thing, but through the ScriptEventQueue
interface.)
"""
queue = ScriptRequestQueue()
session_state = SessionState()
states = WidgetStates()
_create_widget("trigger", states).trigger_value = True
_create_widget("int", states).int_value = 123
queue.enqueue(ScriptRequest.RERUN, RerunData(widget_states=states))
states = WidgetStates()
_create_widget("trigger", states).trigger_value = False
_create_widget("int", states).int_value = 456
session_state.set_metadata(
WidgetMetadata("trigger", lambda x, s: x, None, "trigger_value"))
session_state.set_metadata(
WidgetMetadata("int", lambda x, s: x, lambda x: x, "int_value"))
queue.enqueue(ScriptRequest.RERUN, RerunData(widget_states=states))
event, data = queue.dequeue()
self.assertEqual(event, ScriptRequest.RERUN)
session_state.set_widgets_from_proto(data.widget_states)
# Coalesced triggers should be True if either the old or
# new value was True
self.assertEqual(True, session_state.get("trigger"))
# Other widgets should have their newest value
self.assertEqual(456, session_state.get("int"))
# We should have no more events
self.assertEqual((None, None), queue.dequeue(),
"Expected empty event queue")
# Test that we can coalesce if previous widget state is None
queue.enqueue(ScriptRequest.RERUN, RerunData(widget_states=None))
queue.enqueue(ScriptRequest.RERUN, RerunData(widget_states=None))
states = WidgetStates()
_create_widget("int", states).int_value = 789
queue.enqueue(ScriptRequest.RERUN, RerunData(widget_states=states))
event, data = queue.dequeue()
session_state.set_widgets_from_proto(data.widget_states)
self.assertEqual(event, ScriptRequest.RERUN)
self.assertEqual(789, session_state.get("int"))
# We should have no more events
self.assertEqual((None, None), queue.dequeue(),
"Expected empty event queue")
# Test that we can coalesce if our *new* widget state is None
states = WidgetStates()
_create_widget("int", states).int_value = 101112
queue.enqueue(ScriptRequest.RERUN, RerunData(widget_states=states))
queue.enqueue(ScriptRequest.RERUN, RerunData(widget_states=None))
event, data = queue.dequeue()
session_state.set_widgets_from_proto(data.widget_states)
self.assertEqual(event, ScriptRequest.RERUN)
self.assertEqual(101112, session_state.get("int"))
# We should have no more events
self.assertEqual((None, None), queue.dequeue(),
"Expected empty event queue")
def test_get_prev_widget_value_nonexistent(self):
session_state = SessionState()
self.assertIsNone(session_state.get("fake_widget_id"))
