diff tests/test_container.py @ 0:1fd2201f5c36

Initial commit of parpg-core.
author M. George Hansen <technopolitica@gmail.com>
date Sat, 14 May 2011 01:12:35 -0700
parents
children c017773e8443
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test_container.py	Sat May 14 01:12:35 2011 -0700
@@ -0,0 +1,94 @@
+#!/usr/bin/env python
+
+# This file is part of PARPG.
+# 
+# PARPG is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# PARPG is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with PARPG.  If not, see <http://www.gnu.org/licenses/>.
+
+import unittest
+from parpg.objects.base import GameObject, Container, Scriptable, Carryable
+
+class  TestContainer(unittest.TestCase):
+    class ScriptableContainer (GameObject, Container, Scriptable):
+        def __init__ (self, ID, **kwargs):
+            GameObject.__init__(self, ID, **kwargs)
+            Container.__init__(self, **kwargs)
+            Scriptable.__init__(self, **kwargs)
+
+    class NonScriptableContainer (GameObject, Container):
+        def __init__ (self, ID, **kwargs):
+            GameObject.__init__(self, ID, **kwargs)
+            Container.__init__(self, **kwargs)
+
+    class CarryableObject (GameObject, Carryable):
+        def __init__ (self, ID, **kwargs):
+            GameObject.__init__(self, ID, **kwargs)
+            Carryable.__init__(self, **kwargs)
+
+
+    def setUp(self):
+        self.ranOnPlaceItem = False
+        self.ranOnTakeItem = False
+        self.item = self.CarryableObject(6)
+        self.item2 = self.CarryableObject(7)
+
+    def tearDown(self):
+        self.item = None
+        self.item2 = None
+
+
+    def onPlaceItem(self):
+        self.ranOnPlaceItem = True
+
+    def onTakeItem(self):
+        self.ranOnTakeItem = True
+    
+
+    def testPlaceTake(self):
+        """ Test Container mixin Place/Take functions """
+        container = self.NonScriptableContainer(5)
+        self.assertEqual(container.items,{})
+        self.assertEqual(self.item.in_container,None)
+        self.assertEqual(container.count(), 0)
+        container.placeItem(self.item)
+        self.assertEqual(container.items,{0:self.item})
+        self.assertEqual(self.item.in_container, container)
+        self.assertEqual(container.count(), 1)
+        self.assertRaises(Container.SlotBusy, container.placeItem, self.item2, 0)
+        self.assertRaises(ValueError, container.takeItem, self.item2)
+        container.takeItem(self.item)
+        self.assertEqual(container.items, {})
+        self.assertEqual(container.count(), 0)
+
+    def testBulk(self):
+        container = self.NonScriptableContainer(5, capacity=15)
+        self.item.bulk=10
+        self.item2.bulk=7
+        container.placeItem(self.item)
+        self.assertRaises(Container.TooBig, container.placeItem, self.item2)
+
+    def testScripting(self):
+        container = self.ScriptableContainer(5,parpg={'onPlaceItem':(self.onPlaceItem,(),{}),'onTakeItem':(self.onTakeItem,(),{})})
+        self.assertFalse(self.ranOnPlaceItem)
+        self.assertFalse(self.ranOnTakeItem)
+        container.placeItem(self.item)
+        self.assertTrue(self.ranOnPlaceItem)
+        self.assertFalse(self.ranOnTakeItem)
+        self.ranOnPlaceItem = False
+        container.takeItem(self.item)
+        self.assertFalse(self.ranOnPlaceItem)
+        self.assertTrue(self.ranOnTakeItem)
+
+if __name__ == '__main__':
+    unittest.main()
+