1 files changed, 35 insertions, 0 deletions
diff --git a/tests/TestSwitchBehavior.py b/tests/TestSwitchBehavior.py
new file mode 100644
index 0000000..f130431
--- /dev/null
+++ b/tests/TestSwitchBehavior.py
@@ -0,0 +1,35 @@
+import unittest
+import util.ComponentRegistry as compReg
+
+from behaviors.SwitchBehavior import SwitchBehavior
+from behaviors.EchoBehavior import EchoBehavior
+from behaviors.DebugBehavior import DebugBehavior
+
+class TestSwitchBehavior(unittest.TestCase):
+    def setUp(self):
+        compReg.initRegistry()
+
+        # add a test registry
+        self.behavior1 = EchoBehavior({'Id': 'behavior1'})
+        self.behavior2 = DebugBehavior({'Id': 'behavior2'})
+        compReg.registerComponent(self.behavior1)
+        compReg.registerComponent(self.behavior2)
+
+        self.switchBehavior = SwitchBehavior({'Id': 'switch', '1': 'behavior1', '2': 'behavior2', 'DefaultBehavior': 'behavior1'})
+        compReg.registerComponent(self.switchBehavior)
+
+    def tearDown(self):
+        pass
+
+    def test_switch_to_behavior1(self):
+        inputs = [{'Data': '1something', 'Location': 'someloc'}]
+        returned = self.switchBehavior.processResponse(inputs, [])
+        assert returned[0][0]['Location'] == 'someloc'
+
+    def test_switch_to_behavior2(self):
+        inputs = [{'Data': '2something'}]
+        returned = self.switchBehavior.processResponse(inputs, [])
+        assert returned[0] == []
+
+if __name__ == '__main__':
+    unittest.main()