def test_get_payment_method_source_type(self):
"""
Source Type name should be displayed as the Payment method
when the credit card wasn't used to purchase a product.
"""
order = self.create_order()
source = factories.SourceFactory(order=order)
payment_method = ReceiptResponseView().get_payment_method(order)
self.assertEqual(payment_method, source.source_type.name)
def test_get_payment_method_credit_card_purchase(self):
"""
Credit card type and Source label should be displayed as the Payment method
when a Credit card was used to purchase a product.
"""
order = self.create_order()
source = factories.SourceFactory(order=order, card_type='Dummy Card', label='Test')
payment_method = ReceiptResponseView().get_payment_method(order)
self.assertEqual(payment_method, '{} {}'.format(source.card_type, source.label))
def test_get_payment_method_no_source(self):
""" Payment method should be None when an Order has no Payment source. """
order = self.create_order()
payment_method = ReceiptResponseView().get_payment_method(order)
self.assertEqual(payment_method, None)
