From 9e342041f05e88f8d1987a48fdcdc10c14ef095f Mon Sep 17 00:00:00 2001
From: eugue <eug.sun@gmail.com>
Date: Sat, 12 Feb 2011 20:07:15 -0500
Subject: Added SwitchBehavior and its unit test file. Could be used to control
 games.

---
 tests/TestSwitchBehavior.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 tests/TestSwitchBehavior.py

(limited to 'tests/TestSwitchBehavior.py')

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()
-- 
cgit v1.2.3


From d8ba03006e2ea1400e80caded738e79c186e8de3 Mon Sep 17 00:00:00 2001
From: eugue <eug.sun@gmail.com>
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.

---
 behaviors/SwitchBehavior.py | 28 ++++++++++++++++++++--------
 tests/TestSwitchBehavior.py | 12 +++++++++---
 2 files changed, 29 insertions(+), 11 deletions(-)

(limited to 'tests/TestSwitchBehavior.py')

diff --git a/behaviors/SwitchBehavior.py b/behaviors/SwitchBehavior.py
index 1fb755d..1377b42 100644
--- a/behaviors/SwitchBehavior.py
+++ b/behaviors/SwitchBehavior.py
@@ -1,24 +1,36 @@
 from operationscore.Behavior import *
 import util.ComponentRegistry as compReg
+import json
 
 class SwitchBehavior(Behavior):
     """
     SwitchBehavior is a behavior that transform into different behaviors base on the input data.
-    The behavior expects Args in the form of <Prefix>s mapping to <Behavior ID>s. The behavior detects the prefix on the data and use the corresponding Behavior to process the data and return the outputs.
+    The behavior expects a JSON formatted argument 'PrefixToBehavior' that maps prefixes to behaviors. The behavior detects the prefix on the data and use the corresponding Behavior to process the data and return the outputs.
     In Config file, include:
-    <Prefix>Behavior's ID<Prefix>
-    <DefaultBehavior>Default behavior's ID<DefaultBehavior>
+      <PrefixToBehavior>JSON format dict with prefix keys and behavior ID values</PrefixToBehavior>
+      <DefaultBehavior>Default behavior's ID</DefaultBehavior>
+    An example config excerpt:
+      <Behavior>
+        <Class>behaviors.SwitchBehavior</Class>
+          <Args>
+            <Id>switch</Id>
+            <PrefixToBehavior>{'@':'game1', '#':'game2', '$':'game3'}</PrefixToBehavior>
+            <DefaultBehavior>game1</DefaultBehavior>
+          </Args>
+      </Behavior>
     """
     def behaviorInit(self):
         self.defaultBehavior = compReg.getComponent(self['DefaultBehavior'])
+        self.prefixDict = json.loads(self['PrefixToBehavior'])
         self.currBehavior = None
+        self.setBehavior(self.defaultBehavior)
         
     def processResponse(self, sInputs, rInputs):
         dataStr = sInputs[-1]['Data']
-        if dataStr[0] in self.argDict:
-            self.currBehavior = compReg.getComponent(self[dataStr[0]])
+        if dataStr[0] in self.prefixDict:
+            self.setBehavior(compReg.getComponent(self.prefixDict[dataStr[0]]))
             sInputs[-1]['Data'] = sInputs[-1]['Data'][1:] # remove prefix
-            return self.currBehavior.processResponse(sInputs, rInputs)
-        else:
-            return self.defaultBehavior.processsResponse(sInputs, rInputs)
+        return self.currBehavior.processResponse(sInputs, rInputs)
     
+    def setBehavior(self, behavior):
+        self.currBehavior = behavior
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