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"))