diff options
Diffstat (limited to 'tests/TestSwitchBehavior.py')
-rw-r--r-- | tests/TestSwitchBehavior.py | 12 |
1 files changed, 9 insertions, 3 deletions
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() |