def create(self, validated_data):
"""Planet creation method overwrite in order to get the number of movie appearances"""
movie_appearances = get_planet_appearances(validated_data['name'])
planet = Planet(
terrain=validated_data['terrain'],
climate=validated_data['climate'],
name=validated_data['name'],
movie_appearances=movie_appearances
)
planet.save()
return planet
def test_check_invalid_planet_count(self):
"""Check a name with a space between it"""
invalid = get_planet_appearances('Lorem Ipsum')
self.assertEqual(invalid, 0)
def test_check_alderaan_count(self):
"""First Test case to Check an normal name"""
alderaan = get_planet_appearances('Alderaan')
self.assertEqual(alderaan, 2)
def test_check_polis_massa_count(self):
"""Check a name with a space between it"""
polis_massa = get_planet_appearances('Polis Massa')
self.assertEqual(polis_massa, 1)