From d8ba03006e2ea1400e80caded738e79c186e8de3 Mon Sep 17 00:00:00 2001 From: eugue Date: Mon, 14 Feb 2011 17:13:33 -0500 Subject: change SwitchBehavior to take in a JSON dict to avoid weird XML parsing. Also added a public function to manually set current behavior. --- tests/TestSwitchBehavior.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'tests') diff --git a/tests/TestSwitchBehavior.py b/tests/TestSwitchBehavior.py index f130431..774dbbc 100644 --- a/tests/TestSwitchBehavior.py +++ b/tests/TestSwitchBehavior.py @@ -15,21 +15,27 @@ class TestSwitchBehavior(unittest.TestCase): compReg.registerComponent(self.behavior1) compReg.registerComponent(self.behavior2) - self.switchBehavior = SwitchBehavior({'Id': 'switch', '1': 'behavior1', '2': 'behavior2', 'DefaultBehavior': 'behavior1'}) + self.switchBehavior = SwitchBehavior({'Id': 'switch', 'PrefixToBehavior': '{"@": "behavior1", "#": "behavior2"}', 'DefaultBehavior': 'behavior1'}) compReg.registerComponent(self.switchBehavior) def tearDown(self): pass def test_switch_to_behavior1(self): - inputs = [{'Data': '1something', 'Location': 'someloc'}] + inputs = [{'Data': '@something', 'Location': 'someloc'}] returned = self.switchBehavior.processResponse(inputs, []) assert returned[0][0]['Location'] == 'someloc' def test_switch_to_behavior2(self): - inputs = [{'Data': '2something'}] + inputs = [{'Data': '#something'}] returned = self.switchBehavior.processResponse(inputs, []) assert returned[0] == [] + def test_default_behavior(self): + inputs = [{'Data': 'something', 'Location': 'someloc'}] + returned = self.switchBehavior.processResponse(inputs, []) + assert returned[0][0]['Location'] == 'someloc' + + if __name__ == '__main__': unittest.main() -- cgit v1.2.3