Mercurial > parpg-core
changeset 48:c017773e8443
Added unit tests for the container functions.
| author | KarstenBock@gmx.net |
|---|---|
| date | Fri, 02 Sep 2011 15:56:35 +0200 |
| parents | 5a0b7b1f0230 |
| children | 3024ded5ae61 |
| files | tests/test_container.py |
| diffstat | 1 files changed, 112 insertions(+), 66 deletions(-) [+] |
line wrap: on
line diff
--- a/tests/test_container.py Fri Sep 02 15:54:56 2011 +0200 +++ b/tests/test_container.py Fri Sep 02 15:56:35 2011 +0200 @@ -16,79 +16,125 @@ # along with PARPG. If not, see <http://www.gnu.org/licenses/>. import unittest -from parpg.objects.base import GameObject, Container, Scriptable, Carryable + +from parpg.grease.world import BaseWorld +from parpg.grease.entity import Entity +from parpg.components import containable, container 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 GameWorld(BaseWorld): + """GameWorld""" - 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 configure(self): + """Set up the world""" + self.components.containable = containable.Containable() + self.components.container = container.Container() + class Inventory(Entity): + """Enity representing an Iventory""" + + def __init__(self, world, max_bulk, slots): + """Constructor""" + self.container.children = slots + self.container.max_bulk = max_bulk + + + class Item(Entity): + def __init__(self, world, bulk): + """Constructor""" + self.containable.bulk = bulk + 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 - + unittest.TestCase.setUp(self) + self.world = self.GameWorld() + slots_15 = list() + slots_25 = list() + for i in xrange(3): + slots_15.append(None) + + for i in xrange(10): + slots_25.append(None) - 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) + self.inv_15 = self.Inventory(self.world, 15, slots_15) + self.inv_25 = self.Inventory(self.world, 25, slots_25) + self.dagger_1 = self.Item(self.world, 2) + self.sword_1 = self.Item(self.world, 4) + self.axe_1 = self.Item(self.world, 4) + self.mace_1 = self.Item(self.world, 5) + self.spear_1 = self.Item(self.world, 8) + self.dagger_2 = self.Item(self.world, 2) + self.sword_2 = self.Item(self.world, 4) + self.axe_2 = self.Item(self.world, 4) + self.mace_2 = self.Item(self.world, 5) + self.spear_2 = self.Item(self.world, 8) + + def tearDown(self): + self.dagger_1 = None + self.sword_1 = None + self.axe_1 = None + self.mace_1 = None + self.spear_1 = None + self.dagger_2 = None + self.sword_2 = None + self.axe_2 = None + self.mace_2 = None + self.spear_2 = None + self.inv_15 = None + self.inv_25 = None + self.world = None + + def test_State(self): + for child in self.inv_15.container.children: + self.assertIsNone(child) - 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) + for child in self.inv_25.container.children: + self.assertIsNone(child) + + def test_PutTake(self): + self.assertIsNone(container.get_item(self.inv_15.container, 0)) + + container.put_item(self.inv_15.container, self.sword_1.containable, 0) + self.assertIsNotNone(container.get_item(self.inv_15.container, 0)) + self.assertListEqual(self.inv_15.container.children, self.sword_1.containable.container.children) + self.assertEqual(self.inv_15.container.children[0].container, self.sword_1.containable.container) + self.assertEqual(self.inv_15.container.children[0].slot, self.sword_1.containable.slot) + + container.take_item(self.inv_15.container, 0) + self.assertIsNone(self.inv_15.container.children[0]) + + def test_Swap(self): + self.assertIsNone(self.inv_15.container.children[0]) + + container.put_item(self.inv_15.container, self.sword_1.containable, 0) + sword1 = container.get_item(self.inv_15.container, 0) + self.assertEqual(sword1.container, self.sword_1.containable.container) + self.assertEqual(sword1.slot, self.sword_1.containable.slot) + + sword2 = container.put_item(self.inv_15.container, self.dagger_1.containable, 0) + self.assertEqual(sword2.container, sword1.container) + self.assertEqual(sword2.slot, sword1.slot) - 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) + self.assertIsNotNone(container.get_item(self.inv_15.container, 0)) + self.assertListEqual(self.inv_15.container.children, self.dagger_1.containable.container.children) + self.assertEqual(self.inv_15.container.children[0].container, self.dagger_1.containable.container) + self.assertEqual(self.inv_15.container.children[0].slot, self.dagger_1.containable.slot) + + def test_BulkSlots(self): + container.put_item(self.inv_15.container, self.sword_1.containable) + container.put_item(self.inv_25.container, self.sword_2.containable) + self.assertEqual(container.get_total_bulk(self.inv_15.container), self.sword_1.containable.bulk) + self.assertEqual(container.get_total_bulk(self.inv_25.container), self.sword_2.containable.bulk) + + container.put_item(self.inv_15.container, self.axe_1.containable) + container.put_item(self.inv_25.container, self.axe_2.containable) + self.assertEqual(container.get_total_bulk(self.inv_15.container), container.get_total_bulk(self.inv_25.container)) + + self.assertRaises(container.BulkLimitError, container.put_item, self.inv_15.container, self.spear_1.containable) + container.put_item(self.inv_25.container, self.spear_2.containable) -if __name__ == '__main__': - unittest.main() + container.put_item(self.inv_15.container, self.dagger_1.containable) + container.put_item(self.inv_25.container, self.dagger_2.containable) + self.assertNotEqual(container.get_total_bulk(self.inv_15.container), container.get_total_bulk(self.inv_25.container)) + self.assertRaises(container.NoFreeSlotError, container.put_item, self.inv_15.container, self.mace_1.containable) + container.put_item(self.inv_25.container, self.mace_2.containable)