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()))
