def test_model_train_duplicate(self):
train = Train(name='t1',
from_city=self.city_A,
to_city=self.city_B,
travel_time=111)
with self.assertRaises(ValidationError):
train.full_clean()
def test_model_train_duplicate(self):
'''Testing Trains for doppelgangers (testing for and to city parameters)'''
train = Train(name='t1',
from_city=self.city_A,
to_city=self.city_B,
travel_time=228)
with self.assertRaises(ValidationError):
train.full_clean()
def test_modul_train_dublicate(self):
# Тестирование возникновения ошибки при создании дубля поезда
train = Train(name='t1',
from_city=self.city_A,
to_city=self.city_B,
travel_time=129)
with self.assertRaises(ValidationError):
train.full_clean()
def test_model_train_duplicate(self):
'''Тестирование при возникновении ошибок при создании дупликата поезда'''
train = Train(name='t165',
from_city=self.city_A,
to_city=self.city_B,
travel_time=129)
with self.assertRaises(ValidationError):
train.full_clean()
def index(request):
error_messages = None
if request.method == 'POST':
try:
train = Train(description = request.POST['description'],
username = request.POST['username'],
departure = request.POST['departure'],
status = Status.objects.get(name = 'scheduled'))
train.full_clean()
if train.departure < datetime.now():
exception = ValidationError('Departure should be in future')
raise exception
train.save()
return redirect(reverse('trains.views.view', kwargs = {
'train_id': train.pk
}))
except ValidationError, error:
error_messages = error.messages
def setUp(self) -> None:
self.city_A = City.objects.create(name='A')
self.city_B = City.objects.create(name='B')
self.city_C = City.objects.create(name='C')
self.city_D = City.objects.create(name='D')
self.city_E = City.objects.create(name='E')
trains = [
Train(name='t1',
from_city=self.city_A,
to_city=self.city_B,
travel_time=9),
Train(name='t2',
from_city=self.city_A,
to_city=self.city_C,
travel_time=7),
Train(name='t3',
from_city=self.city_A,
to_city=self.city_C,
travel_time=10),
Train(name='t4',
from_city=self.city_B,
to_city=self.city_A,
travel_time=11),
Train(name='t5',
from_city=self.city_C,
to_city=self.city_B,
travel_time=6),
Train(name='t6',
from_city=self.city_B,
to_city=self.city_D,
travel_time=8),
Train(name='t7',
from_city=self.city_B,
to_city=self.city_E,
travel_time=3),
Train(name='t8',
from_city=self.city_D,
to_city=self.city_E,
travel_time=4),
Train(name='t9',
from_city=self.city_E,
to_city=self.city_D,
travel_time=5),
]
Train.objects.bulk_create(trains)
def add(request):
form = TrainForm()
if request.method == 'POST':
form = TrainForm(request.POST)
if form.is_valid():
new_train = Train()
new_train.name = form.cleaned_data['name']
new_train.from_city = form.cleaned_data['from_city']
new_train.to_city = form.cleaned_data['to_city']
new_train.travel_time = form.cleaned_data['travel_time']
new_train.save()
messages.success(request, 'Поезд успешно добавлен')
return redirect('train:home')
return render(request, 'trains/add.html', {'form': form})
def test_model_train_duplicate(self):
try:
train = Train(name='t2',
from_city=self.city_B,
to_city=self.city_D,
travel_time=4)
train.full_clean()
except ValidationError as e:
self.assertEqual(
{'name': ['Потяг with this Номер потягу already exists.']},
e.message_dict)
try:
train = Train(name='t12',
from_city=self.city_B,
to_city=self.city_D,
travel_time=8)
train.full_clean()
except ValidationError as e:
self.assertEqual({'__all__': ['Виправте час в дорозі']},
e.message_dict)
def test_model_train_duplicate(self):
try:
train = Train(name='t2',
from_city=self.city_B,
to_city=self.city_D,
travel_time=4)
train.full_clean()
except ValidationError as e:
self.assertEqual(
{'name': ['Поезд with this Номер поезда already exists.']},
e.message_dict)
try:
train = Train(name='t12',
from_city=self.city_B,
to_city=self.city_D,
travel_time=8)
train.full_clean()
except ValidationError as e:
self.assertEqual({'__all__': ['Измените время в пути']},
e.message_dict)
def test_model_train_train_duplicate(self):
train = Train(name='t1234',
from_city=self.city_A,
to_city=self.city_B,
travel_time=9)
with self.assertRaises(ValidationError):
train.full_clean()
try:
train.full_clean()
except ValidationError as e:
self.assertEqual({'__all__': ['Change the travel time']},
e.message_dict)
self.assertIn('Change the travel time', e.messages)
def test_model_train_train2_duplicate(self):
'''Тестирование при возникновении ошибок при создании дупликата поезда'''
train = Train(name='t15',
from_city=self.city_A,
to_city=self.city_B,
travel_time=9)
with self.assertRaises(ValidationError):
train.full_clean()
try:
train.full_clean()
except ValidationError as e:
self.assertEqual({'__all__': ['Измените время в пути']},
e.message_dict)
self.assertIn('Измените время в пути', e.message)
def test_model_train_time(self):
'''Тестирование возникновения ошибки времени в пути поезда'''
train = Train(name='t153',
from_city=self.city_A,
to_city=self.city_B,
travel_time=9)
with self.assertRaises(ValidationError):
train.full_clean()
try:
train.full_clean()
except ValidationError as e:
self.assertEqual({'__all__': ['Нужно изменить время в пути']},
e.message_dict)
self.assertIn('Нужно изменить время в пути', e.messages)
def test_modul_train_train_dublicate(self):
# Тестирование возникновения ошибки при дублировании времени
train = Train(name='t1234',
from_city=self.city_A,
to_city=self.city_B,
travel_time=9)
with self.assertRaises(ValidationError):
train.full_clean()
try:
train.full_clean()
except ValidationError as e:
self.assertEqual({'__all__': ['Измените время в пути']},
e.message_dict)
self.assertIn('Измените время в пути', e.messages)
def test_model_train_duplicate(self):
# функция проверки дублирования номеров поездов
try:
train = Train(name='t2',
from_city=self.city_B,
to_city=self.city_D,
travel_time=4)
train.full_clean()
except ValidationError as e:
self.assertEqual(
{'name': ['Поезд with this Номер поезда already exists.']},
e.message_dict)
# функция проверки дублирования времени поездов при одоинаковых городах
try:
train = Train(name='t12',
from_city=self.city_B,
to_city=self.city_D,
travel_time=8)
train.full_clean()
except ValidationError as e:
self.assertEqual({'__all__': ['Измените время в пути']},
e.message_dict)
def test_model_train_time_duplicate(self):
'''Testing Trains for doppelgangers (testing travel time)'''
train = Train(name='t1',
from_city=self.city_A,
to_city=self.city_B,
travel_time=228)
with self.assertRaises(ValidationError):
train.full_clean()
try:
train.full_clean()
except ValidationError as err:
self.assertEqual(
{'name': ['Train with this Train already exists.']},
err.message_dict)
self.assertIn('Train with this Train already exists.',
err.messages)
def setUp(self):
self.city_A = City.objects.create(name='A')
self.city_B = City.objects.create(name='B')
self.city_C = City.objects.create(name='C')
self.city_D = City.objects.create(name='D')
self.city_E = City.objects.create(name='E')
t1 = Train(name='t1',
from_city=self.city_A,
to_city=self.city_B,
travel_time=9)
t1.save()
t2 = Train(name='t2',
from_city=self.city_B,
to_city=self.city_D,
travel_time=8)
t2.save()
t3 = Train(name='t3',
from_city=self.city_A,
to_city=self.city_C,
travel_time=7)
t3.save()
t4 = Train(name='t4',
from_city=self.city_C,
to_city=self.city_B,
travel_time=6)
t4.save()
t5 = Train(name='t5',
from_city=self.city_B,
to_city=self.city_E,
travel_time=3)
t5.save()
t6 = Train(name='t6',
from_city=self.city_B,
to_city=self.city_A,
travel_time=11)
t6.save()
t7 = Train(name='t7',
from_city=self.city_A,
to_city=self.city_C,
travel_time=10)
t7.save()
t8 = Train(name='t8',
from_city=self.city_E,
to_city=self.city_D,
travel_time=5)
t8.save()
t9 = Train(name='t9',
from_city=self.city_D,
to_city=self.city_E,
travel_time=4)
t9.save()
name = "LLHQ",
location = "Emeryville, CA",
latitude = "37.842139", longitude = "-122.289215").save()
dt = datetime.datetime.now() - datetime.timedelta(days = 5)
numDests = Restaurant.objects.all().count()
rid = random.randint(0, numDests)
numDests = Restaurant.objects.all().count()
numUsers = User.objects.all().count()
for cap in User.objects.all():
rid = random.randint(1, numDests)
dest = Restaurant.objects.get(id = rid)
id1 = random.randint(1, numUsers)
id2 = random.randint(1, numUsers)
p1 = User.objects.get(id = id1)
p2 = User.objects.get(id = id2)
t = Train(departure_time = dt,
captain = cap,
destination = dest)
t.save()
t.passengers = []
t.passengers.add(p1)
t.passengers.add(p2)
t.save()
print 'Created initial data in DB!'