def test_movie_get_all(self):
movie = MovieModel(movie_schema.load(self.movie1))
movie.save()
movie = MovieModel(movie_schema.load(self.movie2))
movie.save()
res = MovieService.get_all()
json_data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(json_data.__len__(), 2)
def test_movie_create(self):
res = MovieService.create(movie_schema.load(self.movie1))
json_data = json.loads(res.data)
self.assertEqual(res.status_code, 201)
self.assertEqual(json_data['title'], 'Star Wars')
res = MovieService.get_all()
json_data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(json_data.__len__(), 1)
def get(self):
"""Get all movies"""
return MovieService.get_all()