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")