class TaskTest(EngineTestCase): def setUp(self): super(TaskTest, self).setUp() self.c_corporation_market = self.c.get_random_corporation_market() self.c2_corporation_market = self.c2.get_random_corporation_market() self.v = VoteOrder( player=self.p, corporation_market_up=self.c_corporation_market, corporation_market_down=self.c2_corporation_market, ) self.v.save() def test_vote(self): begin_assets_1 = self.c.assets begin_assets_2 = self.c2.assets self.g.resolve_current_turn() self.c = self.reload(self.c) self.c2 = self.reload(self.c2) self.assertEqual(self.c.assets, begin_assets_1 + 1) self.assertEqual(self.c2.assets, begin_assets_2 - 1)
def test_cant_vote_more_than_once(self): o = VoteOrder(corporation_up=self.c, corporation_down=self.c2, player=self.p) # assertNoRaises o.save() o2 = VoteOrder(corporation_up=self.c, corporation_down=self.c2, player=self.p) self.assertRaises(OrderNotAvailable, o2.clean)
def test_corporation_up_and_down(self): begin_assets_1 = self.c.assets begin_assets_2 = self.c2.assets o = VoteOrder(corporation_up=self.c, corporation_down=self.c2, player=self.p) o.save() o.resolve() self.assertEqual(self.reload(self.c).assets, begin_assets_1 + 1) self.assertEqual(self.reload(self.c2).assets, begin_assets_2 - 1)
def test_corporation_up_and_down(self): begin_assets_1 = self.c.assets begin_assets_2 = self.c2.assets o = VoteOrder( corporation_market_up=self.c_corporation_market, corporation_market_down=self.c2_corporation_market, player=self.p ) o.save() o.resolve() self.assertEqual(self.reload(self.c).assets, begin_assets_1 + 1) self.assertEqual(self.reload(self.c2).assets, begin_assets_2 - 1)
def test_cant_vote_more_than_once(self): o = VoteOrder( corporation_market_up=self.c_corporation_market, corporation_market_down=self.c2_corporation_market, player=self.p ) # assertNoRaises o.save() o2 = VoteOrder( corporation_market_up=self.c_corporation_market, corporation_market_down=self.c2_corporation_market, player=self.p ) self.assertRaises(OrderNotAvailable, o2.clean)
class TaskTest(EngineTestCase): def setUp(self): super(TaskTest, self).setUp() self.v = VoteOrder(player=self.p, corporation_up=self.c, corporation_down=self.c2) self.v.save() def test_vote(self): begin_assets_1 = self.c.assets begin_assets_2 = self.c2.assets self.g.resolve_current_turn() self.c = self.reload(self.c) self.c2 = self.reload(self.c2) self.assertEqual(self.c.assets, begin_assets_1 + 1) self.assertEqual(self.c2.assets, begin_assets_2 - 1)