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)
