def test1(self):
aux1 = Vuelos(destino='BCN')
aux2 = Vuelos(destino='ITA')
aux_vuelo = [aux1, aux2]
x = Viajes(lista_pasajeros=['p1', 'p2', 'p3'], vuelos=aux_vuelo)
y = PaymentData('MASTERCARD', 'Pepito', '4546', '50')
x.payment_V2(y)
assert (x.payment_data.tipo_tarjeta == 'MASTERCARD')
def test2(self):
aux1 = Vuelos(destino='BCN')
aux2 = Vuelos(destino='ITA')
aux_vuelo = [aux1, aux2]
y = PaymentData('MASTERCARD', 'Pepito', '4546', '50')
x = Viajes(user=User,
lista_pasajeros=['p1', 'p2', 'p3'],
vuelos=aux_vuelo)
aux = User('Pol', '12345678J', '08390', '123456789', '*****@*****.**')
fallo = x.payment_V2(y, 1)
Bank.do_payment = MagicMock(return_value=False)
i = Bank()
assert i.do_payment(aux, y) == fallo