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 test_movie_update(self):
movie = MovieModel(movie_schema.load(self.movie1))
movie.save()
res = MovieService.update(1, movie_schema.load(self.movie2))
json_data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(json_data.get('title'), 'Singularity')
def test_movie_get_by_title(self):
movie = MovieModel(movie_schema.load(self.movie1))
movie.save()
res = MovieService.get_by_title(movie.title)
json_data = json.loads(res.data)
self.assertEqual(res.status_code, 200)
self.assertEqual(json_data[0]['title'], 'Star Wars')
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 delete(self, movie_id: int):
"""Delete Movie"""
return MovieService.delete_by_id(movie_id)
def put(self, movie_id: int):
"""Update Move"""
return MovieService.update(movie_id, request.parsed_obj)
def post(self):
"""Search by title"""
req_data = request.json
return MovieService.get_by_title(req_data['title'])
def get(self):
"""Get all movies"""
return MovieService.get_all()
def post(self):
"""Create Movie"""
return MovieService.create(request.parsed_obj)
def test_delete_movie(self):
movie = MovieModel(movie_schema.load(self.movie1))
movie.save()
res = MovieService.delete_by_id(1)
self.assertEqual(res.status_code, 204)