Mercurial > parpg-core
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() +