def setUp(self): Subscription.objects.create(name="Anderson Marques", cpf="00644321229", email="*****@*****.**", phone="984593967") self.model_admin = SubscriptionModelAdmin(Subscription, admin.site)
class SubscriptionModelTestAdmin(TestCase): def setUp(self): Subscription.objects.create(name="Anderson Marques", cpf="00644321229", email="*****@*****.**", phone="984593967") self.model_admin = SubscriptionModelAdmin(Subscription, admin.site) def test_has_action(self): """action mark_as_paid should be installed""" self.assertIn('mark_as_paid', self.model_admin.actions) def test_mark_all(self): """Should mark all selected subscription as paid""" self.call_action() self.assertEqual(1, Subscription.objects.filter(paid=True).count()) def test_message(self): """It should send a message to the user""" mock = self.call_action() mock.assert_called_once_with(None, "1 inscrição foi marcada como paga.") def call_action(self): query_set = Subscription.objects.all() mock = Mock() old_message_user = SubscriptionModelAdmin.message_user SubscriptionModelAdmin.message_user = mock self.model_admin.mark_as_paid(None, query_set) SubscriptionModelAdmin.message_user = old_message_user return mock
def setUp(self): Subscription.objects.create( name="Anderson Marques", cpf="00644321229", email="*****@*****.**", phone="984593967" ) self.model_admin = SubscriptionModelAdmin(Subscription, admin.site)
class SubscriptionModelTestAdmin(TestCase): def setUp(self): Subscription.objects.create( name="Anderson Marques", cpf="00644321229", email="*****@*****.**", phone="984593967" ) self.model_admin = SubscriptionModelAdmin(Subscription, admin.site) def test_has_action(self): """action mark_as_paid should be installed""" self.assertIn('mark_as_paid', self.model_admin.actions) def test_mark_all(self): """Should mark all selected subscription as paid""" self.call_action() self.assertEqual(1, Subscription.objects.filter(paid=True).count()) def test_message(self): """It should send a message to the user""" mock = self.call_action() mock.assert_called_once_with(None, "1 inscrição foi marcada como paga.") def call_action(self): query_set = Subscription.objects.all() mock = Mock() old_message_user = SubscriptionModelAdmin.message_user SubscriptionModelAdmin.message_user = mock self.model_admin.mark_as_paid(None, query_set) SubscriptionModelAdmin.message_user = old_message_user return mock