from datetime import timedelta from unittest import mock from django.contrib.auth import get_user_model from django.core import mail from django.test import TestCase from django.utils import timezone from bda.models import ( Attribution, Participant, Salle, Spectacle, SpectacleRevente, Tirage, ) User = get_user_model() class SpectacleReventeTests(TestCase): fixtures = ['gestioncof/management/data/custommail.json'] def setUp(self): now = timezone.now() self.t = Tirage.objects.create( title='Tirage', ouverture=now - timedelta(days=7), fermeture=now - timedelta(days=3), active=True, ) self.s = Spectacle.objects.create( title='Spectacle', date=now + timedelta(days=20), location=Salle.objects.create(name='Salle', address='Address'), price=10.5, slots=5, tirage=self.t, listing=False, ) self.seller = Participant.objects.create( user=User.objects.create( username='seller', email='seller@mail.net'), tirage=self.t, ) self.p1 = Participant.objects.create( user=User.objects.create(username='part1', email='part1@mail.net'), tirage=self.t, ) self.p2 = Participant.objects.create( user=User.objects.create(username='part2', email='part2@mail.net'), tirage=self.t, ) self.p3 = Participant.objects.create( user=User.objects.create(username='part3', email='part3@mail.net'), tirage=self.t, ) self.attr = Attribution.objects.create( participant=self.seller, spectacle=self.s, ) self.rev = SpectacleRevente.objects.create( attribution=self.attr, seller=self.seller, ) def test_tirage(self): revente = self.rev wanted_by = [self.p1, self.p2, self.p3] revente.confirmed_entry = wanted_by with mock.patch('bda.models.random.choice') as mc: # Set winner to self.p1. mc.return_value = self.p1 revente.tirage() # Call to random.choice used participants in wanted_by. mc_args, _ = mc.call_args self.assertEqual(set(mc_args[0]), set(wanted_by)) self.assertEqual(revente.soldTo, self.p1) self.assertTrue(revente.tirage_done) mails = {m.to[0]: m for m in mail.outbox} self.assertEqual(len(mails), 4) m_seller = mails['seller@mail.net'] self.assertListEqual(m_seller.to, ['seller@mail.net']) self.assertListEqual(m_seller.reply_to, ['part1@mail.net']) m_winner = mails['part1@mail.net'] self.assertListEqual(m_winner.to, ['part1@mail.net']) self.assertCountEqual( [mails['part2@mail.net'].to, mails['part3@mail.net'].to], [['part2@mail.net'], ['part3@mail.net']], )