def setUpClass(self): # Override to use Manhattan MANHATTAN_BASE_DIR = "%s/data/cities/manhattan" % settings.BASE_DIR BaseCorporation.BASE_CORPORATION_DIR = BaseCorporation.BASE_CORPORATION_TEMPLATE % MANHATTAN_BASE_DIR BaseCorporation.build_dict() self.g = Game() self.g.save()
def test_corporation_auto_created(self):
"""
Corporation should have been created alongside the game
"""
corporations = Corporation.objects.all().order_by('base_corporation_slug')
self.assertEqual(len(corporations), len(BaseCorporation.retrieve_all()))
self.assertEqual(corporations[0].base_corporation.slug, 'c')
def auto_create_corporation(sender, instance, **kwargs): """ Create influence model for new player Takes all the Base corporations and create corresponding Corporations """ base_corporations = BaseCorporation.retrieve_all() for base_corporation in base_corporations: Corporation( base_corporation_slug=base_corporation.slug, game=instance, assets=base_corporation.initials_assets ).save()
def test_corporation_auto_created(self):
"""
Corporation should have been created alongside the game
"""
corporations = Corporation.objects.all().order_by(
'base_corporation_slug')
self.assertEqual(len(corporations),
len(BaseCorporation.retrieve_all()))
self.assertEqual(corporations[0].base_corporation.slug, 'ares')
self.assertEqual(corporations[0].base_corporation.datasteal, 10)
def auto_create_markets_and_corporation(sender, instance, **kwargs):
"""
Takes all the Base corporations and create corresponding Corporations
"""
instance.corporations = {}
markets = {}
base_corporations = BaseCorporation.retrieve_all()
# List distinct market types
for base_corporation in base_corporations:
for market_name in base_corporation.markets.keys():
markets[market_name] = True
# Save them on DB
for market_name in markets.keys():
markets[market_name] = Market(game=instance, name=market_name)
markets[market_name].save()
corporation_markets = []
for base_corporation in base_corporations:
instance.corporations[base_corporation.slug] = Corporation(
base_corporation_slug=base_corporation.slug,
game=instance,
assets=base_corporation.initials_assets,
market_assets=base_corporation.initials_assets,
assets_modifier=0,
)
instance.corporations[base_corporation.slug].save()
for market_name in base_corporation.markets.keys():
cm = CorporationMarket(
corporation=instance.corporations[base_corporation.slug],
market=markets[market_name],
turn=0,
value=base_corporation.markets[market_name]
)
corporation_markets.append(cm)
CorporationMarket.objects.bulk_create(corporation_markets)
def list_corporation_as_markdown(): strings = ["* [%s](corporations/%s.md)" % (c.name, c.slug) for c in BaseCorporation.retrieve_all()] return "\n".join(strings)
def get_base_data():
return {
"base_corporations": BaseCorporation.retrieve_all(),
"city": settings.CITY,
"pods": ["corporation_documentation"]
}
def tearDownClass(self): BaseCorporation.BASE_CORPORATION_DIR = BaseCorporation.BASE_CORPORATION_TEMPLATE % settings.CITY_BASE_DIR BaseCorporation.build_dict() self.g.delete()
