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)