def test_volverRealizarPago(self, mock_Bank):
num_passatgers = 3
viaje = Viaje(usr, num_passatgers)
viaje.addDestino(vuelo_1)
viaje.addDestino(vuelo_2)
viaje.user.seleccioMetodePagament(metode_pagament)
mock_Bank.do_payment.return_value = False
self.assertTrue(mock_Bank.reservarYpagar())
def test_errorPago(self):
with pytest.raises(ValueError):
viaje1 = Viaje(usr, 2)
viaje1.addDestino(Flights("1234", "Sant Esteve de les Roures",
100))
viaje1.addDestino(Flights(4737, "Talavera de la Reina", 50))
viaje1.user.seleccioMetodePagament(metode_pagament)
viaje1.reservarYpagar()
def test_errorReservaVols(self):
with pytest.raises(ValueError):
viaje1 = Viaje(usr, 2)
metode_pagament = PaymentData("Mastercard", "Jesus Gil Padre",
"123456", "4242")
viaje1.addDestino(Flights("1234", "Sant Esteve de les Roures",
100))
viaje1.addDestino(Flights(4737, "Talavera de la Reina", 50))
viaje1.user.seleccioMetodePagament(metode_pagament)
viaje1.reservarYpagar()
def test_vuelos_ok(self):
viaje1 = Viaje(usr, 2)
dest1 = Flights("84522", "Villa Puri", 100)
dest2 = Flights("5745", "Guarroman", 50)
dests = [dest1, dest2]
ok = [dest1, dest2]
self.assertEqual(len(viaje1.lista_vuelos), 0)
for i in range(len(dests)):
viaje1.addDestino(dests[i])
for k, j in enumerate(viaje1.lista_vuelos):
self.assertEqual(j, ok[k])
def test_destinos_ok(self):
viaje1 = Viaje(usr, 2)
dest1 = Flights("1234", "Sant Esteve de les Roures", 100)
dest2 = Flights("4737", "Talavera de la Reina", 50)
dests = [dest1, dest2]
ok = ["Sant Esteve de les Roures", "Talavera de la Reina"]
self.assertEqual(len(viaje1.lista_destinos), 0)
for i in range(len(dests)):
viaje1.addDestino(dests[i])
for k, j in enumerate(viaje1.lista_destinos):
self.assertEqual(j, ok[k])
def test_rmDestinoVuelo(self):
viaje1 = Viaje(usr, 2)
viaje2 = Viaje(usr, 4)
vuelos = [
Flights("58555", "España", 75),
Flights("12345", "Italia", 60)
]
for i in range(len(vuelos)):
viaje1.addDestino(vuelos[i])
viaje2.addDestino(vuelos[i])
viajes = [viaje1, viaje2]
for destino in range(len(viajes)):
viajes[destino].rmDestino("58555", "España", 75)
for i in range(len(viajes[destino].lista_vuelos)):
self.assertEqual(viajes[destino].lista_vuelos[i], vuelos[-1])
def test_rmDestinoPrecio(self):
viaje1 = Viaje(usr, 2)
viaje2 = Viaje(usr, 4)
vuelos = [
Flights("58555", "España", 100),
Flights("12345", "Italia", 200)
]
soladd = [0, 0, 200, 400]
solT = [600, 400, 1200, 800]
aux = []
for i in range(len(vuelos)):
aux.append(viaje1.sumaPrecios())
aux.append(viaje2.sumaPrecios())
viaje1.addDestino(vuelos[i])
viaje2.addDestino(vuelos[i])
self.assertEqual(aux, soladd)
viajes = [viaje1, viaje2]
aux = []
for destino in range(len(viajes)):
aux.append(viajes[destino].sumaPrecios())
viajes[destino].rmDestino("58555", "España", 100)
aux.append(viajes[destino].sumaPrecios())
self.assertEqual(solT, aux)
def test_pagoEsperado(self):
viaje1 = Viaje(usr, 2)
viaje1.addDestino(Flights("1234", "Sant Esteve de les Roures", 100))
viaje1.addDestino(Flights("4737", "Talavera de la Reina", 50))
viaje1.user.seleccioMetodePagament(metode_pagament)
self.assertEqual(metode_pagament, viaje1.user.dades_pagament)