def setUp(self): self.inventory = GenericStorage() self.owner_inventory = GenericStorage() class Instance(object): def __init__(self, comp): self.comp = comp def get_component(self, x): class Comp(object): inventory = self.comp return Comp() self.tradepost = TradePostComponent() self.tradepost.instance = Instance(self.inventory) self.tradepost.instance.owner = Instance(self.owner_inventory) self.tradepost.initialize() class Timer(object): def add_call(self, x): pass def get_ticks(self, x): return 100 Scheduler.create_instance(timer=Timer())
def test_reset_all(self): s = GenericStorage() s.alter(1, 5) s.alter(2, 3) s.reset_all() self.assertEqual(s[1], 0) self.assertEqual(s[2], 0)
def test_alter(self): s = GenericStorage() self.assertEqual(s.alter(1, 5), 0) self.assertEqual(s.alter(2, 3), 0) self.assertEqual(s[1], 5) self.assertEqual(s[2], 3) self.assertEqual(s.alter(1, 10), 0) self.assertEqual(s[1], 15)
def test_get_item(self): s = GenericStorage() s.alter(1, 5) self.assertEqual(s[1], 5) self.assertEqual(s[2], 0)
def test_sum_of_stored_resources(self): s = GenericStorage() s.alter(1, 5) s.alter(2, 3) self.assertEqual(s.get_sum_of_stored_resources(), 8)
def test_limit(self): s = GenericStorage() self.assertEqual(s.get_limit(), sys.maxint) self.assertEqual(s.get_limit(1), sys.maxint)