def test_get_subsribers(self):
user1= User.objects.create_user('Mar2','*****@*****.**','j1')
user3= User.objects.create_user('koupepia','*****@*****.**','m1')
user4= User.objects.create_user('Ginnis1','*****@*****.**','m1')
tour = Tour()
tour.owner_id = user1
tour.save()
tour.subscribers.add(user3, user4)
tour.name = "Open Day"
# tour.tour_date = "2010-12-12"
tour.save()
self.assertEqual(get_subscribers("Open Day").count(),2)
def test_get_owner(self):
user1 = User.objects.create_user('John','*****@*****.**','j1')
user2= User.objects.create_user('Mike','*****@*****.**','m1')
user3= User.objects.create_user('kokoko','*****@*****.**','m1')
user4= User.objects.create_user('lalal','*****@*****.**','m1')
tour = Tour()
tour.id = "1"
tour.name = "hello"
tour.owner_id = user1
tour.save()
self.assertEqual(get_owner("hello"),'John')
def test_delete_tour(self):
user1 = User.objects.create_user('kokokoko','*****@*****.**','j1')
user2= User.objects.create_user('kk','*****@*****.**','m1')
user3= User.objects.create_user('ff','*****@*****.**','m1')
user4= User.objects.create_user('dd','*****@*****.**','m1')
tour = Tour()
tour.id = "1"
n = tour.name = "hello"
tour.owner_id = user1
tour.save()
delete_tour(n)
tour1 = Tour()
n1 = tour1.name = "hello"
tour1.owner_id = user1
tour1.save()
self.assertEqual(get_owner("hello"), "kokokoko")
