def test_current_portfolio_value(self): """ Current portfolio value """ user = UserFactory() self.assertEqual(0, user.current_portfolio_value) event = EventFactory() bet = BetFactory(user=user, event=event) self.assertEqual(50, user.current_portfolio_value) bet.outcome = Bet.BET_OUTCOME_CHOICES.NO bet.has = 2 bet.save() self.assertEqual(100, user.current_portfolio_value)
def test_update_portfolio_value(self): """ Update portfolio_value """ price = 90 user = UserFactory() event = EventFactory(current_sell_for_price=price) BetFactory(user=user, event=event, has=1, outcome=True) self.assertEqual(0, user.portfolio_value) update_portfolio_value() user.refresh_from_db()
def test_current_portfolio_value(self): """ Current portfolio value """ user = UserFactory() self.assertEqual(0, user.current_portfolio_value) event = EventFactory() bet = BetFactory(user=user, event=event) self.assertEqual(50, user.current_portfolio_value) bet.outcome = Bet.NO bet.has = 2 bet.save() self.assertEqual(100, user.current_portfolio_value)
def test_get_newest_results(self): """ Get newest results """ users = UserFactory.create_batch(2) events = EventFactory.create_batch(5) BetFactory(user=users[0], event=events[0]) bet2 = BetFactory(user=users[0], event=events[1]) bet3 = BetFactory(user=users[0], event=events[2]) bet4 = BetFactory(user=users[0], event=events[3]) bet5 = BetFactory(user=users[1], event=events[4]) events[1].outcome = Event.CANCELLED events[1].save() events[2].outcome = Event.FINISHED_YES events[2].save() events[3].outcome = Event.FINISHED_NO events[3].save() events[4].outcome = Event.FINISHED_YES events[4].save() bet2.is_new_resolved = True bet2.save() bet3.is_new_resolved = True bet3.save() bet4.is_new_resolved = True bet4.save() bet5.is_new_resolved = True bet5.save() self.assertEqual([bet2, bet3, bet4], list(users[0].get_newest_results())) self.assertEqual([bet5], list(users[1].get_newest_results()))
def test_get_newest_results(self): """ Get newest results """ users = UserFactory.create_batch(2) events = EventFactory.create_batch(5) BetFactory(user=users[0], event=events[0]) bet2 = BetFactory(user=users[0], event=events[1]) bet3 = BetFactory(user=users[0], event=events[2]) bet4 = BetFactory(user=users[0], event=events[3]) bet5 = BetFactory(user=users[1], event=events[4]) events[1].outcome = Event.EVENT_OUTCOME_CHOICES.CANCELLED events[1].save() events[2].outcome = Event.EVENT_OUTCOME_CHOICES.FINISHED_YES events[2].save() events[3].outcome = Event.EVENT_OUTCOME_CHOICES.FINISHED_NO events[3].save() events[4].outcome = Event.EVENT_OUTCOME_CHOICES.FINISHED_YES events[4].save() bet2.is_new_resolved = True bet2.save() bet3.is_new_resolved = True bet3.save() bet4.is_new_resolved = True bet4.save() bet5.is_new_resolved = True bet5.save() self.assertEqual([bet2, bet3, bet4], list(users[0].get_newest_results())) self.assertEqual([bet5], list(users[1].get_newest_results()))