def test__get_actor_info__actor_not_found__None(self): # Arrange personID = 1 self.imdb_get_person_mock.return_value = Person(personID=personID, data={}) # Act actor = get_actor_info(personID) # Assert self.assertIsNone(actor)
def test__get_actor_info__actor_found__actor(self): # Arrange person = Person( personID=1, data={ "name": "Actor1", "headshot": "hs", "actor": [IMDbMovie(movieID=1, data={"title": "mymovie", "year": 2015, "kind": "movie"})], }, ) self.imdb_get_person_mock.return_value = person expected_actor = Actor.create(person, True) # Act actor = get_actor_info(1) # Assert self.assertEquals(actor, expected_actor)