mirror of
https://github.com/PokeAPI/pokeapi
synced 2024-11-22 19:33:10 +00:00
4005 lines
175 KiB
Python
4005 lines
175 KiB
Python
from __future__ import unicode_literals
|
|
# from django.test import TestCase
|
|
from rest_framework import status
|
|
from rest_framework.test import APITestCase
|
|
from rest_framework.test import APIClient
|
|
from rest_framework.reverse import reverse
|
|
from pokemon_v2.models import *
|
|
import json
|
|
|
|
test_host = 'http://testserver'
|
|
media_sprites = '/media/sprites/{}'
|
|
api_v2 = '/api/v2'
|
|
|
|
|
|
"""
|
|
Data Initializers
|
|
"""
|
|
|
|
class APIData():
|
|
|
|
"""
|
|
Gender Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_gender_data(self, name='gndr'):
|
|
|
|
gender = Gender.objects.create (
|
|
name = name,
|
|
)
|
|
gender.save()
|
|
|
|
return gender
|
|
|
|
|
|
"""
|
|
Language Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_language_data(self, name='lang'):
|
|
|
|
language = Language.objects.create (
|
|
iso639 = 'ts',
|
|
iso3166 = 'tt',
|
|
name = name,
|
|
official = True,
|
|
order = 1,
|
|
)
|
|
language.save()
|
|
|
|
return language
|
|
|
|
@classmethod
|
|
def setup_language_name_data(self, language, name='lang nm'):
|
|
|
|
local_language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
language_name = LanguageName.objects.create (
|
|
language = language,
|
|
local_language = local_language,
|
|
name = name
|
|
)
|
|
language_name.save()
|
|
|
|
return language_name
|
|
|
|
|
|
"""
|
|
Region Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_region_data(self, name='reg'):
|
|
|
|
region = Region.objects.create (
|
|
name = name
|
|
)
|
|
region.save()
|
|
|
|
return region
|
|
|
|
@classmethod
|
|
def setup_region_name_data(self, region, name='reg nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
region_name = RegionName.objects.create (
|
|
region = region,
|
|
language = language,
|
|
name = name
|
|
)
|
|
region_name.save()
|
|
|
|
return region_name
|
|
|
|
|
|
"""
|
|
Generation Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_generation_data(self, region=None, name='gen'):
|
|
|
|
region = region or self.setup_region_data(
|
|
name='reg for '+name)
|
|
|
|
generation = Generation.objects.create (
|
|
region = region,
|
|
name = name
|
|
)
|
|
generation.save()
|
|
|
|
return generation
|
|
|
|
@classmethod
|
|
def setup_generation_name_data(self, generation, name='gen nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
generation_name = GenerationName.objects.create (
|
|
generation = generation,
|
|
language = language,
|
|
name = name
|
|
)
|
|
generation_name.save()
|
|
|
|
return generation_name
|
|
|
|
|
|
"""
|
|
Version Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_version_group_data(self, name='ver grp', generation=None):
|
|
|
|
generation = generation or self.setup_generation_data(
|
|
name='gen for '+name)
|
|
|
|
version_group = VersionGroup.objects.create (
|
|
name = name,
|
|
generation = generation,
|
|
order = 1
|
|
)
|
|
version_group.save()
|
|
|
|
return version_group
|
|
|
|
@classmethod
|
|
def setup_version_group_region_data(self, version_group=None, region=None):
|
|
|
|
version_group_region = VersionGroupRegion.objects.create (
|
|
version_group = version_group,
|
|
region = region
|
|
)
|
|
version_group_region.save()
|
|
|
|
return version_group_region
|
|
|
|
@classmethod
|
|
def setup_version_data(self, version_group=None, name='ver'):
|
|
|
|
version = Version.objects.create (
|
|
name = name,
|
|
version_group = version_group,
|
|
)
|
|
version.save()
|
|
|
|
return version
|
|
|
|
@classmethod
|
|
def setup_version_name_data(self, version, name='ver nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
version_name = VersionName.objects.create (
|
|
version = version,
|
|
language = language,
|
|
name = name
|
|
)
|
|
version_name.save()
|
|
|
|
return version_name
|
|
|
|
|
|
"""
|
|
Ability Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_ability_data(self, name='ablty', generation=None):
|
|
|
|
generation = generation or self.setup_generation_data(
|
|
name='gen for '+name)
|
|
|
|
ability = Ability.objects.create (
|
|
name = name,
|
|
generation = generation,
|
|
is_main_series = False
|
|
)
|
|
ability.save()
|
|
|
|
return ability
|
|
|
|
@classmethod
|
|
def setup_ability_name_data(self, ability, name='ablty nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
ability_name = AbilityName.objects.create (
|
|
ability = ability,
|
|
language = language,
|
|
name = name
|
|
)
|
|
ability_name.save()
|
|
|
|
return ability_name
|
|
|
|
@classmethod
|
|
def setup_ability_effect_text_data(self, ability, short_effect='ablty shrt efct', effect='ablty efct'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+effect)
|
|
|
|
ability_effect_text = AbilityEffectText.objects.create (
|
|
ability = ability,
|
|
language = language,
|
|
short_effect = short_effect,
|
|
effect = effect
|
|
)
|
|
ability_effect_text.save()
|
|
|
|
return ability_effect_text
|
|
|
|
@classmethod
|
|
def setup_ability_change_data(self, ability):
|
|
|
|
version_group = self.setup_version_group_data(
|
|
name='ver grp for ablty chng')
|
|
|
|
ability_change = AbilityChange.objects.create (
|
|
ability = ability,
|
|
version_group = version_group
|
|
)
|
|
ability_change.save()
|
|
|
|
return ability_change
|
|
|
|
@classmethod
|
|
def setup_ability_change_effect_text_data(self, ability_change, effect='ablty change efct'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+effect)
|
|
|
|
ability_change_effect_text = AbilityChangeEffectText.objects.create (
|
|
ability_change = ability_change,
|
|
language = language,
|
|
effect = effect
|
|
)
|
|
ability_change_effect_text.save()
|
|
|
|
return ability_change_effect_text
|
|
|
|
@classmethod
|
|
def setup_ability_flavor_text_data(self, ability, flavor_text='ablty flvr txt'):
|
|
|
|
version_group = self.setup_version_group_data(
|
|
name='ver grp for '+flavor_text)
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+flavor_text)
|
|
|
|
ability_flavor_text = AbilityFlavorText.objects.create (
|
|
ability = ability,
|
|
version_group = version_group,
|
|
language = language,
|
|
flavor_text = flavor_text
|
|
)
|
|
ability_flavor_text.save()
|
|
|
|
return ability_flavor_text
|
|
|
|
|
|
"""
|
|
Item Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_item_attribute_data(self, name="itm attr"):
|
|
|
|
item_attribute = ItemAttribute.objects.create (
|
|
name = name,
|
|
)
|
|
item_attribute.save()
|
|
|
|
return item_attribute
|
|
|
|
@classmethod
|
|
def setup_item_attribute_name_data(self, item_attribute, name='itm attr nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
item_attribute_name = ItemAttributeName.objects.create (
|
|
item_attribute = item_attribute,
|
|
name = name,
|
|
language = language
|
|
)
|
|
item_attribute_name.save()
|
|
|
|
return item_attribute_name
|
|
|
|
@classmethod
|
|
def setup_item_attribute_description_data(self, item_attribute, description='itm attr desc'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
item_attribute_description = ItemAttributeDescription.objects.create (
|
|
item_attribute = item_attribute,
|
|
description = description,
|
|
language = language
|
|
)
|
|
item_attribute_description.save()
|
|
|
|
return item_attribute_description
|
|
|
|
@classmethod
|
|
def setup_item_attribute_map_data(self, item, item_attribute):
|
|
|
|
item_attribute_map = ItemAttributeMap (
|
|
item = item,
|
|
item_attribute = item_attribute
|
|
)
|
|
item_attribute_map.save()
|
|
|
|
return item_attribute_map
|
|
|
|
@classmethod
|
|
def setup_item_fling_effect_data(self, name="itm flng efct"):
|
|
|
|
item_fling_effect = ItemFlingEffect.objects.create (
|
|
name = name,
|
|
)
|
|
item_fling_effect.save()
|
|
|
|
return item_fling_effect
|
|
|
|
def setup_item_fling_effect_effect_text_data(self, item_fling_effect, effect='itm flng efct efct txt'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+effect)
|
|
|
|
item_fling_effect_effect_text = ItemFlingEffectEffectText.objects.create (
|
|
item_fling_effect = item_fling_effect,
|
|
effect = effect,
|
|
language = language
|
|
)
|
|
item_fling_effect_effect_text.save()
|
|
|
|
return item_fling_effect_effect_text
|
|
|
|
@classmethod
|
|
def setup_item_pocket_data(self, name='itm pkt'):
|
|
|
|
item_pocket = ItemPocket.objects.create (
|
|
name = name,
|
|
)
|
|
item_pocket.save()
|
|
|
|
return item_pocket
|
|
|
|
@classmethod
|
|
def setup_item_pocket_name_data(self, item_pocket, name='itm pkt nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
item_pocket_name = ItemPocketName.objects.create (
|
|
item_pocket = item_pocket,
|
|
name = name,
|
|
language = language
|
|
)
|
|
item_pocket_name.save()
|
|
|
|
return item_pocket_name
|
|
|
|
@classmethod
|
|
def setup_item_category_data(self, name="itm ctgry", item_pocket=None):
|
|
|
|
item_pocket = item_pocket or self.setup_item_pocket_data(
|
|
name='itm pkt for '+name)
|
|
|
|
item_category = ItemCategory.objects.create (
|
|
name = name,
|
|
item_pocket = item_pocket
|
|
)
|
|
item_category.save()
|
|
|
|
return item_category
|
|
|
|
def setup_item_category_name_data(self, item_category, name='itm ctgry nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
item_category_name = ItemCategoryName.objects.create (
|
|
item_category = item_category,
|
|
name = name,
|
|
language = language
|
|
)
|
|
item_category_name.save()
|
|
|
|
return item_category_name
|
|
|
|
@classmethod
|
|
def setup_item_sprites_data(self, item, default=True):
|
|
|
|
sprite_path = '/media/sprites/items/%s.png'
|
|
|
|
sprites = {
|
|
'default': sprite_path % item.id if default else None,
|
|
}
|
|
|
|
item_sprites = ItemSprites.objects.create (
|
|
item = item,
|
|
sprites = json.dumps(sprites)
|
|
)
|
|
item_sprites.save()
|
|
|
|
return item_sprites
|
|
|
|
@classmethod
|
|
def setup_item_data(self, item_category=None, item_fling_effect=None, name='itm', cost=100, fling_power=100):
|
|
|
|
item = Item.objects.create (
|
|
name = name,
|
|
item_category = item_category,
|
|
cost = 300,
|
|
fling_power = 100,
|
|
item_fling_effect = item_fling_effect
|
|
)
|
|
item.save()
|
|
|
|
return item
|
|
|
|
@classmethod
|
|
def setup_item_name_data(self, item, name='itm nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
item_name = ItemName.objects.create (
|
|
item = item,
|
|
name = name,
|
|
language = language
|
|
)
|
|
item_name.save()
|
|
|
|
return item_name
|
|
|
|
@classmethod
|
|
def setup_item_effect_text_data(self, item, short_effect='ablty shrt efct', effect='ablty efct'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+effect)
|
|
|
|
item_effect_text = ItemEffectText.objects.create (
|
|
item = item,
|
|
language = language,
|
|
short_effect = short_effect,
|
|
effect = effect
|
|
)
|
|
item_effect_text.save()
|
|
|
|
return item_effect_text
|
|
|
|
@classmethod
|
|
def setup_item_flavor_text_data(self, item, flavor_text='itm flvr txt'):
|
|
|
|
version_group = self.setup_version_group_data(
|
|
name='ver grp for '+flavor_text)
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+flavor_text)
|
|
|
|
item_flavor_text = ItemFlavorText.objects.create (
|
|
item = item,
|
|
version_group = version_group,
|
|
language = language,
|
|
flavor_text = flavor_text
|
|
)
|
|
item_flavor_text.save()
|
|
|
|
return item_flavor_text
|
|
|
|
@classmethod
|
|
def setup_item_game_index_data(self, item, game_index=0):
|
|
|
|
generation = self.setup_generation_data(
|
|
name='gen for itm gm indx')
|
|
|
|
item_game_index = ItemGameIndex.objects.create (
|
|
item = item,
|
|
game_index = game_index,
|
|
generation = generation
|
|
)
|
|
item_game_index.save()
|
|
|
|
return item_game_index
|
|
|
|
|
|
|
|
"""
|
|
Contest Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_contest_type_data(self, name='cntst tp'):
|
|
|
|
contest_type = ContestType.objects.create (
|
|
name = name,
|
|
)
|
|
contest_type.save()
|
|
|
|
return contest_type
|
|
|
|
@classmethod
|
|
def setup_contest_type_name_data(self, contest_type, name='cntst tp nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
contest_type_name = ContestTypeName.objects.create (
|
|
language = language,
|
|
name = name,
|
|
contest_type = contest_type
|
|
)
|
|
contest_type_name.save()
|
|
|
|
return contest_type_name
|
|
|
|
@classmethod
|
|
def setup_contest_effect_data(self, appeal=2, jam=0):
|
|
|
|
contest_effect = ContestEffect.objects.create (
|
|
appeal = appeal,
|
|
jam = jam
|
|
)
|
|
contest_effect.save()
|
|
|
|
return contest_effect
|
|
|
|
@classmethod
|
|
def setup_contest_effect_flavor_text_data(self, contest_effect, flavor_text='cntst efct flvr txt'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+flavor_text)
|
|
|
|
contest_effect_flavor_text = ContestEffectFlavorText.objects.create (
|
|
language = language,
|
|
flavor_text = flavor_text,
|
|
contest_effect = contest_effect
|
|
)
|
|
contest_effect_flavor_text.save()
|
|
|
|
return contest_effect_flavor_text
|
|
|
|
@classmethod
|
|
def setup_contest_effect_effect_text_data(self, contest_effect, effect='cntst efct efct txt'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+effect)
|
|
|
|
contest_effect_effect_text = ContestEffectEffectText.objects.create (
|
|
language = language,
|
|
effect = effect,
|
|
contest_effect = contest_effect
|
|
)
|
|
contest_effect_effect_text.save()
|
|
|
|
return contest_effect_effect_text
|
|
|
|
@classmethod
|
|
def setup_super_contest_effect_data(self, appeal=2):
|
|
|
|
super_contest_effect = SuperContestEffect.objects.create (
|
|
appeal = appeal,
|
|
)
|
|
super_contest_effect.save()
|
|
|
|
return super_contest_effect
|
|
|
|
@classmethod
|
|
def setup_super_contest_effect_flavor_text_data(self, super_contest_effect, flavor_text='spr cntst efct flvr txt'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+flavor_text)
|
|
|
|
super_contest_effect_flavor_text = SuperContestEffectFlavorText.objects.create (
|
|
language = language,
|
|
flavor_text = flavor_text,
|
|
super_contest_effect = super_contest_effect
|
|
)
|
|
super_contest_effect_flavor_text.save()
|
|
|
|
return super_contest_effect_flavor_text
|
|
|
|
|
|
"""
|
|
Berry Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_berry_flavor_data(self, contest_type=None, name="bry flvr"):
|
|
|
|
contest_type = contest_type or self.setup_contest_type_data(
|
|
name='cntst tp for bry flvr')
|
|
|
|
berry_flavor = BerryFlavor.objects.create (
|
|
name = name,
|
|
contest_type = contest_type
|
|
)
|
|
berry_flavor.save()
|
|
|
|
return berry_flavor
|
|
|
|
@classmethod
|
|
def setup_berry_flavor_name_data(self, berry_flavor, name='bry flvr nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
berry_flavor_name = BerryFlavorName.objects.create (
|
|
language = language,
|
|
name = name,
|
|
berry_flavor = berry_flavor
|
|
)
|
|
berry_flavor_name.save()
|
|
|
|
return berry_flavor_name
|
|
|
|
@classmethod
|
|
def setup_berry_firmness_data(self, name='bry frmns'):
|
|
|
|
berry_firmness = BerryFirmness.objects.create (
|
|
name = name,
|
|
)
|
|
berry_firmness.save()
|
|
|
|
return berry_firmness
|
|
|
|
@classmethod
|
|
def setup_berry_firmness_name_data(self, berry_firmness, name='bry frmns nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
berry_firmness_name = BerryFirmnessName.objects.create (
|
|
language = language,
|
|
name = name,
|
|
berry_firmness = berry_firmness
|
|
)
|
|
berry_firmness_name.save()
|
|
|
|
return berry_firmness_name
|
|
|
|
|
|
@classmethod
|
|
def setup_berry_flavor_map_data(self, berry, berry_flavor, potency=20):
|
|
|
|
berry_flavor_map = BerryFlavorMap (
|
|
berry = berry,
|
|
berry_flavor = berry_flavor,
|
|
potency = potency
|
|
)
|
|
berry_flavor_map.save()
|
|
|
|
return berry_flavor_map
|
|
|
|
|
|
@classmethod
|
|
def setup_berry_data(self, berry_firmness=None, item=None, natural_gift_type=None, name='bry', natural_gift_power=50, size=20, max_harvest=5, growth_time=2, soil_dryness=15, smoothness=25):
|
|
|
|
item = item or self.setup_item_data(
|
|
name='itm for '+name)
|
|
|
|
berry_firmness = berry_firmness or self.setup_berry_firmness_data(
|
|
name='bry frmns for '+name)
|
|
|
|
berry = Berry.objects.create (
|
|
name = name,
|
|
item = item,
|
|
berry_firmness = berry_firmness,
|
|
natural_gift_power = natural_gift_power,
|
|
natural_gift_type = natural_gift_type,
|
|
size = size,
|
|
max_harvest = max_harvest,
|
|
growth_time = growth_time,
|
|
soil_dryness = soil_dryness,
|
|
smoothness = smoothness
|
|
)
|
|
berry.save()
|
|
|
|
return berry
|
|
|
|
|
|
"""
|
|
Egg Group Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_egg_group_data(self, name='egg grp'):
|
|
|
|
egg_group = EggGroup.objects.create (
|
|
name = name,
|
|
)
|
|
egg_group.save()
|
|
|
|
return egg_group
|
|
|
|
@classmethod
|
|
def setup_egg_group_name_data(self, egg_group, name='ntr nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
egg_group_name = EggGroupName.objects.create (
|
|
egg_group = egg_group,
|
|
language = language,
|
|
name = name
|
|
)
|
|
egg_group_name.save()
|
|
|
|
return egg_group_name
|
|
|
|
|
|
"""
|
|
Growth Rate Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_growth_rate_data(self, name='grth rt', formula="pie*1000"):
|
|
|
|
growth_rate = GrowthRate (
|
|
name = name,
|
|
formula = formula
|
|
)
|
|
growth_rate.save()
|
|
|
|
return growth_rate
|
|
|
|
def setup_growth_rate_description_data(self, growth_rate, description='grth rt desc'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
growth_rate_description = GrowthRateDescription.objects.create (
|
|
growth_rate = growth_rate,
|
|
description = description,
|
|
language = language
|
|
)
|
|
growth_rate_description.save()
|
|
|
|
return growth_rate_description
|
|
|
|
|
|
"""
|
|
Location Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_location_data(self, region=None, name='lctn'):
|
|
|
|
region = region or self.setup_region_data(
|
|
name='rgn for '+name)
|
|
|
|
location = Location (
|
|
name = name,
|
|
region = region
|
|
)
|
|
location.save()
|
|
|
|
return location
|
|
|
|
@classmethod
|
|
def setup_location_game_index_data(self, location, game_index=0):
|
|
|
|
generation = self.setup_generation_data(
|
|
name='gen for itm gm indx')
|
|
|
|
location_game_index = LocationGameIndex.objects.create (
|
|
location = location,
|
|
game_index = game_index,
|
|
generation = generation
|
|
)
|
|
location_game_index.save()
|
|
|
|
return location_game_index
|
|
|
|
@classmethod
|
|
def setup_location_name_data(self, location, name='lctn nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
location_name = LocationName.objects.create (
|
|
language = language,
|
|
name = name,
|
|
location = location
|
|
)
|
|
location_name.save()
|
|
|
|
return location_name
|
|
|
|
@classmethod
|
|
def setup_location_area_data(self, location=None, name='lctn area', game_index=0):
|
|
|
|
location = location or self.setup_location_data(
|
|
name='lctn for '+name)
|
|
|
|
location_area = LocationArea (
|
|
location = location,
|
|
name = name,
|
|
game_index = game_index
|
|
)
|
|
location_area.save()
|
|
|
|
return location_area
|
|
|
|
@classmethod
|
|
def setup_location_area_name_data(self, location_area, name='lctn area nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
location_area_name = LocationAreaName.objects.create (
|
|
language = language,
|
|
name = name,
|
|
location_area = location_area
|
|
)
|
|
location_area_name.save()
|
|
|
|
return location_area_name
|
|
|
|
|
|
"""
|
|
Type Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_type_data(self, name='tp', move_damage_class=None, generation=None):
|
|
|
|
move_damage_class = move_damage_class or self.setup_move_damage_class_data(
|
|
name="mv dmg cls for "+name)
|
|
|
|
generation = generation or self.setup_generation_data(
|
|
name='rgn for '+name)
|
|
|
|
type = Type (
|
|
name = name,
|
|
generation = generation,
|
|
move_damage_class = move_damage_class
|
|
)
|
|
type.save()
|
|
|
|
return type
|
|
|
|
@classmethod
|
|
def setup_type_name_data(self, type, name='tp nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
type_name = TypeName.objects.create (
|
|
language = language,
|
|
name = name,
|
|
type = type
|
|
)
|
|
type_name.save()
|
|
|
|
return type_name
|
|
|
|
@classmethod
|
|
def setup_type_game_index_data(self, type, game_index=0):
|
|
|
|
generation = self.setup_generation_data(
|
|
name='gen for tp gm indx')
|
|
|
|
type_game_index = TypeGameIndex.objects.create (
|
|
type = type,
|
|
game_index = game_index,
|
|
generation = generation
|
|
)
|
|
type_game_index.save()
|
|
|
|
return type_game_index
|
|
|
|
|
|
"""
|
|
Move Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_move_ailment_data(self, name='mv almnt'):
|
|
|
|
move_ailment = MoveMetaAilment.objects.create (
|
|
name = name
|
|
)
|
|
move_ailment.save()
|
|
|
|
return move_ailment
|
|
|
|
@classmethod
|
|
def setup_move_ailment_name_data(self, move_ailment, name='mv almnt nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
move_ailment_name = MoveMetaAilmentName.objects.create (
|
|
move_meta_ailment = move_ailment,
|
|
language = language,
|
|
name = name
|
|
)
|
|
move_ailment_name.save()
|
|
|
|
return move_ailment_name
|
|
|
|
@classmethod
|
|
def setup_move_battle_style_data(self, name='mv btl stl'):
|
|
|
|
move_battle_style = MoveBattleStyle.objects.create (
|
|
name = name
|
|
)
|
|
move_battle_style.save()
|
|
|
|
return move_battle_style
|
|
|
|
@classmethod
|
|
def setup_move_battle_style_name_data(self, move_battle_style, name='mv almnt nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
move_battle_style_name = MoveBattleStyleName.objects.create (
|
|
move_battle_style = move_battle_style,
|
|
language = language,
|
|
name = name
|
|
)
|
|
move_battle_style_name.save()
|
|
|
|
return move_battle_style_name
|
|
|
|
@classmethod
|
|
def setup_move_category_data(self, name='mv ctgry'):
|
|
|
|
move_category = MoveMetaCategory.objects.create (
|
|
name = name
|
|
)
|
|
move_category.save()
|
|
|
|
return move_category
|
|
|
|
@classmethod
|
|
def setup_move_category_description_data(self, move_category, description='mv ctgry desc'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
move_category_description = MoveMetaCategoryDescription.objects.create (
|
|
move_meta_category = move_category,
|
|
language = language,
|
|
description = description
|
|
)
|
|
move_category_description.save()
|
|
|
|
return move_category_description
|
|
|
|
@classmethod
|
|
def setup_move_effect_data(self):
|
|
|
|
move_effect = MoveEffect.objects.create ()
|
|
move_effect.save()
|
|
|
|
return move_effect
|
|
|
|
@classmethod
|
|
def setup_move_effect_effect_text_data(self, move_effect, effect='mv efct efct txt', short_effect='mv efct shrt efct txt'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+effect)
|
|
|
|
effect_effect_text = MoveEffectEffectText.objects.create (
|
|
effect = effect,
|
|
short_effect = short_effect,
|
|
move_effect = move_effect,
|
|
language = language
|
|
)
|
|
effect_effect_text.save()
|
|
|
|
return effect_effect_text
|
|
|
|
@classmethod
|
|
def setup_move_damage_class_data(self, name='mv dmg cls'):
|
|
|
|
move_damage_class = MoveDamageClass.objects.create (
|
|
name = name
|
|
)
|
|
move_damage_class.save()
|
|
|
|
return move_damage_class
|
|
|
|
@classmethod
|
|
def setup_move_damage_class_name_data(self, move_damage_class, name='mv dmg cls nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
move_damage_class_name = MoveDamageClassName.objects.create (
|
|
move_damage_class = move_damage_class,
|
|
language = language,
|
|
name = name
|
|
)
|
|
move_damage_class_name.save()
|
|
|
|
return move_damage_class_name
|
|
|
|
@classmethod
|
|
def setup_move_damage_class_description_data(self, move_damage_class, description='mv dmg cls desc'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
move_damage_class_description = MoveDamageClassDescription.objects.create (
|
|
move_damage_class = move_damage_class,
|
|
language = language,
|
|
description = description
|
|
)
|
|
move_damage_class_description.save()
|
|
|
|
return move_damage_class_description
|
|
|
|
@classmethod
|
|
def setup_move_learn_method_data(self, name='mv lrn mthd'):
|
|
|
|
move_learn_method = MoveLearnMethod.objects.create (
|
|
name = name
|
|
)
|
|
move_learn_method.save()
|
|
|
|
return move_learn_method
|
|
|
|
@classmethod
|
|
def setup_move_learn_method_name_data(self, move_learn_method, name='mv lrn mthd nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
move_learn_method_name = MoveLearnMethodName.objects.create (
|
|
move_learn_method = move_learn_method,
|
|
language = language,
|
|
name = name
|
|
)
|
|
move_learn_method_name.save()
|
|
|
|
return move_learn_method_name
|
|
|
|
@classmethod
|
|
def setup_move_learn_method_description_data(self, move_learn_method, description='mv lrn mthd desc'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
move_learn_method_description = MoveLearnMethodDescription.objects.create (
|
|
move_learn_method = move_learn_method,
|
|
language = language,
|
|
description = description
|
|
)
|
|
move_learn_method_description.save()
|
|
|
|
return move_learn_method_description
|
|
|
|
@classmethod
|
|
def setup_move_target_data(self, name='mv trgt'):
|
|
|
|
move_target = MoveTarget.objects.create (
|
|
name = name
|
|
)
|
|
move_target.save()
|
|
|
|
return move_target
|
|
|
|
@classmethod
|
|
def setup_move_target_name_data(self, move_target, name='mv trgt nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
move_target_name = MoveTargetName.objects.create (
|
|
move_target = move_target,
|
|
language = language,
|
|
name = name
|
|
)
|
|
move_target_name.save()
|
|
|
|
return move_target_name
|
|
|
|
@classmethod
|
|
def setup_move_target_description_data(self, move_target, description='mv trgt desc'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
move_target_description = MoveTargetDescription.objects.create (
|
|
move_target = move_target,
|
|
language = language,
|
|
description = description
|
|
)
|
|
move_target_description.save()
|
|
|
|
return move_target_description
|
|
|
|
@classmethod
|
|
def setup_contest_combo_data(self, first_move, second_move):
|
|
|
|
contest_combo = ContestCombo.objects.create (
|
|
first_move = first_move,
|
|
second_move = second_move
|
|
)
|
|
contest_combo.save()
|
|
|
|
return contest_combo
|
|
|
|
@classmethod
|
|
def setup_version_group_move_learn_method_data(self, version_group=None, move_learn_method=None):
|
|
|
|
version_group_move_learn_method = VersionGroupMoveLearnMethod.objects.create (
|
|
version_group = version_group,
|
|
move_learn_method = move_learn_method
|
|
)
|
|
version_group_move_learn_method.save()
|
|
|
|
return version_group_move_learn_method
|
|
|
|
@classmethod
|
|
def setup_super_contest_combo_data(self, first_move, second_move):
|
|
|
|
super_contest_combo = SuperContestCombo.objects.create (
|
|
first_move = first_move,
|
|
second_move = second_move
|
|
)
|
|
super_contest_combo.save()
|
|
|
|
return super_contest_combo
|
|
|
|
@classmethod
|
|
def setup_move_data(self, contest_type=None, contest_effect=None, super_contest_effect=None, generation=None, move_damage_class=None, move_effect=None, move_target=None, type=None, name='mv', power=20, pp=20, accuracy=80, priority=0, effect_chance=50):
|
|
|
|
contest_type = contest_type or self.setup_contest_type_data(
|
|
name='cntst tp for '+name)
|
|
|
|
contest_effect = contest_effect or self.setup_contest_effect_data()
|
|
|
|
super_contest_effect = super_contest_effect or self.setup_super_contest_effect_data()
|
|
|
|
generation = generation or self.setup_generation_data(
|
|
name='gen for '+name)
|
|
|
|
type = type or self.setup_type_data(
|
|
name='tp for '+name)
|
|
|
|
move_target = move_target or self.setup_move_target_data(
|
|
name='mv trgt for '+name)
|
|
|
|
move_damage_class = move_damage_class or self.setup_move_damage_class_data(
|
|
name='mv dmg cls for '+name)
|
|
|
|
move = Move.objects.create (
|
|
name = name,
|
|
generation = generation,
|
|
type = type,
|
|
power = power,
|
|
pp = pp,
|
|
accuracy = accuracy,
|
|
priority = priority,
|
|
move_target = move_target,
|
|
move_damage_class = move_damage_class,
|
|
move_effect = move_effect,
|
|
move_effect_chance = effect_chance,
|
|
contest_type = contest_type,
|
|
contest_effect = contest_effect,
|
|
super_contest_effect = super_contest_effect
|
|
)
|
|
move.save()
|
|
|
|
return move
|
|
|
|
@classmethod
|
|
def setup_move_name_data(self, move, name='mv nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
move_name = MoveName.objects.create (
|
|
move = move,
|
|
language = language,
|
|
name = name
|
|
)
|
|
move_name.save()
|
|
|
|
return move_name
|
|
|
|
@classmethod
|
|
def setup_move_meta_data(self, move, move_ailment=None, move_category=None, min_hits=1, max_hits=1, min_turns=1, max_turns=1, drain=0, healing=0, crit_rate=0, ailment_chance=0, flinch_chance=0, stat_chance=0):
|
|
|
|
move_ailment = move_ailment or self.setup_move_ailment_data()
|
|
|
|
move_category = move_category or self.setup_move_category_data()
|
|
|
|
move_meta = MoveMeta (
|
|
move = move,
|
|
move_meta_category = move_category,
|
|
move_meta_ailment = move_ailment,
|
|
min_hits = min_hits,
|
|
max_hits = max_hits,
|
|
min_turns = min_turns,
|
|
max_turns = max_turns,
|
|
drain = drain,
|
|
healing = healing,
|
|
crit_rate = crit_rate,
|
|
ailment_chance = ailment_chance,
|
|
flinch_chance = flinch_chance,
|
|
stat_chance = stat_chance
|
|
)
|
|
move_meta.save()
|
|
|
|
return move_meta
|
|
|
|
@classmethod
|
|
def setup_move_change_data(self, move=None, type=None, move_effect=None, version_group=None, power=20, pp=20, accuracy=80, priority=0, effect_chance=50):
|
|
|
|
version_group = version_group or self.setup_version_group_data(
|
|
name='ver grp for mv chng')
|
|
|
|
move_change = MoveChange.objects.create (
|
|
move = move,
|
|
version_group = version_group,
|
|
type = type,
|
|
power = power,
|
|
pp = pp,
|
|
accuracy = accuracy,
|
|
move_effect = move_effect,
|
|
move_effect_chance = effect_chance
|
|
)
|
|
move_change.save()
|
|
|
|
return move_change
|
|
|
|
@classmethod
|
|
def setup_move_effect_change_data(self, move_effect=None):
|
|
|
|
version_group = self.setup_version_group_data(
|
|
name='ver grp for mv chng')
|
|
|
|
move_effect_change = MoveEffectChange.objects.create (
|
|
move_effect = move_effect,
|
|
version_group = version_group
|
|
)
|
|
move_effect_change.save()
|
|
|
|
return move_effect_change
|
|
|
|
@classmethod
|
|
def setup_move_effect_change_effect_text_data(self, move_effect_change=None, effect='mv efct chng efct txt'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+effect)
|
|
|
|
move_effect_change_effect_text = MoveEffectChangeEffectText.objects.create (
|
|
move_effect_change = move_effect_change,
|
|
language = language,
|
|
effect = effect
|
|
)
|
|
move_effect_change_effect_text.save()
|
|
|
|
return move_effect_change_effect_text
|
|
|
|
|
|
|
|
"""
|
|
Stat Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_stat_data(self, name='stt', is_battle_only=True, game_index=1):
|
|
|
|
move_damage_class = self.setup_move_damage_class_data(
|
|
name='mv dmg cls for '+name)
|
|
|
|
stat = Stat.objects.create (
|
|
name = name,
|
|
is_battle_only = is_battle_only,
|
|
move_damage_class = move_damage_class,
|
|
game_index = game_index
|
|
)
|
|
stat.save()
|
|
|
|
return stat
|
|
|
|
@classmethod
|
|
def setup_stat_name_data(self, stat, name='stt nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
stat_name = StatName.objects.create (
|
|
stat = stat,
|
|
language = language,
|
|
name = name
|
|
)
|
|
stat_name.save()
|
|
|
|
return stat_name
|
|
|
|
@classmethod
|
|
def setup_move_stat_change_data(self, move, stat=None, change=1):
|
|
|
|
stat = stat or self.setup_stat_data(
|
|
name='stt for mv')
|
|
|
|
move_stat_change = MoveMetaStatChange.objects.create (
|
|
move = move,
|
|
stat = stat,
|
|
change = change
|
|
)
|
|
move_stat_change.save()
|
|
|
|
return move_stat_change
|
|
|
|
@classmethod
|
|
def setup_pokeathlon_stat_data(self, name='pkathln stt'):
|
|
|
|
pokeathlon_stat = PokeathlonStat.objects.create (
|
|
name = name
|
|
)
|
|
pokeathlon_stat.save()
|
|
|
|
return pokeathlon_stat
|
|
|
|
@classmethod
|
|
def setup_pokeathlon_stat_name_data(self, pokeathlon_stat, name='pkathln stt nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
pokeathlon_stat_name = PokeathlonStatName.objects.create (
|
|
pokeathlon_stat = pokeathlon_stat,
|
|
language = language,
|
|
name = name
|
|
)
|
|
pokeathlon_stat_name.save()
|
|
|
|
return pokeathlon_stat_name
|
|
|
|
|
|
"""
|
|
Characteristic Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_characteristic_data(self, gene_mod_5=0, stat=None):
|
|
|
|
stat = stat or self.setup_stat_data(
|
|
name='stt for char')
|
|
|
|
characteristic = Characteristic.objects.create (
|
|
stat = stat,
|
|
gene_mod_5 = gene_mod_5
|
|
)
|
|
characteristic.save()
|
|
|
|
return characteristic
|
|
|
|
@classmethod
|
|
def setup_characteristic_description_data(self, characteristic, description='char desc'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
characteristic_description = CharacteristicDescription.objects.create (
|
|
characteristic = characteristic,
|
|
language = language,
|
|
description = description
|
|
)
|
|
characteristic_description.save()
|
|
|
|
return characteristic_description
|
|
|
|
|
|
"""
|
|
Nature Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_nature_data(self, decreased_stat=None, increased_stat=None, likes_flavor=None, hates_flavor=None, name='ntr', game_index=1):
|
|
|
|
nature = Nature.objects.create (
|
|
name = name,
|
|
decreased_stat = decreased_stat,
|
|
increased_stat = increased_stat,
|
|
hates_flavor = hates_flavor,
|
|
likes_flavor = likes_flavor,
|
|
game_index = game_index
|
|
)
|
|
nature.save()
|
|
|
|
return nature
|
|
|
|
@classmethod
|
|
def setup_nature_name_data(self, nature, name='ntr nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
nature_name = NatureName.objects.create (
|
|
nature = nature,
|
|
language = language,
|
|
name = name
|
|
)
|
|
nature_name.save()
|
|
|
|
return nature_name
|
|
|
|
@classmethod
|
|
def setup_nature_pokeathlon_stat_data(self, nature=None, pokeathlon_stat=None, max_change=1):
|
|
|
|
nature_pokeathlon_stat = NaturePokeathlonStat.objects.create (
|
|
nature = nature,
|
|
pokeathlon_stat = pokeathlon_stat,
|
|
max_change = max_change,
|
|
)
|
|
nature_pokeathlon_stat.save()
|
|
|
|
return nature_pokeathlon_stat
|
|
|
|
@classmethod
|
|
def setup_nature_battle_style_preference_data(self, nature=None, move_battle_style=None, low_hp_preference=10, high_hp_preference=20):
|
|
|
|
nature_battle_style_preference = NatureBattleStylePreference.objects.create (
|
|
nature = nature,
|
|
move_battle_style = move_battle_style,
|
|
low_hp_preference = low_hp_preference,
|
|
high_hp_preference = high_hp_preference
|
|
)
|
|
nature_battle_style_preference.save()
|
|
|
|
return nature_battle_style_preference
|
|
|
|
|
|
|
|
"""
|
|
Pokedex Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_pokedex_data(self, region=None, name='pkdx'):
|
|
|
|
region = region or self.setup_region_data(
|
|
name='rgn for '+name)
|
|
|
|
pokedex = Pokedex.objects.create (
|
|
name = name,
|
|
region = region,
|
|
)
|
|
pokedex.save()
|
|
|
|
return pokedex
|
|
|
|
@classmethod
|
|
def setup_pokedex_name_data(self, pokedex, name='pkdx nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
pokedex_name = PokedexName.objects.create (
|
|
pokedex = pokedex,
|
|
language = language,
|
|
name = name
|
|
)
|
|
pokedex_name.save()
|
|
|
|
return pokedex_name
|
|
|
|
@classmethod
|
|
def setup_pokedex_description_data(self, pokedex, description='pkdx desc'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
pokedex_description = PokedexDescription.objects.create (
|
|
pokedex = pokedex,
|
|
language = language,
|
|
description = description
|
|
)
|
|
pokedex_description.save()
|
|
|
|
return pokedex_description
|
|
|
|
@classmethod
|
|
def setup_pokedex_version_group_data(self, pokedex, version_group=None):
|
|
|
|
version_group = version_group or self.setup_language_data(
|
|
name='ver grp for pkdx')
|
|
|
|
pokedex_version_group = PokedexVersionGroup.objects.create (
|
|
pokedex = pokedex,
|
|
version_group = version_group
|
|
)
|
|
pokedex_version_group.save()
|
|
|
|
return pokedex_version_group
|
|
|
|
|
|
"""
|
|
Pokemon Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_pokemon_habitat_data(self, name='pkm hbtt'):
|
|
|
|
pokemon_habitat = PokemonHabitat.objects.create (
|
|
name = name,
|
|
)
|
|
pokemon_habitat.save()
|
|
|
|
return pokemon_habitat
|
|
|
|
@classmethod
|
|
def setup_pokemon_habitat_name_data(self, pokemon_habitat, name='pkm hbtt nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
pokemon_habitat_name = PokemonHabitatName.objects.create (
|
|
pokemon_habitat = pokemon_habitat,
|
|
language = language,
|
|
name = name
|
|
)
|
|
pokemon_habitat_name.save()
|
|
|
|
return pokemon_habitat_name
|
|
|
|
@classmethod
|
|
def setup_pokemon_color_data(self, name='pkm clr'):
|
|
|
|
pokemon_color = PokemonColor.objects.create (
|
|
name = name,
|
|
)
|
|
pokemon_color.save()
|
|
|
|
return pokemon_color
|
|
|
|
@classmethod
|
|
def setup_pokemon_color_name_data(self, pokemon_color, name='pkm clr nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
pokemon_color_name = PokemonColorName.objects.create (
|
|
pokemon_color = pokemon_color,
|
|
language = language,
|
|
name = name
|
|
)
|
|
pokemon_color_name.save()
|
|
|
|
return pokemon_color_name
|
|
|
|
@classmethod
|
|
def setup_pokemon_shape_data(self, name='pkm shp'):
|
|
|
|
pokemon_shape = PokemonShape.objects.create (
|
|
name = name,
|
|
)
|
|
pokemon_shape.save()
|
|
|
|
return pokemon_shape
|
|
|
|
@classmethod
|
|
def setup_pokemon_shape_name_data(self, pokemon_shape, name='pkm shp nm', awesome_name='pkm shp awsm nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
pokemon_shape_name = PokemonShapeName.objects.create (
|
|
pokemon_shape = pokemon_shape,
|
|
language = language,
|
|
name = name,
|
|
awesome_name = awesome_name
|
|
)
|
|
pokemon_shape_name.save()
|
|
|
|
return pokemon_shape_name
|
|
|
|
@classmethod
|
|
def setup_pokemon_species_form_description_data(self, pokemon_species=None, description='pkm spcs frm dscr'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+description)
|
|
|
|
pokemon_species_form_description = PokemonSpeciesDescription.objects.create (
|
|
pokemon_species = pokemon_species,
|
|
language = language,
|
|
description = description
|
|
)
|
|
pokemon_species_form_description.save()
|
|
|
|
return pokemon_species_form_description
|
|
|
|
@classmethod
|
|
def setup_pokemon_species_flavor_text_data(self, pokemon_species, flavor_text='pkmn spcs flvr txt'):
|
|
|
|
version = self.setup_version_data(
|
|
name='ver for '+flavor_text)
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+flavor_text)
|
|
|
|
pokemon_species_flavor_text = PokemonSpeciesFlavorText.objects.create (
|
|
pokemon_species = pokemon_species,
|
|
version = version,
|
|
language = language,
|
|
flavor_text = flavor_text
|
|
)
|
|
pokemon_species_flavor_text.save()
|
|
|
|
return pokemon_species_flavor_text
|
|
|
|
@classmethod
|
|
def setup_pokemon_species_data(self, generation=None, evolves_from_species=None, evolution_chain=None, growth_rate=None, pokemon_color=None, pokemon_habitat=None, pokemon_shape=None, name='pkm spcs', gender_rate=50, capture_rate=20, base_happiness=20, is_baby=False, hatch_counter=10, has_gender_differences=True, forms_switchable=False, order=1):
|
|
|
|
generation = generation or self.setup_generation_data(
|
|
name='gen for '+name)
|
|
|
|
growth_rate = growth_rate or self.setup_growth_rate_data(
|
|
name='grth rt for '+name)
|
|
|
|
pokemon_shape = pokemon_shape or self.setup_pokemon_shape_data(
|
|
name='pkmn shp for '+name)
|
|
|
|
pokemon_color = pokemon_color or self.setup_pokemon_color_data(
|
|
name='pkmn clr for '+name)
|
|
|
|
pokemon_habitat = pokemon_habitat or self.setup_pokemon_habitat_data(
|
|
name='pkm hbtt for '+name)
|
|
|
|
pokemon_species = PokemonSpecies.objects.create (
|
|
name = name,
|
|
generation = generation,
|
|
evolves_from_species = evolves_from_species,
|
|
evolution_chain = evolution_chain,
|
|
pokemon_color = pokemon_color,
|
|
pokemon_shape = pokemon_shape,
|
|
pokemon_habitat = pokemon_habitat,
|
|
gender_rate = gender_rate,
|
|
capture_rate = capture_rate,
|
|
base_happiness = base_happiness,
|
|
is_baby = is_baby,
|
|
hatch_counter = hatch_counter,
|
|
has_gender_differences = has_gender_differences,
|
|
growth_rate = growth_rate,
|
|
forms_switchable = forms_switchable,
|
|
order = forms_switchable
|
|
)
|
|
pokemon_species.save()
|
|
|
|
return pokemon_species
|
|
|
|
@classmethod
|
|
def setup_pokemon_species_name_data(self, pokemon_species, name='pkmn spcs nm', genus='pkmn spcs gns'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
pokemon_species_name = PokemonSpeciesName.objects.create (
|
|
pokemon_species = pokemon_species,
|
|
language = language,
|
|
name = name,
|
|
genus = genus
|
|
)
|
|
pokemon_species_name.save()
|
|
|
|
return pokemon_species_name
|
|
|
|
@classmethod
|
|
def setup_pokemon_dex_entry_data(self, pokemon_species, pokedex, entry_number=100):
|
|
|
|
dex_number = PokemonDexNumber (
|
|
pokemon_species = pokemon_species,
|
|
pokedex = pokedex,
|
|
pokedex_number = entry_number
|
|
)
|
|
dex_number.save()
|
|
|
|
return dex_number
|
|
|
|
@classmethod
|
|
def setup_pokemon_egg_group_data(self, pokemon_species, egg_group):
|
|
|
|
pokemon_egg_group = PokemonEggGroup (
|
|
pokemon_species = pokemon_species,
|
|
egg_group = egg_group
|
|
)
|
|
pokemon_egg_group.save()
|
|
|
|
return pokemon_egg_group
|
|
|
|
@classmethod
|
|
def setup_pokemon_data(self, pokemon_species=None, name='pkmn', height=100, weight=100, base_experience=0, order=1, is_default=False):
|
|
|
|
pokemon_species = pokemon_species or self.setup_pokemon_species_data(
|
|
name='pkmn spcs for '+name)
|
|
|
|
pokemon = Pokemon.objects.create (
|
|
name = name,
|
|
pokemon_species = pokemon_species,
|
|
height = height,
|
|
weight = weight,
|
|
base_experience = base_experience,
|
|
order = order,
|
|
is_default = is_default
|
|
)
|
|
pokemon.save()
|
|
|
|
return pokemon
|
|
|
|
@classmethod
|
|
def setup_pokemon_game_index_data(self, pokemon, game_index=0):
|
|
|
|
version = self.setup_version_data(
|
|
name='ver for pkmn gm indx')
|
|
|
|
pokemon_game_index = PokemonGameIndex.objects.create (
|
|
pokemon = pokemon,
|
|
game_index = game_index,
|
|
version = version
|
|
)
|
|
pokemon_game_index.save()
|
|
|
|
return pokemon_game_index
|
|
|
|
@classmethod
|
|
def setup_pokemon_form_sprites_data(
|
|
self, pokemon_form,
|
|
front_default=True, front_shiny=False,
|
|
back_default=False, back_shiny=False):
|
|
|
|
sprite_path = '/media/sprites/pokemon/%s.png'
|
|
|
|
sprites = {
|
|
'front_default' : sprite_path % pokemon_form.id if front_default else None,
|
|
'front_shiny' : sprite_path % pokemon_form.id if front_shiny else None,
|
|
'back_default' : sprite_path % pokemon_form.id if back_default else None,
|
|
'back_shiny' : sprite_path % pokemon_form.id if back_shiny else None,
|
|
}
|
|
|
|
pokemon_form_sprites = PokemonFormSprites.objects.create (
|
|
pokemon_form = pokemon_form,
|
|
sprites = json.dumps(sprites)
|
|
)
|
|
pokemon_form_sprites.save()
|
|
|
|
return pokemon_form_sprites
|
|
|
|
|
|
@classmethod
|
|
def setup_pokemon_form_data(self, pokemon, name='pkmn nrml frm', form_name='nrml', order=1, is_default=True, is_battle_only=True, form_order=1, is_mega=False):
|
|
|
|
version_group = self.setup_version_group_data(
|
|
name='ver grp for '+name)
|
|
|
|
pokemon_form = PokemonForm (
|
|
name = name,
|
|
form_name = form_name,
|
|
pokemon = pokemon,
|
|
version_group = version_group,
|
|
is_default = is_default,
|
|
is_battle_only = is_battle_only,
|
|
is_mega = is_mega,
|
|
form_order = form_order,
|
|
order = order
|
|
)
|
|
pokemon_form.save()
|
|
|
|
return pokemon_form
|
|
|
|
@classmethod
|
|
def setup_pokemon_ability_data(self, pokemon, ability=None, is_hidden=False, slot=1):
|
|
|
|
ability = ability or self.setup_ability_data(name='ablty for pkmn')
|
|
|
|
pokemon_ability = PokemonAbility (
|
|
pokemon = pokemon,
|
|
ability = ability,
|
|
is_hidden = is_hidden,
|
|
slot = slot
|
|
)
|
|
pokemon_ability.save()
|
|
|
|
return pokemon_ability
|
|
|
|
@classmethod
|
|
def setup_pokemon_stat_data(self, pokemon, base_stat=10, effort=10):
|
|
|
|
stat = self.setup_stat_data(
|
|
name='stt for pkmn')
|
|
|
|
pokemon_stat = PokemonStat (
|
|
pokemon = pokemon,
|
|
stat = stat,
|
|
base_stat = base_stat,
|
|
effort = effort
|
|
)
|
|
pokemon_stat.save()
|
|
|
|
return pokemon_stat
|
|
|
|
@classmethod
|
|
def setup_pokemon_type_data(self, pokemon, type=None, slot=1):
|
|
|
|
type = type or self.setup_type_data(
|
|
name='tp for pkmn')
|
|
|
|
pokemon_type = PokemonType (
|
|
pokemon = pokemon,
|
|
type = type,
|
|
slot = slot
|
|
)
|
|
pokemon_type.save()
|
|
|
|
return pokemon_type
|
|
|
|
@classmethod
|
|
def setup_pokemon_item_data(self, pokemon=None, item=None, version=None, rarity=50):
|
|
|
|
item = item or self.setup_item_data(
|
|
name='itm for pkmn')
|
|
|
|
pokemon = pokemon or self.setup_pokemon_data(
|
|
name='pkmn for pkmn')
|
|
|
|
version = version or self.setup_version_data(
|
|
name='ver grp for pkmn itm')
|
|
|
|
pokemon_item = PokemonItem (
|
|
pokemon = pokemon,
|
|
version = version,
|
|
item = item,
|
|
rarity = rarity
|
|
)
|
|
pokemon_item.save()
|
|
|
|
return pokemon_item
|
|
|
|
@classmethod
|
|
def setup_pokemon_move_data(self, pokemon, move, version_group, level=0, order=1):
|
|
|
|
move_learn_method = self.setup_move_learn_method_data(
|
|
name='mv lrn mthd for pkmn')
|
|
|
|
pokemon_move = PokemonMove.objects.create (
|
|
pokemon = pokemon,
|
|
version_group = version_group,
|
|
move = move,
|
|
move_learn_method = move_learn_method,
|
|
level = level,
|
|
order = order
|
|
)
|
|
pokemon_move.save()
|
|
|
|
return pokemon_move
|
|
|
|
@classmethod
|
|
def setup_pokemon_sprites_data(
|
|
self, pokemon, front_default=True,
|
|
front_female=False, front_shiny=False,
|
|
front_shiny_female=False, back_default=False,
|
|
back_female=False, back_shiny=False,
|
|
back_shiny_female=False):
|
|
|
|
sprite_path = '/media/sprites/pokemon/%s.png'
|
|
|
|
sprites = {
|
|
'front_default' : sprite_path % pokemon.id if front_default else None,
|
|
'front_female' : sprite_path % pokemon.id if front_female else None,
|
|
'front_shiny' : sprite_path % pokemon.id if front_shiny else None,
|
|
'front_shiny_female' : sprite_path % pokemon.id if front_shiny_female else None,
|
|
'back_default' : sprite_path % pokemon.id if back_default else None,
|
|
'back_female' : sprite_path % pokemon.id if back_female else None,
|
|
'back_shiny' : sprite_path % pokemon.id if back_shiny else None,
|
|
'back_shiny_female' : sprite_path % pokemon.id if back_shiny_female else None,
|
|
}
|
|
|
|
pokemon_sprites = PokemonSprites.objects.create (
|
|
pokemon = pokemon,
|
|
sprites = json.dumps(sprites)
|
|
)
|
|
pokemon_sprites.save()
|
|
|
|
return pokemon_sprites
|
|
|
|
|
|
"""
|
|
Evolution Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_evolution_trigger_data(self, name='evltn trgr'):
|
|
|
|
evolution_trigger = EvolutionTrigger.objects.create (
|
|
name = name,
|
|
)
|
|
evolution_trigger.save()
|
|
|
|
return evolution_trigger
|
|
|
|
@classmethod
|
|
def setup_evolution_trigger_name_data(self, evolution_trigger, name='evltn trgr nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
evolution_trigger_name = EvolutionTriggerName.objects.create (
|
|
evolution_trigger = evolution_trigger,
|
|
language = language,
|
|
name = name
|
|
)
|
|
evolution_trigger_name.save()
|
|
|
|
return evolution_trigger_name
|
|
|
|
@classmethod
|
|
def setup_evolution_chain_data(self, baby_trigger_item=None):
|
|
|
|
evolution_chain = EvolutionChain.objects.create (
|
|
baby_trigger_item=baby_trigger_item,
|
|
)
|
|
evolution_chain.save()
|
|
|
|
return evolution_chain
|
|
|
|
@classmethod
|
|
def setup_pokemon_evolution_data(self, evolved_species=None, evolution_trigger=None, party_species=None, trade_species=None, evolution_item=None, party_type=None, min_level=0, gender=None, location=None, held_item=None, time_of_day='', known_move=None, known_move_type=None, min_happiness=0, min_beauty=0, min_affection=0, relative_physical_stats=0, needs_overworld_rain=False, turn_upside_down=False):
|
|
|
|
evolved_species = evolved_species or self.setup_pokemon_species_data(
|
|
name='pkmn spcs for pkmn evltn')
|
|
|
|
evolution_trigger = evolution_trigger or self.setup_evolution_trigger_data(
|
|
name='evltn trgr for pkmn evltn')
|
|
|
|
pokemon_evolution = PokemonEvolution.objects.create (
|
|
evolved_species = evolved_species,
|
|
evolution_trigger = evolution_trigger,
|
|
evolution_item = evolution_item,
|
|
min_level = min_level,
|
|
gender = gender,
|
|
location = location,
|
|
held_item = held_item,
|
|
time_of_day = time_of_day,
|
|
known_move = known_move,
|
|
known_move_type = known_move_type,
|
|
min_happiness = min_happiness,
|
|
min_beauty = min_beauty,
|
|
min_affection = min_affection,
|
|
relative_physical_stats = relative_physical_stats,
|
|
party_species = party_species,
|
|
party_type = party_type,
|
|
trade_species = trade_species,
|
|
needs_overworld_rain = needs_overworld_rain,
|
|
turn_upside_down = turn_upside_down
|
|
)
|
|
pokemon_evolution.save()
|
|
|
|
return pokemon_evolution
|
|
|
|
|
|
"""
|
|
Encounter Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_encounter_method_data(self, name='encntr mthd', order=0):
|
|
|
|
encounter_method = EncounterMethod.objects.create (
|
|
name = name,
|
|
order = order
|
|
)
|
|
encounter_method.save()
|
|
|
|
return encounter_method
|
|
|
|
classmethod
|
|
def setup_encounter_method_name_data(self, encounter_method, name='encntr mthd nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
encounter_method_name = EncounterMethodName.objects.create (
|
|
encounter_method = encounter_method,
|
|
language = language,
|
|
name = name
|
|
)
|
|
encounter_method_name.save()
|
|
|
|
return encounter_method_name
|
|
|
|
@classmethod
|
|
def setup_encounter_condition_data(self, name='encntr cndtn', order=0):
|
|
|
|
encounter_condition = EncounterCondition.objects.create (
|
|
name = name
|
|
)
|
|
encounter_condition.save()
|
|
|
|
return encounter_condition
|
|
|
|
@classmethod
|
|
def setup_encounter_condition_name_data(self, encounter_condition, name='encntr cndtn nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
encounter_condition_name = EncounterConditionName.objects.create (
|
|
encounter_condition = encounter_condition,
|
|
language = language,
|
|
name = name
|
|
)
|
|
encounter_condition_name.save()
|
|
|
|
return encounter_condition_name
|
|
|
|
@classmethod
|
|
def setup_encounter_condition_value_data(self, encounter_condition, name='encntr cndtn vlu', is_default=False):
|
|
|
|
encounter_condition_value = EncounterConditionValue.objects.create (
|
|
encounter_condition = encounter_condition,
|
|
name = name,
|
|
is_default = is_default
|
|
)
|
|
encounter_condition_value.save()
|
|
|
|
return encounter_condition_value
|
|
|
|
@classmethod
|
|
def setup_encounter_condition_value_name_data(self, encounter_condition_value, name='encntr cndtn vlu nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
encounter_condition_value_name = EncounterConditionValueName.objects.create (
|
|
encounter_condition_value = encounter_condition_value,
|
|
language = language,
|
|
name = name
|
|
)
|
|
encounter_condition_value_name.save()
|
|
|
|
return encounter_condition_value_name
|
|
|
|
@classmethod
|
|
def setup_encounter_condition_value_map_data(self, encounter, encounter_condition_value):
|
|
|
|
encounter_condition_value_map = EncounterConditionValue.objects.create (
|
|
encounter = encounter,
|
|
encounter_condition_value = encounter_condition_value
|
|
)
|
|
encounter_condition_value_map.save()
|
|
|
|
return encounter_condition_value_map
|
|
|
|
@classmethod
|
|
def setup_encounter_slot_data(self, encounter_method=None, slot=0, rarity=0):
|
|
|
|
encounter_method = encounter_method or self.setup_encounter_method_data(
|
|
name='encntr mthd for encntr slt')
|
|
|
|
version_group = self.setup_version_group_data(
|
|
name='ver grp for encntr slt')
|
|
|
|
encounter_slot = EncounterSlot.objects.create (
|
|
encounter_method = encounter_method,
|
|
version_group = version_group,
|
|
slot = slot,
|
|
rarity = rarity
|
|
)
|
|
encounter_slot.save()
|
|
|
|
return encounter_slot
|
|
|
|
@classmethod
|
|
def setup_location_area_encounter_rate_data(self, location_area, encounter_method, rate=0):
|
|
|
|
version = self.setup_version_data(
|
|
name='ver for lctn area')
|
|
|
|
location_area_encounter_rate = LocationAreaEncounterRate.objects.create (
|
|
encounter_method = encounter_method,
|
|
version = version,
|
|
location_area = location_area,
|
|
rate = rate
|
|
)
|
|
location_area_encounter_rate.save()
|
|
|
|
return location_area_encounter_rate
|
|
|
|
@classmethod
|
|
def setup_encounter_data(self, location_area=None, encounter_slot=None, pokemon=None, version=None, min_level=10, max_level=15):
|
|
|
|
location_area = location_area or self.setup_location_area_data(
|
|
name='ver for encntr')
|
|
|
|
encounter_slot = encounter_slot or self.setup_encounter_slot_data()
|
|
|
|
pokemon = pokemon or self.setup_pokemon_data(
|
|
name='pkmn for encntr')
|
|
|
|
version = version or self.setup_version_data(
|
|
name='ver for encntr')
|
|
|
|
encounter = Encounter.objects.create (
|
|
version = version,
|
|
location_area = location_area,
|
|
encounter_slot = encounter_slot,
|
|
pokemon = pokemon,
|
|
min_level = min_level,
|
|
max_level = max_level
|
|
)
|
|
encounter.save()
|
|
|
|
return encounter
|
|
|
|
|
|
"""
|
|
Pal Park Data
|
|
"""
|
|
|
|
@classmethod
|
|
def setup_pal_park_area_data(self, name='pl prk area'):
|
|
|
|
pal_park_area = PalParkArea.objects.create (
|
|
name = name
|
|
)
|
|
pal_park_area.save()
|
|
|
|
return pal_park_area
|
|
|
|
classmethod
|
|
def setup_pal_park_area_name_data(self, pal_park_area, name='pl prk area nm'):
|
|
|
|
language = self.setup_language_data(
|
|
name='lang for '+name)
|
|
|
|
pal_park_area_name = PalParkAreaName.objects.create (
|
|
pal_park_area = pal_park_area,
|
|
language = language,
|
|
name = name
|
|
)
|
|
pal_park_area_name.save()
|
|
|
|
return pal_park_area_name
|
|
|
|
@classmethod
|
|
def setup_pal_park_data(self, pokemon_species=None, pal_park_area=None, base_score=10, rate=10):
|
|
|
|
pal_park_area = pal_park_area or self.setup_pal_park_area_data(
|
|
name='pl prk area')
|
|
|
|
pal_park = PalPark.objects.create (
|
|
base_score = base_score,
|
|
pokemon_species = pokemon_species,
|
|
pal_park_area = pal_park_area,
|
|
rate = rate
|
|
)
|
|
pal_park.save()
|
|
|
|
return pal_park
|
|
|
|
|
|
|
|
"""
|
|
Tests
|
|
"""
|
|
|
|
class APITests(APIData, APITestCase):
|
|
|
|
"""
|
|
Gender Tests
|
|
"""
|
|
|
|
def test_gender_api(self):
|
|
|
|
gender = self.setup_gender_data(name='female')
|
|
pokemon_species = self.setup_pokemon_species_data(name='pkmn spcs for gndr', gender_rate=8)
|
|
evolved_species = self.setup_pokemon_species_data(name='evlvd pkmn spcs for gndr')
|
|
self.setup_pokemon_evolution_data(evolved_species=evolved_species, gender=gender)
|
|
|
|
response = self.client.get('{}/gender/{}/'.format(api_v2, gender.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], gender.pk)
|
|
self.assertEqual(response.data['name'], gender.name)
|
|
# species params
|
|
self.assertEqual(response.data['pokemon_species_details'][0]['rate'], pokemon_species.gender_rate)
|
|
self.assertEqual(response.data['pokemon_species_details'][0]['pokemon_species']['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_species_details'][0]['pokemon_species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
self.assertEqual(response.data['required_for_evolution'][0]['name'], evolved_species.name)
|
|
self.assertEqual(response.data['required_for_evolution'][0]['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, evolved_species.pk))
|
|
|
|
|
|
"""
|
|
Language Tests
|
|
"""
|
|
|
|
def test_language_api(self):
|
|
|
|
language = self.setup_language_data(name='base lang')
|
|
language_name = self.setup_language_name_data(language, name='base lang name')
|
|
|
|
response = self.client.get('{}/language/{}/'.format(api_v2, language.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], language.pk)
|
|
self.assertEqual(response.data['iso639'], language.iso639)
|
|
self.assertEqual(response.data['iso3166'], language.iso3166)
|
|
self.assertEqual(response.data['name'], language.name)
|
|
self.assertEqual(response.data['official'], language.official)
|
|
#name params
|
|
self.assertEqual(response.data['names'][0]['name'], language_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], language_name.local_language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, language_name.local_language.pk))
|
|
|
|
|
|
"""
|
|
Region Tests
|
|
"""
|
|
|
|
def test_region_api(self):
|
|
|
|
region = self.setup_region_data(name='base reg')
|
|
region_name = self.setup_region_name_data(region, name='base reg name')
|
|
location = self.setup_location_data(region=region, name="lctn for base rgn")
|
|
generation = self.setup_generation_data(region=region, name="gnrtn for base rgn")
|
|
pokedex = self.setup_pokedex_data(region=region, name="pkdx for base rgn")
|
|
version_group = self.setup_version_group_data(name="ver grp for base rgn")
|
|
version_group_region = self.setup_version_group_region_data(region=region, version_group=version_group)
|
|
|
|
response = self.client.get('{}/region/{}/'.format(api_v2, region.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], region.pk)
|
|
self.assertEqual(response.data['name'], region.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], region_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], region_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, region_name.language.pk))
|
|
# location params
|
|
self.assertEqual(response.data['locations'][0]['name'], location.name)
|
|
self.assertEqual(response.data['locations'][0]['url'], '{}{}/location/{}/'.format(test_host, api_v2, location.pk))
|
|
# generation params
|
|
self.assertEqual(response.data['main_generation']['name'], generation.name)
|
|
self.assertEqual(response.data['main_generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, generation.pk))
|
|
# pokedex params
|
|
self.assertEqual(response.data['pokedexes'][0]['name'], pokedex.name)
|
|
self.assertEqual(response.data['pokedexes'][0]['url'], '{}{}/pokedex/{}/'.format(test_host, api_v2, pokedex.pk))
|
|
# version group params
|
|
self.assertEqual(response.data['version_groups'][0]['name'], version_group.name)
|
|
self.assertEqual(response.data['version_groups'][0]['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, version_group.pk))
|
|
|
|
|
|
|
|
"""
|
|
Generation Tests
|
|
"""
|
|
|
|
def test_generation_api(self):
|
|
|
|
generation = self.setup_generation_data(name='base gen')
|
|
generation_name = self.setup_generation_name_data(generation, name='base reg name')
|
|
ability = self.setup_ability_data(name='ablty for base gen', generation=generation)
|
|
move = self.setup_move_data(name='mv for base gen', generation=generation)
|
|
pokemon_species = self.setup_pokemon_species_data(name='pkmn spcs for base gen', generation=generation)
|
|
type = self.setup_type_data(name='tp for base gen', generation=generation)
|
|
version_group = self.setup_version_group_data(name='ver grp for base gen', generation=generation)
|
|
|
|
response = self.client.get('{}/generation/{}/'.format(api_v2, generation.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], generation.pk)
|
|
self.assertEqual(response.data['name'], generation.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], generation_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], generation_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, generation_name.language.pk))
|
|
# region params
|
|
self.assertEqual(response.data['main_region']['name'], generation.region.name)
|
|
self.assertEqual(response.data['main_region']['url'], '{}{}/region/{}/'.format(test_host, api_v2, generation.region.pk))
|
|
# ability params
|
|
self.assertEqual(response.data['abilities'][0]['name'], ability.name)
|
|
self.assertEqual(response.data['abilities'][0]['url'], '{}{}/ability/{}/'.format(test_host, api_v2, ability.pk))
|
|
# move params
|
|
self.assertEqual(response.data['moves'][0]['name'], move.name)
|
|
self.assertEqual(response.data['moves'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, move.pk))
|
|
# type params
|
|
self.assertEqual(response.data['types'][0]['name'], type.name)
|
|
self.assertEqual(response.data['types'][0]['url'], '{}{}/type/{}/'.format(test_host, api_v2, type.pk))
|
|
# species params
|
|
self.assertEqual(response.data['pokemon_species'][0]['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_species'][0]['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
# version group params
|
|
self.assertEqual(response.data['version_groups'][0]['name'], version_group.name)
|
|
self.assertEqual(response.data['version_groups'][0]['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, version_group.pk))
|
|
|
|
|
|
|
|
"""
|
|
Version Tests
|
|
"""
|
|
|
|
def test_version_api(self):
|
|
|
|
version_group = self.setup_version_group_data(name='ver grp for ver')
|
|
version = self.setup_version_data(name='base ver', version_group=version_group)
|
|
version_name = self.setup_version_name_data(version, name='base ver name')
|
|
|
|
response = self.client.get('{}/version/{}/'.format(api_v2, version.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], version.pk)
|
|
self.assertEqual(response.data['name'], version.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], version_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], version_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, version_name.language.pk))
|
|
# version group params
|
|
self.assertEqual(response.data['version_group']['name'], version.version_group.name)
|
|
self.assertEqual(response.data['version_group']['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, version.version_group.pk))
|
|
|
|
def test_version_group_api(self):
|
|
|
|
version_group = self.setup_version_group_data(name='base ver grp')
|
|
move_learn_method = self.setup_move_learn_method_data(name='mv lrn mthd for ')
|
|
version_group_move_learn_method = self.setup_version_group_move_learn_method_data(version_group=version_group, move_learn_method=move_learn_method)
|
|
region = self.setup_region_data(name='rgn for ver grp')
|
|
version_group_region = self.setup_version_group_region_data(version_group=version_group, region=region)
|
|
version = self.setup_version_data(name='ver for base ver grp', version_group=version_group)
|
|
pokedex = self.setup_pokedex_data(name='pkdx for base ver group')
|
|
pokedex_version_group = self.setup_pokedex_version_group_data(pokedex=pokedex, version_group=version_group)
|
|
|
|
response = self.client.get('{}/version-group/{}/'.format(api_v2, version_group.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], version_group.pk)
|
|
self.assertEqual(response.data['name'], version_group.name)
|
|
self.assertEqual(response.data['order'], version_group.order)
|
|
# version params
|
|
self.assertEqual(response.data['versions'][0]['name'], version.name)
|
|
self.assertEqual(response.data['versions'][0]['url'], '{}{}/version/{}/'.format(test_host, api_v2, version.pk))
|
|
# generation params
|
|
self.assertEqual(response.data['generation']['name'], version_group.generation.name)
|
|
self.assertEqual(response.data['generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, version_group.generation.pk))
|
|
# region params
|
|
self.assertEqual(response.data['regions'][0]['name'], region.name)
|
|
self.assertEqual(response.data['regions'][0]['url'], '{}{}/region/{}/'.format(test_host, api_v2, region.pk))
|
|
# move learn method params
|
|
self.assertEqual(response.data['move_learn_methods'][0]['name'], move_learn_method.name)
|
|
self.assertEqual(response.data['move_learn_methods'][0]['url'], '{}{}/move-learn-method/{}/'.format(test_host, api_v2, move_learn_method.pk))
|
|
# pokedex group
|
|
self.assertEqual(response.data['pokedexes'][0]['name'], pokedex.name)
|
|
self.assertEqual(response.data['pokedexes'][0]['url'], '{}{}/pokedex/{}/'.format(test_host, api_v2, pokedex.pk))
|
|
|
|
|
|
|
|
"""
|
|
Egg Group Tests
|
|
"""
|
|
|
|
def test_egg_group_api(self):
|
|
|
|
egg_group = self.setup_egg_group_data(name='base egg grp')
|
|
egg_group_name = self.setup_egg_group_name_data(egg_group, name='base egg grp name')
|
|
pokemon_species = self.setup_pokemon_species_data()
|
|
self.setup_pokemon_egg_group_data(pokemon_species=pokemon_species, egg_group=egg_group)
|
|
|
|
response = self.client.get('{}/egg-group/{}/'.format(api_v2, egg_group.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], egg_group.pk)
|
|
self.assertEqual(response.data['name'], egg_group.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], egg_group_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], egg_group_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, egg_group_name.language.pk))
|
|
# species params
|
|
self.assertEqual(response.data['pokemon_species'][0]['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_species'][0]['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
|
|
|
|
|
|
"""
|
|
Ability Tests
|
|
"""
|
|
|
|
def test_ability_api(self):
|
|
|
|
ability = self.setup_ability_data(name='base ablty')
|
|
ability_name = self.setup_ability_name_data(ability, name='base ablty name')
|
|
ability_effect_text = self.setup_ability_effect_text_data(ability, effect='base ablty efct')
|
|
ability_flavor_text = self.setup_ability_flavor_text_data(ability, flavor_text='base flvr txt')
|
|
ability_change = self.setup_ability_change_data(ability)
|
|
ability_change_effect_text = self.setup_ability_change_effect_text_data(ability_change, effect='base ablty chng efct')
|
|
pokemon = self.setup_pokemon_data(name='pkmn for ablty')
|
|
pokemon_ability = self.setup_pokemon_ability_data(ability=ability, pokemon=pokemon)
|
|
|
|
response = self.client.get('{}/ability/{}/'.format(api_v2, ability.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], ability.pk)
|
|
self.assertEqual(response.data['name'], ability.name)
|
|
self.assertEqual(response.data['is_main_series'], ability.is_main_series)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], ability_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], ability_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, ability_name.language.pk))
|
|
# description params
|
|
self.assertEqual(response.data['effect_entries'][0]['effect'], ability_effect_text.effect)
|
|
self.assertEqual(response.data['effect_entries'][0]['short_effect'], ability_effect_text.short_effect)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['name'], ability_effect_text.language.name)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, ability_effect_text.language.pk))
|
|
# flavor text params
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['flavor_text'], ability_flavor_text.flavor_text)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['version_group']['name'], ability_flavor_text.version_group.name)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['name'], ability_flavor_text.language.name)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, ability_flavor_text.language.pk))
|
|
# generation params
|
|
self.assertEqual(response.data['generation']['name'], ability.generation.name)
|
|
self.assertEqual(response.data['generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, ability.generation.pk))
|
|
# change params
|
|
self.assertEqual(response.data['effect_changes'][0]['version_group']['name'], ability_change.version_group.name)
|
|
self.assertEqual(response.data['effect_changes'][0]['version_group']['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, ability_change.version_group.pk))
|
|
self.assertEqual(response.data['effect_changes'][0]['effect_entries'][0]['effect'], ability_change_effect_text.effect)
|
|
self.assertEqual(response.data['effect_changes'][0]['effect_entries'][0]['language']['name'], ability_change_effect_text.language.name)
|
|
self.assertEqual(response.data['effect_changes'][0]['effect_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, ability_change_effect_text.language.pk))
|
|
# pokemon params
|
|
self.assertEqual(response.data['pokemon'][0]['is_hidden'], pokemon_ability.is_hidden)
|
|
self.assertEqual(response.data['pokemon'][0]['slot'], pokemon_ability.slot)
|
|
self.assertEqual(response.data['pokemon'][0]['pokemon']['name'], pokemon.name)
|
|
self.assertEqual(response.data['pokemon'][0]['pokemon']['url'], '{}{}/pokemon/{}/'.format(test_host, api_v2, pokemon.pk))
|
|
|
|
|
|
|
|
"""
|
|
Item Tests
|
|
"""
|
|
|
|
def test_item_attribute_api(self):
|
|
|
|
# item attribute data
|
|
item_attribute = self.setup_item_attribute_data(name='base itm attr')
|
|
item_attribute_name = self.setup_item_attribute_name_data(item_attribute, name='base itm attr nm')
|
|
item_attribute_description = self.setup_item_attribute_description_data(item_attribute, description='base itm attr desc')
|
|
item = self.setup_item_data(name='itm fr base itm attr')
|
|
self.setup_item_attribute_map_data(item_attribute=item_attribute, item=item)
|
|
|
|
response = self.client.get('{}/item-attribute/{}/'.format(api_v2, item_attribute.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], item_attribute.pk)
|
|
self.assertEqual(response.data['name'], item_attribute.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], item_attribute_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], item_attribute_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, item_attribute_name.language.pk))
|
|
# description params
|
|
self.assertEqual(response.data['descriptions'][0]['description'], item_attribute_description.description)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['name'], item_attribute_description.language.name)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, item_attribute_description.language.pk))
|
|
# item params
|
|
self.assertEqual(response.data['items'][0]['name'], item.name)
|
|
self.assertEqual(response.data['items'][0]['url'], '{}{}/item/{}/'.format(test_host, api_v2, item.pk))
|
|
|
|
|
|
def test_item_category_api(self):
|
|
|
|
# item category data
|
|
item_category = self.setup_item_category_data(name='base itm ctgry')
|
|
item_category_name = self.setup_item_category_name_data(item_category, name='base itm ctgry nm')
|
|
item = self.setup_item_data(item_category=item_category, name='itm fr base itm ctgry')
|
|
|
|
response = self.client.get('{}/item-category/{}/'.format(api_v2, item_category.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], item_category.pk)
|
|
self.assertEqual(response.data['name'], item_category.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], item_category_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], item_category_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, item_category_name.language.pk))
|
|
# pocket params
|
|
self.assertEqual(response.data['pocket']['name'], item_category.item_pocket.name)
|
|
self.assertEqual(response.data['pocket']['url'], '{}{}/item-pocket/{}/'.format(test_host, api_v2, item_category.item_pocket.pk))
|
|
# item params
|
|
self.assertEqual(response.data['items'][0]['name'], item.name)
|
|
self.assertEqual(response.data['items'][0]['url'], '{}{}/item/{}/'.format(test_host, api_v2, item.pk))
|
|
|
|
|
|
def test_item_fling_effect_api(self):
|
|
|
|
# item category data
|
|
item_fling_effect = self.setup_item_fling_effect_data(name='base itm flng efct')
|
|
item_fling_effect_effect_text = self.setup_item_fling_effect_effect_text_data(item_fling_effect, effect='base itm flng efct nm')
|
|
item = self.setup_item_data(item_fling_effect=item_fling_effect, name='itm fr base itm attr')
|
|
|
|
response = self.client.get('{}/item-fling-effect/{}/'.format(api_v2, item_fling_effect.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], item_fling_effect.pk)
|
|
self.assertEqual(response.data['name'], item_fling_effect.name)
|
|
# description params
|
|
self.assertEqual(response.data['effect_entries'][0]['effect'], item_fling_effect_effect_text.effect)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['name'], item_fling_effect_effect_text.language.name)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, item_fling_effect_effect_text.language.pk))
|
|
# item params
|
|
self.assertEqual(response.data['items'][0]['name'], item.name)
|
|
self.assertEqual(response.data['items'][0]['url'], '{}{}/item/{}/'.format(test_host, api_v2, item.pk))
|
|
|
|
|
|
def test_item_pocket_api(self):
|
|
|
|
# item pocket data
|
|
item_pocket = self.setup_item_pocket_data(name='base itm pkt')
|
|
item_pocket_name = self.setup_item_pocket_name_data(item_pocket, name='base itm pkt nm')
|
|
item_category = self.setup_item_category_data(name='itm ctgry for base itm pckt', item_pocket=item_pocket)
|
|
|
|
response = self.client.get('{}/item-pocket/{}/'.format(api_v2, item_pocket.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], item_pocket.pk)
|
|
self.assertEqual(response.data['name'], item_pocket.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], item_pocket_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], item_pocket_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, item_pocket_name.language.pk))
|
|
|
|
def test_item_api(self):
|
|
|
|
item_category = self.setup_item_category_data(name='itm ctgry for base itm')
|
|
item_fling_effect = self.setup_item_fling_effect_data(name='itm flng efct for base itm')
|
|
item = self.setup_item_data(item_category, item_fling_effect, name='base itm')
|
|
item_name = self.setup_item_name_data(item, name='base itm name')
|
|
item_flavor_text = self.setup_item_flavor_text_data(item, flavor_text='base itm flvr txt')
|
|
item_effect_text = self.setup_item_effect_text_data(item, effect='base nrml efct', short_effect='base shrt efct')
|
|
item_attribute = self.setup_item_attribute_data()
|
|
item_game_index = self.setup_item_game_index_data(item, game_index=10)
|
|
item_sprites = self.setup_item_sprites_data(item)
|
|
pokemon = self.setup_pokemon_data(name='pkmn for base itm')
|
|
pokemon_item = self.setup_pokemon_item_data(pokemon=pokemon, item=item)
|
|
evolution_chain = self.setup_evolution_chain_data(baby_trigger_item=item)
|
|
|
|
# map item attribute to item
|
|
item_attribute_map = ItemAttributeMap (
|
|
item = item,
|
|
item_attribute = item_attribute
|
|
)
|
|
item_attribute_map.save()
|
|
|
|
sprites_data = json.loads(item_sprites.sprites)
|
|
|
|
response = self.client.get('{}/item/{}/'.format(api_v2, item.pk), HTTP_HOST='testserver')
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], item.pk)
|
|
self.assertEqual(response.data['name'], item.name)
|
|
self.assertEqual(response.data['cost'], item.cost)
|
|
self.assertEqual(response.data['fling_power'], item.fling_power)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], item_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], item_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, item_name.language.pk))
|
|
# flavor text params
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['text'], item_flavor_text.flavor_text)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['version_group']['name'], item_flavor_text.version_group.name)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['version_group']['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, item_flavor_text.version_group.pk))
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['name'], item_flavor_text.language.name)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, item_flavor_text.language.pk))
|
|
# effect text params
|
|
self.assertEqual(response.data['effect_entries'][0]['effect'], item_effect_text.effect)
|
|
self.assertEqual(response.data['effect_entries'][0]['short_effect'], item_effect_text.short_effect)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['name'], item_effect_text.language.name)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, item_effect_text.language.pk))
|
|
# category params
|
|
self.assertEqual(response.data['category']['name'], item_category.name)
|
|
self.assertEqual(response.data['category']['url'], '{}{}/item-category/{}/'.format(test_host, api_v2, item_category.pk))
|
|
# fling effect params
|
|
self.assertEqual(response.data['fling_effect']['name'], item_fling_effect.name)
|
|
self.assertEqual(response.data['fling_effect']['url'], '{}{}/item-fling-effect/{}/'.format(test_host, api_v2, item_fling_effect.pk))
|
|
# attribute params
|
|
self.assertEqual(response.data['attributes'][0]['name'], item_attribute.name)
|
|
self.assertEqual(response.data['attributes'][0]['url'], '{}{}/item-attribute/{}/'.format(test_host, api_v2, item_attribute.pk))
|
|
# game indices params
|
|
self.assertEqual(response.data['game_indices'][0]['game_index'], item_game_index.game_index)
|
|
self.assertEqual(response.data['game_indices'][0]['generation']['name'], item_game_index.generation.name)
|
|
self.assertEqual(response.data['game_indices'][0]['generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, item_game_index.generation.pk))
|
|
# held by params
|
|
self.assertEqual(response.data['held_by_pokemon'][0]['pokemon']['name'], pokemon.name)
|
|
self.assertEqual(response.data['held_by_pokemon'][0]['pokemon']['url'], '{}{}/pokemon/{}/'.format(test_host, api_v2, pokemon.pk))
|
|
self.assertEqual(response.data['held_by_pokemon'][0]['version_details'][0]['rarity'], pokemon_item.rarity)
|
|
self.assertEqual(response.data['held_by_pokemon'][0]['version_details'][0]['version']['name'], pokemon_item.version.name)
|
|
self.assertEqual(response.data['held_by_pokemon'][0]['version_details'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, pokemon_item.version.pk))
|
|
# baby trigger params
|
|
self.assertEqual(response.data['baby_trigger_for']['url'], '{}{}/evolution-chain/{}/'.format(test_host, api_v2, evolution_chain.pk))
|
|
# sprites
|
|
self.assertEqual(response.data['sprites']['default'], '{}{}'.format(test_host, sprites_data['default']))
|
|
|
|
|
|
"""
|
|
Berry Tests
|
|
"""
|
|
|
|
def test_berry_firmness_api(self):
|
|
|
|
berry_firmness = self.setup_berry_firmness_data(name='base bry frmns')
|
|
berry_firmness_name = self.setup_berry_firmness_name_data(berry_firmness, name='base bry frmns nm')
|
|
berry = self.setup_berry_data(berry_firmness=berry_firmness, name='bry for base frmns')
|
|
|
|
response = self.client.get('{}/berry-firmness/{}/'.format(api_v2, berry_firmness.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], berry_firmness.pk)
|
|
self.assertEqual(response.data['name'], berry_firmness.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], berry_firmness_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], berry_firmness_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, berry_firmness_name.language.pk))
|
|
# berry params
|
|
self.assertEqual(response.data['berries'][0]['name'], berry.name)
|
|
self.assertEqual(response.data['berries'][0]['url'], '{}{}/berry/{}/'.format(test_host, api_v2, berry.pk))
|
|
|
|
|
|
def test_berry_flavor_api(self):
|
|
|
|
berry_flavor = self.setup_berry_flavor_data(name='base bry flvr')
|
|
berry_flavor_name = self.setup_berry_flavor_name_data(berry_flavor, name='base bry flvr nm')
|
|
berry = self.setup_berry_data(name='bry for base bry flvr')
|
|
berry_flavor_map = self.setup_berry_flavor_map_data(berry=berry, berry_flavor=berry_flavor, potency=50)
|
|
|
|
response = self.client.get('{}/berry-flavor/{}/'.format(api_v2, berry_flavor.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], berry_flavor.pk)
|
|
self.assertEqual(response.data['name'], berry_flavor.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], berry_flavor_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], berry_flavor_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, berry_flavor_name.language.pk))
|
|
# contest type params
|
|
self.assertEqual(response.data['contest_type']['name'], berry_flavor.contest_type.name)
|
|
self.assertEqual(response.data['contest_type']['url'], '{}{}/contest-type/{}/'.format(test_host, api_v2, berry_flavor.contest_type.pk))
|
|
# berry params
|
|
self.assertEqual(response.data['berries'][0]['potency'], berry_flavor_map.potency)
|
|
self.assertEqual(response.data['berries'][0]['berry']['name'], berry.name)
|
|
self.assertEqual(response.data['berries'][0]['berry']['url'], '{}{}/berry/{}/'.format(test_host, api_v2, berry.pk))
|
|
|
|
|
|
|
|
def test_berry_api(self):
|
|
|
|
type = self.setup_type_data(name="tp fr base bry")
|
|
berry = self.setup_berry_data(name='base bry', natural_gift_type=type)
|
|
berry_flavor = self.setup_berry_flavor_data(name='bry flvr for base bry')
|
|
berry_flavor_map = self.setup_berry_flavor_map_data(berry=berry, berry_flavor=berry_flavor)
|
|
|
|
response = self.client.get('{}/berry/{}/'.format(api_v2, berry.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], berry.pk)
|
|
self.assertEqual(response.data['name'], berry.name)
|
|
self.assertEqual(response.data['growth_time'], berry.growth_time)
|
|
self.assertEqual(response.data['max_harvest'], berry.max_harvest)
|
|
self.assertEqual(response.data['natural_gift_power'], berry.natural_gift_power)
|
|
self.assertEqual(response.data['size'], berry.size)
|
|
self.assertEqual(response.data['smoothness'], berry.smoothness)
|
|
self.assertEqual(response.data['soil_dryness'], berry.soil_dryness)
|
|
# firmness params
|
|
self.assertEqual(response.data['firmness']['name'], berry.berry_firmness.name)
|
|
self.assertEqual(response.data['firmness']['url'], '{}{}/berry-firmness/{}/'.format(test_host, api_v2, berry.berry_firmness.pk))
|
|
# item params
|
|
self.assertEqual(response.data['item']['name'], berry.item.name)
|
|
self.assertEqual(response.data['item']['url'], '{}{}/item/{}/'.format(test_host, api_v2, berry.item.pk))
|
|
# flavor params
|
|
self.assertEqual(response.data['flavors'][0]['potency'], berry_flavor_map.potency)
|
|
self.assertEqual(response.data['flavors'][0]['flavor']['name'], berry_flavor.name)
|
|
self.assertEqual(response.data['flavors'][0]['flavor']['url'], '{}{}/berry-flavor/{}/'.format(test_host, api_v2, berry_flavor.pk))
|
|
# natural gift type
|
|
self.assertEqual(response.data['natural_gift_type']['name'], type.name)
|
|
self.assertEqual(response.data['natural_gift_type']['url'], '{}{}/type/{}/'.format(test_host, api_v2, type.pk))
|
|
|
|
|
|
|
|
"""
|
|
Growth Rate Tests
|
|
"""
|
|
|
|
def test_growth_rate_api(self):
|
|
|
|
# item pocket data
|
|
growth_rate = self.setup_growth_rate_data(name='base grth rt')
|
|
growth_rate_description = self.setup_growth_rate_description_data(growth_rate, description='base grth rt desc')
|
|
pokemon_species = self.setup_pokemon_species_data(name='pkmn spcs for grth rt', growth_rate=growth_rate)
|
|
|
|
# map item attribute to item
|
|
experience = Experience (
|
|
growth_rate = growth_rate,
|
|
level = 10,
|
|
experience = 3000
|
|
)
|
|
experience.save()
|
|
|
|
response = self.client.get('{}/growth-rate/{}/'.format(api_v2, growth_rate.pk))
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], growth_rate.pk)
|
|
self.assertEqual(response.data['name'], growth_rate.name)
|
|
self.assertEqual(response.data['formula'], growth_rate.formula)
|
|
# description params
|
|
self.assertEqual(response.data['descriptions'][0]['description'], growth_rate_description.description)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['name'], growth_rate_description.language.name)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, growth_rate_description.language.pk))
|
|
# experience params
|
|
self.assertEqual(response.data['levels'][0]['level'], experience.level)
|
|
self.assertEqual(response.data['levels'][0]['experience'], experience.experience)
|
|
# species params
|
|
self.assertEqual(response.data['pokemon_species'][0]['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_species'][0]['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
|
|
|
|
|
|
"""
|
|
Location Tests
|
|
"""
|
|
|
|
def test_location_api(self):
|
|
|
|
location = self.setup_location_data(name='base lctn')
|
|
location_name = self.setup_location_name_data(location, name='base lctn name')
|
|
location_game_index = self.setup_location_game_index_data(location, game_index=10)
|
|
|
|
response = self.client.get('{}/location/{}/'.format(api_v2, location.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], location.pk)
|
|
self.assertEqual(response.data['name'], location.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], location_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], location_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, location_name.language.pk))
|
|
# region params
|
|
self.assertEqual(response.data['region']['name'], location.region.name)
|
|
self.assertEqual(response.data['region']['url'], '{}{}/region/{}/'.format(test_host, api_v2, location.region.pk))
|
|
# game indices params
|
|
self.assertEqual(response.data['game_indices'][0]['game_index'], location_game_index.game_index)
|
|
self.assertEqual(response.data['game_indices'][0]['generation']['name'], location_game_index.generation.name)
|
|
self.assertEqual(response.data['game_indices'][0]['generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, location_game_index.generation.pk))
|
|
|
|
|
|
def test_location_area_api(self):
|
|
|
|
location = self.setup_location_data(name='lctn for base lctn area')
|
|
location_area = self.setup_location_area_data(location, name="base lctn area")
|
|
location_area_name = self.setup_location_area_name_data(location_area, name='base lctn area name')
|
|
|
|
encounter_method = self.setup_encounter_method_data(name='encntr mthd for lctn area')
|
|
location_area_encounter_rate = self.setup_location_area_encounter_rate_data(location_area, encounter_method, rate=20)
|
|
|
|
pokemon_species1 = self.setup_pokemon_species_data(name='spcs for pkmn1')
|
|
pokemon1 = self.setup_pokemon_data(name='pkmn1 for base encntr', pokemon_species=pokemon_species1)
|
|
encounter_slot1 = self.setup_encounter_slot_data(encounter_method, slot=1, rarity=30)
|
|
encounter1 = self.setup_encounter_data(pokemon=pokemon1, location_area=location_area, encounter_slot=encounter_slot1, min_level=30, max_level=35)
|
|
|
|
pokemon_species2 = self.setup_pokemon_species_data(name='spcs for pkmn2')
|
|
pokemon2 = self.setup_pokemon_data(name='pkmn2 for base encntr', pokemon_species=pokemon_species2)
|
|
encounter_slot2 = self.setup_encounter_slot_data(encounter_method, slot=2, rarity=40)
|
|
encounter2 = self.setup_encounter_data(pokemon=pokemon2, location_area=location_area, encounter_slot=encounter_slot2, min_level=32, max_level=36)
|
|
|
|
response = self.client.get('{}/location-area/{}/'.format(api_v2, location_area.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], location_area.pk)
|
|
self.assertEqual(response.data['name'], location_area.name)
|
|
self.assertEqual(response.data['game_index'], location_area.game_index)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], location_area_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], location_area_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, location_area_name.language.pk))
|
|
# location params
|
|
self.assertEqual(response.data['location']['name'], location.name)
|
|
self.assertEqual(response.data['location']['url'], '{}{}/location/{}/'.format(test_host, api_v2, location.pk))
|
|
# encounter method params
|
|
self.assertEqual(response.data['encounter_method_rates'][0]['encounter_method']['name'], encounter_method.name)
|
|
self.assertEqual(response.data['encounter_method_rates'][0]['encounter_method']['url'], '{}{}/encounter-method/{}/'.format(test_host, api_v2, encounter_method.pk))
|
|
self.assertEqual(response.data['encounter_method_rates'][0]['version_details'][0]['rate'], location_area_encounter_rate.rate)
|
|
self.assertEqual(response.data['encounter_method_rates'][0]['version_details'][0]['version']['name'], location_area_encounter_rate.version.name)
|
|
self.assertEqual(response.data['encounter_method_rates'][0]['version_details'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, location_area_encounter_rate.version.pk))
|
|
# encounter params
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['pokemon']['name'], pokemon1.name)
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['pokemon']['url'], '{}{}/pokemon/{}/'.format(test_host, api_v2, pokemon1.pk))
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['version_details'][0]['max_chance'], encounter_slot1.rarity)
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['version_details'][0]['version']['name'], encounter1.version.name)
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['version_details'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, encounter1.version.pk))
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['version_details'][0]['encounter_details'][0]['chance'], encounter_slot1.rarity)
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['version_details'][0]['encounter_details'][0]['method']['name'], encounter_slot1.encounter_method.name)
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['version_details'][0]['encounter_details'][0]['method']['url'], '{}{}/encounter-method/{}/'.format(test_host, api_v2, encounter_slot1.encounter_method.pk))
|
|
|
|
self.assertEqual(response.data['pokemon_encounters'][1]['pokemon']['name'], pokemon2.name)
|
|
self.assertEqual(response.data['pokemon_encounters'][1]['pokemon']['url'], '{}{}/pokemon/{}/'.format(test_host, api_v2, pokemon2.pk))
|
|
self.assertEqual(response.data['pokemon_encounters'][1]['version_details'][0]['max_chance'], encounter_slot2.rarity)
|
|
self.assertEqual(response.data['pokemon_encounters'][1]['version_details'][0]['version']['name'], encounter2.version.name)
|
|
self.assertEqual(response.data['pokemon_encounters'][1]['version_details'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, encounter2.version.pk))
|
|
self.assertEqual(response.data['pokemon_encounters'][1]['version_details'][0]['encounter_details'][0]['chance'], encounter_slot2.rarity)
|
|
self.assertEqual(response.data['pokemon_encounters'][1]['version_details'][0]['encounter_details'][0]['method']['name'], encounter_slot2.encounter_method.name)
|
|
self.assertEqual(response.data['pokemon_encounters'][1]['version_details'][0]['encounter_details'][0]['method']['url'], '{}{}/encounter-method/{}/'.format(test_host, api_v2, encounter_slot2.encounter_method.pk))
|
|
|
|
|
|
"""
|
|
Contest Tests
|
|
"""
|
|
|
|
def test_contest_type_api(self):
|
|
|
|
contest_type = self.setup_contest_type_data(name='base cntst tp')
|
|
contest_type_name = self.setup_contest_type_name_data(contest_type, name='base cntst tp name')
|
|
berry_flavor = self.setup_berry_flavor_data(name="bry for base cntst tp", contest_type=contest_type)
|
|
|
|
response = self.client.get('{}/contest-type/{}/'.format(api_v2, contest_type.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], contest_type.pk)
|
|
self.assertEqual(response.data['name'], contest_type.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], contest_type_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], contest_type_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, contest_type_name.language.pk))
|
|
# berry params
|
|
self.assertEqual(response.data['berry_flavor']['name'], berry_flavor.name)
|
|
self.assertEqual(response.data['berry_flavor']['url'], '{}{}/berry-flavor/{}/'.format(test_host, api_v2, berry_flavor.pk))
|
|
|
|
|
|
def test_contest_effect_api(self):
|
|
|
|
contest_effect = self.setup_contest_effect_data(appeal=10, jam=20)
|
|
contest_effect_flavor_text = self.setup_contest_effect_flavor_text_data(contest_effect, flavor_text='base cntst efct flvr txt')
|
|
contest_effect_effect_text = self.setup_contest_effect_effect_text_data(contest_effect, effect='base cntst efct eftc txt')
|
|
|
|
response = self.client.get('{}/contest-effect/{}/'.format(api_v2, contest_effect.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], contest_effect.pk)
|
|
self.assertEqual(response.data['appeal'], contest_effect.appeal)
|
|
self.assertEqual(response.data['jam'], contest_effect.jam)
|
|
# effect text params
|
|
self.assertEqual(response.data['effect_entries'][0]['effect'], contest_effect_effect_text.effect)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['name'], contest_effect_effect_text.language.name)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, contest_effect_effect_text.language.pk))
|
|
# flavor text params
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['flavor_text'], contest_effect_flavor_text.flavor_text)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['name'], contest_effect_flavor_text.language.name)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, contest_effect_flavor_text.language.pk))
|
|
|
|
|
|
def test_super_contest_effect_api(self):
|
|
|
|
super_contest_effect = self.setup_super_contest_effect_data(appeal=10)
|
|
super_contest_effect_flavor_text = self.setup_super_contest_effect_flavor_text_data(super_contest_effect, flavor_text='base spr cntst efct flvr txt')
|
|
move = self.setup_move_data(name="mv for base spr cntst efct", super_contest_effect=super_contest_effect)
|
|
|
|
response = self.client.get('{}/super-contest-effect/{}/'.format(api_v2, super_contest_effect.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], super_contest_effect.pk)
|
|
self.assertEqual(response.data['appeal'], super_contest_effect.appeal)
|
|
# flavor text params
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['flavor_text'], super_contest_effect_flavor_text.flavor_text)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['name'], super_contest_effect_flavor_text.language.name)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, super_contest_effect_flavor_text.language.pk))
|
|
# move params
|
|
self.assertEqual(response.data['moves'][0]['name'], move.name)
|
|
self.assertEqual(response.data['moves'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, move.pk))
|
|
|
|
|
|
|
|
"""
|
|
Type Tests
|
|
"""
|
|
|
|
def test_type_api(self):
|
|
|
|
type = self.setup_type_data(name='base tp')
|
|
type_name = self.setup_type_name_data(type, name='base tp nm')
|
|
type_game_index = self.setup_type_game_index_data(type, game_index=10)
|
|
move = self.setup_move_data(name="mv for base tp", type=type)
|
|
pokemon = self.setup_pokemon_data(name="pkmn for base tp")
|
|
pokemon_type = self.setup_pokemon_type_data(pokemon=pokemon, type=type)
|
|
|
|
no_damage_to = self.setup_type_data(name='no damage to tp')
|
|
half_damage_to = self.setup_type_data(name='half damage to tp')
|
|
double_damage_to = self.setup_type_data(name='double damage to tp')
|
|
no_damage_from = self.setup_type_data(name='no damage from tp')
|
|
half_damage_from = self.setup_type_data(name='half damage from tp')
|
|
double_damage_from = self.setup_type_data(name='double damage from tp')
|
|
|
|
# type relations
|
|
no_damage_to_relation = TypeEfficacy (
|
|
damage_type = type,
|
|
target_type = no_damage_to,
|
|
damage_factor = 0
|
|
)
|
|
no_damage_to_relation.save()
|
|
|
|
half_damage_to_type_relation = TypeEfficacy (
|
|
damage_type = type,
|
|
target_type = half_damage_to,
|
|
damage_factor = 50
|
|
)
|
|
half_damage_to_type_relation.save()
|
|
|
|
double_damage_to_type_relation = TypeEfficacy (
|
|
damage_type = type,
|
|
target_type = double_damage_to,
|
|
damage_factor = 200
|
|
)
|
|
double_damage_to_type_relation.save()
|
|
|
|
no_damage_from_relation = TypeEfficacy (
|
|
damage_type = no_damage_from,
|
|
target_type = type,
|
|
damage_factor = 0
|
|
)
|
|
no_damage_from_relation.save()
|
|
|
|
half_damage_from_type_relation = TypeEfficacy (
|
|
damage_type = half_damage_from,
|
|
target_type = type,
|
|
damage_factor = 50
|
|
)
|
|
half_damage_from_type_relation.save()
|
|
|
|
double_damage_from_type_relation = TypeEfficacy (
|
|
damage_type = double_damage_from,
|
|
target_type = type,
|
|
damage_factor = 200
|
|
)
|
|
double_damage_from_type_relation.save()
|
|
|
|
response = self.client.get('{}/type/{}/'.format(api_v2, type.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], type.pk)
|
|
self.assertEqual(response.data['name'], type.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], type_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], type_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, type_name.language.pk))
|
|
# generation params
|
|
self.assertEqual(response.data['generation']['name'], type.generation.name)
|
|
self.assertEqual(response.data['generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, type.generation.pk))
|
|
# damage class params
|
|
self.assertEqual(response.data['move_damage_class']['name'], type.move_damage_class.name)
|
|
self.assertEqual(response.data['move_damage_class']['url'], '{}{}/move-damage-class/{}/'.format(test_host, api_v2, type.move_damage_class.pk))
|
|
# move params
|
|
self.assertEqual(response.data['moves'][0]['name'], move.name)
|
|
self.assertEqual(response.data['moves'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, move.pk))
|
|
# pokemon params
|
|
self.assertEqual(response.data['pokemon'][0]['slot'], pokemon_type.slot)
|
|
self.assertEqual(response.data['pokemon'][0]['pokemon']['name'], pokemon.name)
|
|
self.assertEqual(response.data['pokemon'][0]['pokemon']['url'], '{}{}/pokemon/{}/'.format(test_host, api_v2, pokemon.pk))
|
|
# damage relations params
|
|
self.assertEqual(response.data['damage_relations']['no_damage_to'][0]['name'], no_damage_to.name)
|
|
self.assertEqual(response.data['damage_relations']['no_damage_to'][0]['url'], '{}{}/type/{}/'.format(test_host, api_v2, no_damage_to.pk))
|
|
self.assertEqual(response.data['damage_relations']['half_damage_to'][0]['name'], half_damage_to.name)
|
|
self.assertEqual(response.data['damage_relations']['half_damage_to'][0]['url'], '{}{}/type/{}/'.format(test_host, api_v2, half_damage_to.pk))
|
|
self.assertEqual(response.data['damage_relations']['double_damage_to'][0]['name'], double_damage_to.name)
|
|
self.assertEqual(response.data['damage_relations']['double_damage_to'][0]['url'], '{}{}/type/{}/'.format(test_host, api_v2, double_damage_to.pk))
|
|
self.assertEqual(response.data['damage_relations']['no_damage_from'][0]['name'], no_damage_from.name)
|
|
self.assertEqual(response.data['damage_relations']['no_damage_from'][0]['url'], '{}{}/type/{}/'.format(test_host, api_v2, no_damage_from.pk))
|
|
self.assertEqual(response.data['damage_relations']['half_damage_from'][0]['name'], half_damage_from.name)
|
|
self.assertEqual(response.data['damage_relations']['half_damage_from'][0]['url'], '{}{}/type/{}/'.format(test_host, api_v2, half_damage_from.pk))
|
|
self.assertEqual(response.data['damage_relations']['double_damage_from'][0]['name'], double_damage_from.name)
|
|
self.assertEqual(response.data['damage_relations']['double_damage_from'][0]['url'], '{}{}/type/{}/'.format(test_host, api_v2, double_damage_from.pk))
|
|
# game indices params
|
|
self.assertEqual(response.data['game_indices'][0]['game_index'], type_game_index.game_index)
|
|
self.assertEqual(response.data['game_indices'][0]['generation']['name'], type_game_index.generation.name)
|
|
self.assertEqual(response.data['game_indices'][0]['generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, type_game_index.generation.pk))
|
|
|
|
|
|
"""
|
|
Pokedex Tests
|
|
"""
|
|
|
|
def test_pokedex_api(self):
|
|
|
|
pokedex = self.setup_pokedex_data(name='base pkdx')
|
|
pokedex_name = self.setup_pokedex_name_data(pokedex, name='base pkdx name')
|
|
pokedex_description = self.setup_pokedex_description_data(pokedex, description='base pkdx desc')
|
|
pokemon_species = self.setup_pokemon_species_data(name='pkmn spcs for base pkdx')
|
|
dex_entry = self.setup_pokemon_dex_entry_data(pokedex=pokedex, pokemon_species=pokemon_species)
|
|
|
|
response = self.client.get('{}/pokedex/{}/'.format(api_v2, pokedex.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pokedex.pk)
|
|
self.assertEqual(response.data['name'], pokedex.name)
|
|
self.assertEqual(response.data['is_main_series'], pokedex.is_main_series)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], pokedex_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], pokedex_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokedex_name.language.pk))
|
|
# descriptions params
|
|
self.assertEqual(response.data['descriptions'][0]['description'], pokedex_description.description)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['name'], pokedex_description.language.name)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokedex_description.language.pk))
|
|
# region params
|
|
self.assertEqual(response.data['region']['name'], pokedex.region.name)
|
|
self.assertEqual(response.data['region']['url'], '{}{}/region/{}/'.format(test_host, api_v2, pokedex.region.pk))
|
|
# species params
|
|
self.assertEqual(response.data['pokemon_entries'][0]['entry_number'], dex_entry.pokedex_number)
|
|
self.assertEqual(response.data['pokemon_entries'][0]['pokemon_species']['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_entries'][0]['pokemon_species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
|
|
|
|
"""
|
|
Move Tests
|
|
"""
|
|
|
|
def test_move_ailment_api(self):
|
|
|
|
move_ailment = self.setup_move_ailment_data(name='base mv almnt')
|
|
move_ailment_name = self.setup_move_ailment_name_data(move_ailment, name='base mv almnt name')
|
|
move = self.setup_move_data(name='mv for base mv almnt')
|
|
move_meta = self.setup_move_meta_data(move=move, move_ailment=move_ailment)
|
|
|
|
response = self.client.get('{}/move-ailment/{}/'.format(api_v2, move_ailment.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], move_ailment.pk)
|
|
self.assertEqual(response.data['name'], move_ailment.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], move_ailment_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], move_ailment_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_ailment_name.language.pk))
|
|
# move params
|
|
self.assertEqual(response.data['moves'][0]['name'], move.name)
|
|
self.assertEqual(response.data['moves'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, move.pk))
|
|
|
|
|
|
def test_move_battle_style_api(self):
|
|
|
|
move_battle_style = self.setup_move_battle_style_data(name='base mv btl stl')
|
|
move_battle_style_name = self.setup_move_battle_style_name_data(move_battle_style, name='base mv btl stl name')
|
|
|
|
response = self.client.get('{}/move-battle-style/{}/'.format(api_v2, move_battle_style.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], move_battle_style.pk)
|
|
self.assertEqual(response.data['name'], move_battle_style.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], move_battle_style_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], move_battle_style_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_battle_style_name.language.pk))
|
|
|
|
|
|
def test_move_category_api(self):
|
|
|
|
move_category = self.setup_move_category_data(name='base mv ctgry')
|
|
move_category_description = self.setup_move_category_description_data(move_category, description='base mv ctgry description')
|
|
move = self.setup_move_data(name='mv for base mv ctgry')
|
|
move_meta = self.setup_move_meta_data(move=move, move_category=move_category)
|
|
|
|
response = self.client.get('{}/move-category/{}/'.format(api_v2, move_category.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], move_category.pk)
|
|
self.assertEqual(response.data['name'], move_category.name)
|
|
# name params
|
|
self.assertEqual(response.data['descriptions'][0]['description'], move_category_description.description)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['name'], move_category_description.language.name)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_category_description.language.pk))
|
|
# move params
|
|
self.assertEqual(response.data['moves'][0]['name'], move.name)
|
|
self.assertEqual(response.data['moves'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, move.pk))
|
|
|
|
|
|
def test_move_damage_class_api(self):
|
|
|
|
move_damage_class = self.setup_move_damage_class_data(name='base mv dmg cls')
|
|
move_damage_class_name = self.setup_move_damage_class_name_data(move_damage_class, name='base mv dmg cls nm')
|
|
move_damage_class_description = self.setup_move_damage_class_description_data(move_damage_class, description='base mv dmg cls desc')
|
|
move = self.setup_move_data(name='mv for base mv dmg cls', move_damage_class=move_damage_class)
|
|
|
|
response = self.client.get('{}/move-damage-class/{}/'.format(api_v2, move_damage_class.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], move_damage_class.pk)
|
|
self.assertEqual(response.data['name'], move_damage_class.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], move_damage_class_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], move_damage_class_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_damage_class_name.language.pk))
|
|
# description params
|
|
self.assertEqual(response.data['descriptions'][0]['description'], move_damage_class_description.description)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['name'], move_damage_class_description.language.name)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_damage_class_description.language.pk))
|
|
# move params
|
|
self.assertEqual(response.data['moves'][0]['name'], move.name)
|
|
self.assertEqual(response.data['moves'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, move.pk))
|
|
|
|
|
|
def test_move_learn_method_api(self):
|
|
|
|
move_learn_method = self.setup_move_learn_method_data(name='base mv lrn mthd')
|
|
move_learn_method_name = self.setup_move_learn_method_name_data(move_learn_method, name='base mv lrn mthd nm')
|
|
move_learn_method_description = self.setup_move_learn_method_description_data(move_learn_method, description='base mv lrn mthd desc')
|
|
version_group = self.setup_version_group_data(name='ver grp for base mv lrn mthd')
|
|
self.setup_version_group_move_learn_method_data(version_group=version_group, move_learn_method=move_learn_method)
|
|
|
|
response = self.client.get('{}/move-learn-method/{}/'.format(api_v2, move_learn_method.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], move_learn_method.pk)
|
|
self.assertEqual(response.data['name'], move_learn_method.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], move_learn_method_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], move_learn_method_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_learn_method_name.language.pk))
|
|
# description params
|
|
self.assertEqual(response.data['descriptions'][0]['description'], move_learn_method_description.description)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['name'], move_learn_method_description.language.name)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_learn_method_description.language.pk))
|
|
# version group params
|
|
self.assertEqual(response.data['version_groups'][0]['name'], version_group.name)
|
|
self.assertEqual(response.data['version_groups'][0]['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, version_group.pk))
|
|
|
|
|
|
def test_move_target_api(self):
|
|
|
|
move_target = self.setup_move_target_data(name='base mv trgt')
|
|
move_target_name = self.setup_move_target_name_data(move_target, name='base mv trgt nm')
|
|
move_target_description = self.setup_move_target_description_data(move_target, description='base mv trgt desc')
|
|
move = self.setup_move_data(name='mv for base mv trgt', move_target=move_target)
|
|
|
|
response = self.client.get('{}/move-target/{}/'.format(api_v2, move_target.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], move_target.pk)
|
|
self.assertEqual(response.data['name'], move_target.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], move_target_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], move_target_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_target_name.language.pk))
|
|
# description params
|
|
self.assertEqual(response.data['descriptions'][0]['description'], move_target_description.description)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['name'], move_target_description.language.name)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_target_description.language.pk))
|
|
# move params
|
|
self.assertEqual(response.data['moves'][0]['name'], move.name)
|
|
self.assertEqual(response.data['moves'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, move.pk))
|
|
|
|
|
|
def test_move_api(self):
|
|
|
|
move_effect = self.setup_move_effect_data()
|
|
move_effect_effect_text = self.setup_move_effect_effect_text_data(move_effect)
|
|
move = self.setup_move_data(name='base mv', move_effect=move_effect)
|
|
move_name = self.setup_move_name_data(move, name='base mv nm')
|
|
move_meta = self.setup_move_meta_data(move)
|
|
move_stat_change = self.setup_move_stat_change_data(move=move, change=2)
|
|
move_change = self.setup_move_change_data(move, power=10, pp=20, accuracy=30)
|
|
move_effect_change = self.setup_move_effect_change_data(move_effect)
|
|
move_effect_change_effect_text = self.setup_move_effect_change_effect_text_data(move_effect_change=move_effect_change, effect='efct tx for mv efct chng')
|
|
|
|
after_move = self.setup_move_data(name='after mv')
|
|
before_move = self.setup_move_data(name='before mv')
|
|
|
|
before_combo = self.setup_contest_combo_data(move, after_move)
|
|
after_combo = self.setup_contest_combo_data(before_move, move)
|
|
before_super_combo = self.setup_super_contest_combo_data(move, after_move)
|
|
after_super_combo = self.setup_super_contest_combo_data(before_move, move)
|
|
|
|
response = self.client.get('{}/move/{}/'.format(api_v2, move.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], move.pk)
|
|
self.assertEqual(response.data['name'], move.name)
|
|
self.assertEqual(response.data['accuracy'], move.accuracy)
|
|
self.assertEqual(response.data['effect_chance'], move.move_effect_chance)
|
|
self.assertEqual(response.data['power'], move.power)
|
|
self.assertEqual(response.data['pp'], move.pp)
|
|
self.assertEqual(response.data['priority'], move.priority)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], move_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], move_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_name.language.pk))
|
|
# damage class params
|
|
self.assertEqual(response.data['damage_class']['name'], move.move_damage_class.name)
|
|
self.assertEqual(response.data['damage_class']['url'], '{}{}/move-damage-class/{}/'.format(test_host, api_v2, move.move_damage_class.pk))
|
|
# contest type params
|
|
self.assertEqual(response.data['contest_type']['name'], move.contest_type.name)
|
|
self.assertEqual(response.data['contest_type']['url'], '{}{}/contest-type/{}/'.format(test_host, api_v2, move.contest_type.pk))
|
|
# contest effect params
|
|
self.assertEqual(response.data['contest_effect']['url'], '{}{}/contest-effect/{}/'.format(test_host, api_v2, move.contest_effect.pk))
|
|
# super contest effect params
|
|
self.assertEqual(response.data['super_contest_effect']['url'], '{}{}/super-contest-effect/{}/'.format(test_host, api_v2, move.super_contest_effect.pk))
|
|
# generation params
|
|
self.assertEqual(response.data['generation']['name'], move.generation.name)
|
|
self.assertEqual(response.data['generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, move.generation.pk))
|
|
# target params
|
|
self.assertEqual(response.data['target']['name'], move.move_target.name)
|
|
self.assertEqual(response.data['target']['url'], '{}{}/move-target/{}/'.format(test_host, api_v2, move.move_target.pk))
|
|
# type params
|
|
self.assertEqual(response.data['type']['name'], move.type.name)
|
|
self.assertEqual(response.data['type']['url'], '{}{}/type/{}/'.format(test_host, api_v2, move.type.pk))
|
|
# stat change params
|
|
self.assertEqual(response.data['stat_changes'][0]['change'], move_stat_change.change)
|
|
self.assertEqual(response.data['stat_changes'][0]['stat']['name'], move_stat_change.stat.name)
|
|
self.assertEqual(response.data['stat_changes'][0]['stat']['url'], '{}{}/stat/{}/'.format(test_host, api_v2, move_stat_change.stat.pk))
|
|
# effect entries params
|
|
self.assertEqual(response.data['effect_entries'][0]['effect'], move_effect_effect_text.effect)
|
|
self.assertEqual(response.data['effect_entries'][0]['short_effect'], move_effect_effect_text.short_effect)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['name'], move_effect_effect_text.language.name)
|
|
self.assertEqual(response.data['effect_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_effect_effect_text.language.pk))
|
|
# meta data
|
|
self.assertEqual(response.data['meta']['min_hits'], move_meta.min_hits)
|
|
self.assertEqual(response.data['meta']['max_hits'], move_meta.max_hits)
|
|
self.assertEqual(response.data['meta']['min_turns'], move_meta.min_turns)
|
|
self.assertEqual(response.data['meta']['max_turns'], move_meta.max_turns)
|
|
self.assertEqual(response.data['meta']['drain'], move_meta.drain)
|
|
self.assertEqual(response.data['meta']['healing'], move_meta.healing)
|
|
self.assertEqual(response.data['meta']['crit_rate'], move_meta.crit_rate)
|
|
self.assertEqual(response.data['meta']['ailment_chance'], move_meta.ailment_chance)
|
|
self.assertEqual(response.data['meta']['flinch_chance'], move_meta.flinch_chance)
|
|
self.assertEqual(response.data['meta']['stat_chance'], move_meta.stat_chance)
|
|
# ailment params
|
|
self.assertEqual(response.data['meta']['ailment']['name'], move_meta.move_meta_ailment.name)
|
|
self.assertEqual(response.data['meta']['ailment']['url'], '{}{}/move-ailment/{}/'.format(test_host, api_v2, move_meta.move_meta_ailment.pk))
|
|
# category params
|
|
self.assertEqual(response.data['meta']['category']['name'], move_meta.move_meta_category.name)
|
|
self.assertEqual(response.data['meta']['category']['url'], '{}{}/move-category/{}/'.format(test_host, api_v2, move_meta.move_meta_category.pk))
|
|
# combo params
|
|
self.assertEqual(response.data['contest_combos']['normal']['use_before'][0]['name'], after_move.name)
|
|
self.assertEqual(response.data['contest_combos']['normal']['use_before'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, after_move.pk))
|
|
self.assertEqual(response.data['contest_combos']['normal']['use_after'][0]['name'], before_move.name)
|
|
self.assertEqual(response.data['contest_combos']['normal']['use_after'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, before_move.pk))
|
|
self.assertEqual(response.data['contest_combos']['super']['use_before'][0]['name'], after_move.name)
|
|
self.assertEqual(response.data['contest_combos']['super']['use_before'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, after_move.pk))
|
|
self.assertEqual(response.data['contest_combos']['super']['use_after'][0]['name'], before_move.name)
|
|
self.assertEqual(response.data['contest_combos']['super']['use_after'][0]['url'], '{}{}/move/{}/'.format(test_host, api_v2, before_move.pk))
|
|
# change params
|
|
self.assertEqual(response.data['past_values'][0]['accuracy'], move_change.accuracy)
|
|
self.assertEqual(response.data['past_values'][0]['power'], move_change.power)
|
|
self.assertEqual(response.data['past_values'][0]['pp'], move_change.pp)
|
|
self.assertEqual(response.data['past_values'][0]['effect_chance'], move_change.move_effect_chance)
|
|
self.assertEqual(response.data['past_values'][0]['version_group']['name'], move_change.version_group.name)
|
|
self.assertEqual(response.data['past_values'][0]['version_group']['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, move_change.version_group.pk))
|
|
# effect changes
|
|
self.assertEqual(response.data['effect_changes'][0]['version_group']['name'], move_effect_change.version_group.name)
|
|
self.assertEqual(response.data['effect_changes'][0]['version_group']['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, move_effect_change.version_group.pk))
|
|
self.assertEqual(response.data['effect_changes'][0]['effect_entries'][0]['effect'], move_effect_change_effect_text.effect)
|
|
self.assertEqual(response.data['effect_changes'][0]['effect_entries'][0]['language']['name'], move_effect_change_effect_text.language.name)
|
|
self.assertEqual(response.data['effect_changes'][0]['effect_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, move_effect_change_effect_text.language.pk))
|
|
|
|
|
|
|
|
"""
|
|
Stat Tests
|
|
"""
|
|
|
|
def test_stat_api(self):
|
|
|
|
stat = self.setup_stat_data(name='base stt')
|
|
stat_name = self.setup_stat_name_data(stat, name='base stt name')
|
|
increase_move = self.setup_move_data(name="incrs mv for base stt")
|
|
increase_move_stat_change = self.setup_move_stat_change_data(move=increase_move, stat=stat, change=2)
|
|
decrease_move = self.setup_move_data(name="dcrs mv for base stt")
|
|
decrease_move_stat_change = self.setup_move_stat_change_data(move=decrease_move, stat=stat, change=(-2))
|
|
increase_nature = self.setup_nature_data(name="incrs ntr for base stt", increased_stat=stat)
|
|
decrease_nature = self.setup_nature_data(name="dcrs ntr for base stt", decreased_stat=stat)
|
|
characteristic = self.setup_characteristic_data(stat=stat)
|
|
|
|
response = self.client.get('{}/stat/{}/'.format(api_v2, stat.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], stat.pk)
|
|
self.assertEqual(response.data['name'], stat.name)
|
|
self.assertEqual(response.data['game_index'], stat.game_index)
|
|
self.assertEqual(response.data['is_battle_only'], stat.is_battle_only)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], stat_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], stat_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, stat_name.language.pk))
|
|
# move damage class params
|
|
self.assertEqual(response.data['move_damage_class']['name'], stat.move_damage_class.name)
|
|
self.assertEqual(response.data['move_damage_class']['url'], '{}{}/move-damage-class/{}/'.format(test_host, api_v2, stat.move_damage_class.pk))
|
|
# nature params
|
|
self.assertEqual(response.data['affecting_natures']['increase'][0]['name'], increase_nature.name)
|
|
self.assertEqual(response.data['affecting_natures']['increase'][0]['url'], '{}{}/nature/{}/'.format(test_host, api_v2, increase_nature.pk))
|
|
self.assertEqual(response.data['affecting_natures']['decrease'][0]['name'], decrease_nature.name)
|
|
self.assertEqual(response.data['affecting_natures']['decrease'][0]['url'], '{}{}/nature/{}/'.format(test_host, api_v2, decrease_nature.pk))
|
|
# move params
|
|
self.assertEqual(response.data['affecting_moves']['increase'][0]['change'], increase_move_stat_change.change)
|
|
self.assertEqual(response.data['affecting_moves']['increase'][0]['move']['name'], increase_move.name)
|
|
self.assertEqual(response.data['affecting_moves']['increase'][0]['move']['url'], '{}{}/move/{}/'.format(test_host, api_v2, increase_move.pk))
|
|
self.assertEqual(response.data['affecting_moves']['decrease'][0]['change'], decrease_move_stat_change.change)
|
|
self.assertEqual(response.data['affecting_moves']['decrease'][0]['move']['name'], decrease_move.name)
|
|
self.assertEqual(response.data['affecting_moves']['decrease'][0]['move']['url'], '{}{}/move/{}/'.format(test_host, api_v2, decrease_move.pk))
|
|
# characteristics params
|
|
self.assertEqual(response.data['characteristics'][0]['url'], '{}{}/characteristic/{}/'.format(test_host, api_v2, characteristic.pk))
|
|
|
|
|
|
def test_pokeathlon_stat_api(self):
|
|
|
|
pokeathlon_stat = self.setup_pokeathlon_stat_data(name='base pkathln stt')
|
|
pokeathlon_stat_name = self.setup_pokeathlon_stat_name_data(pokeathlon_stat, name='base pkathln stt name')
|
|
|
|
response = self.client.get('{}/pokeathlon-stat/{}/'.format(api_v2, pokeathlon_stat.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pokeathlon_stat.pk)
|
|
self.assertEqual(response.data['name'], pokeathlon_stat.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], pokeathlon_stat_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], pokeathlon_stat_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokeathlon_stat_name.language.pk))
|
|
|
|
|
|
"""
|
|
Characteristic Tests
|
|
"""
|
|
|
|
def test_characteristic_api(self):
|
|
|
|
characteristic = self.setup_characteristic_data(gene_mod_5=5)
|
|
characteristic_description = self.setup_characteristic_description_data(characteristic, description='base char desc')
|
|
|
|
response = self.client.get('{}/characteristic/{}/'.format(api_v2, characteristic.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], characteristic.pk)
|
|
self.assertEqual(response.data['gene_modulo'], characteristic.gene_mod_5)
|
|
# name params
|
|
self.assertEqual(response.data['descriptions'][0]['description'], characteristic_description.description)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['name'], characteristic_description.language.name)
|
|
self.assertEqual(response.data['descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, characteristic_description.language.pk))
|
|
# stat params
|
|
self.assertEqual(response.data['highest_stat']['name'], characteristic.stat.name)
|
|
self.assertEqual(response.data['highest_stat']['url'], '{}{}/stat/{}/'.format(test_host, api_v2, characteristic.stat.pk))
|
|
|
|
|
|
|
|
"""
|
|
Nature Tests
|
|
"""
|
|
|
|
def test_nature_api(self):
|
|
|
|
hates_flavor = self.setup_berry_flavor_data(name='hts flvr for base ntr')
|
|
likes_flavor = self.setup_berry_flavor_data(name='lks flvr for base ntr')
|
|
decreased_stat = self.setup_stat_data(name='dcrs stt for base ntr')
|
|
increased_stat = self.setup_stat_data(name='ncrs stt for base ntr')
|
|
nature = self.setup_nature_data(name='base ntr', hates_flavor=hates_flavor, likes_flavor=likes_flavor, decreased_stat=decreased_stat, increased_stat=increased_stat)
|
|
nature_name = self.setup_nature_name_data(nature, name='base ntr name')
|
|
|
|
pokeathlon_stat = self.setup_pokeathlon_stat_data(name='pkeathln stt for ntr stt')
|
|
nature_pokeathlon_stat = self.setup_nature_pokeathlon_stat_data(nature=nature, pokeathlon_stat=pokeathlon_stat, max_change=1)
|
|
|
|
move_battle_style = self.setup_move_battle_style_data(name='mv btl stl for ntr stt')
|
|
nature_battle_style_preference = self.setup_nature_battle_style_preference_data(nature=nature, move_battle_style=move_battle_style)
|
|
|
|
response = self.client.get('{}/nature/{}/'.format(api_v2, nature.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], nature.pk)
|
|
self.assertEqual(response.data['name'], nature.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], nature_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], nature_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, nature_name.language.pk))
|
|
# stat params
|
|
self.assertEqual(response.data['decreased_stat']['name'], decreased_stat.name)
|
|
self.assertEqual(response.data['decreased_stat']['url'], '{}{}/stat/{}/'.format(test_host, api_v2, decreased_stat.pk))
|
|
self.assertEqual(response.data['increased_stat']['name'], increased_stat.name)
|
|
self.assertEqual(response.data['increased_stat']['url'], '{}{}/stat/{}/'.format(test_host, api_v2, increased_stat.pk))
|
|
# flavor params
|
|
self.assertEqual(response.data['hates_flavor']['name'], hates_flavor.name)
|
|
self.assertEqual(response.data['hates_flavor']['url'], '{}{}/berry-flavor/{}/'.format(test_host, api_v2, hates_flavor.pk))
|
|
self.assertEqual(response.data['likes_flavor']['name'], likes_flavor.name)
|
|
self.assertEqual(response.data['likes_flavor']['url'], '{}{}/berry-flavor/{}/'.format(test_host, api_v2, likes_flavor.pk))
|
|
# pokeathlon stat params
|
|
self.assertEqual(response.data['pokeathlon_stat_changes'][0]['max_change'], nature_pokeathlon_stat.max_change)
|
|
self.assertEqual(response.data['pokeathlon_stat_changes'][0]['pokeathlon_stat']['name'], pokeathlon_stat.name)
|
|
self.assertEqual(response.data['pokeathlon_stat_changes'][0]['pokeathlon_stat']['url'], '{}{}/pokeathlon-stat/{}/'.format(test_host, api_v2, pokeathlon_stat.pk))
|
|
# pokeathlon stat params
|
|
self.assertEqual(response.data['move_battle_style_preferences'][0]['low_hp_preference'], nature_battle_style_preference.low_hp_preference)
|
|
self.assertEqual(response.data['move_battle_style_preferences'][0]['high_hp_preference'], nature_battle_style_preference.high_hp_preference)
|
|
self.assertEqual(response.data['move_battle_style_preferences'][0]['move_battle_style']['name'], move_battle_style.name)
|
|
self.assertEqual(response.data['move_battle_style_preferences'][0]['move_battle_style']['url'], '{}{}/move-battle-style/{}/'.format(test_host, api_v2, move_battle_style.pk))
|
|
|
|
|
|
|
|
"""
|
|
Pokemon Tests
|
|
"""
|
|
|
|
def test_pokemon_habitat_api(self):
|
|
|
|
pokemon_habitat = self.setup_pokemon_habitat_data(name='base pkmn hbtt trgr')
|
|
pokemon_habitat_name = self.setup_pokemon_habitat_name_data(pokemon_habitat, name='base pkmn hbtt name')
|
|
pokemon_species = self.setup_pokemon_species_data(pokemon_habitat=pokemon_habitat, name='pkmn spcs for pkmn hbtt')
|
|
|
|
response = self.client.get('{}/pokemon-habitat/{}/'.format(api_v2, pokemon_habitat.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pokemon_habitat.pk)
|
|
self.assertEqual(response.data['name'], pokemon_habitat.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], pokemon_habitat_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], pokemon_habitat_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokemon_habitat_name.language.pk))
|
|
# species params
|
|
self.assertEqual(response.data['pokemon_species'][0]['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_species'][0]['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
|
|
|
|
def test_pokemon_color_api(self):
|
|
|
|
pokemon_color = self.setup_pokemon_color_data(name='base pkmn clr trgr')
|
|
pokemon_color_name = self.setup_pokemon_color_name_data(pokemon_color, name='base pkmn clr name')
|
|
pokemon_species = self.setup_pokemon_species_data(pokemon_color=pokemon_color, name='pkmn spcs for pkmn clr')
|
|
|
|
response = self.client.get('{}/pokemon-color/{}/'.format(api_v2, pokemon_color.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pokemon_color.pk)
|
|
self.assertEqual(response.data['name'], pokemon_color.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], pokemon_color_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], pokemon_color_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokemon_color_name.language.pk))
|
|
# species params
|
|
self.assertEqual(response.data['pokemon_species'][0]['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_species'][0]['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
|
|
|
|
def test_pokemon_shape_api(self):
|
|
|
|
pokemon_shape = self.setup_pokemon_shape_data(name='base pkmn shp trgr')
|
|
pokemon_shape_name = self.setup_pokemon_shape_name_data(pokemon_shape, name='base pkmn shp name')
|
|
pokemon_species = self.setup_pokemon_species_data(pokemon_shape=pokemon_shape, name='pkmn spcs for pkmn shp')
|
|
|
|
response = self.client.get('{}/pokemon-shape/{}/'.format(api_v2, pokemon_shape.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pokemon_shape.pk)
|
|
self.assertEqual(response.data['name'], pokemon_shape.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], pokemon_shape_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], pokemon_shape_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokemon_shape_name.language.pk))
|
|
# awesome name params
|
|
self.assertEqual(response.data['awesome_names'][0]['awesome_name'], pokemon_shape_name.awesome_name)
|
|
self.assertEqual(response.data['awesome_names'][0]['language']['name'], pokemon_shape_name.language.name)
|
|
self.assertEqual(response.data['awesome_names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokemon_shape_name.language.pk))
|
|
# species params
|
|
self.assertEqual(response.data['pokemon_species'][0]['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_species'][0]['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
|
|
|
|
def test_pokemon_species_api(self):
|
|
|
|
evolves_from_species = self.setup_pokemon_species_data(name='evolves from pkmn spcs')
|
|
pokemon_species = self.setup_pokemon_species_data(evolves_from_species=evolves_from_species, name='base pkmn spcs')
|
|
pokemon_species_name = self.setup_pokemon_species_name_data(pokemon_species, name='base pkmn shp name')
|
|
pokemon_species_form_description = self.setup_pokemon_species_form_description_data(pokemon_species, description='frm dscr for pkmn spcs')
|
|
pokemon_species_flavor_text = self.setup_pokemon_species_flavor_text_data(pokemon_species, flavor_text='flvr txt for pkmn spcs')
|
|
pokedex = self.setup_pokedex_data(name='pkdx for pkmn spcs')
|
|
|
|
pal_park = self.setup_pal_park_data(pokemon_species=pokemon_species)
|
|
|
|
dex_number = self.setup_pokemon_dex_entry_data(pokemon_species=pokemon_species, pokedex=pokedex, entry_number=44)
|
|
|
|
egg_group = self.setup_egg_group_data(name='egg grp for pkmn spcs')
|
|
self.setup_pokemon_egg_group_data(pokemon_species=pokemon_species, egg_group=egg_group)
|
|
|
|
pokemon = self.setup_pokemon_data (pokemon_species=pokemon_species, name = 'pkm for base pkmn spcs')
|
|
pokemon_sprites = self.setup_pokemon_sprites_data(pokemon)
|
|
|
|
response = self.client.get('{}/pokemon-species/{}/'.format(api_v2, pokemon_species.pk), HTTP_HOST='testserver')
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pokemon_species.pk)
|
|
self.assertEqual(response.data['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['order'], pokemon_species.order)
|
|
self.assertEqual(response.data['capture_rate'], pokemon_species.capture_rate)
|
|
self.assertEqual(response.data['gender_rate'], pokemon_species.gender_rate)
|
|
self.assertEqual(response.data['base_happiness'], pokemon_species.base_happiness)
|
|
self.assertEqual(response.data['is_baby'], pokemon_species.is_baby)
|
|
self.assertEqual(response.data['hatch_counter'], pokemon_species.hatch_counter)
|
|
self.assertEqual(response.data['has_gender_differences'], pokemon_species.has_gender_differences)
|
|
self.assertEqual(response.data['forms_switchable'], pokemon_species.forms_switchable)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], pokemon_species_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], pokemon_species_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokemon_species_name.language.pk))
|
|
# growth rate params
|
|
self.assertEqual(response.data['growth_rate']['name'], pokemon_species.growth_rate.name)
|
|
self.assertEqual(response.data['growth_rate']['url'], '{}{}/growth-rate/{}/'.format(test_host, api_v2, pokemon_species.growth_rate.pk))
|
|
# dex number params
|
|
self.assertEqual(response.data['pokedex_numbers'][0]['entry_number'], dex_number.pokedex_number)
|
|
self.assertEqual(response.data['pokedex_numbers'][0]['pokedex']['name'], pokedex.name)
|
|
self.assertEqual(response.data['pokedex_numbers'][0]['pokedex']['url'], '{}{}/pokedex/{}/'.format(test_host, api_v2, pokedex.pk))
|
|
# egg group params
|
|
self.assertEqual(response.data['egg_groups'][0]['name'], egg_group.name)
|
|
self.assertEqual(response.data['egg_groups'][0]['url'], '{}{}/egg-group/{}/'.format(test_host, api_v2, egg_group.pk))
|
|
# generation params
|
|
self.assertEqual(response.data['generation']['name'], pokemon_species.generation.name)
|
|
self.assertEqual(response.data['generation']['url'], '{}{}/generation/{}/'.format(test_host, api_v2, pokemon_species.generation.pk))
|
|
# color params
|
|
self.assertEqual(response.data['color']['name'], pokemon_species.pokemon_color.name)
|
|
self.assertEqual(response.data['color']['url'], '{}{}/pokemon-color/{}/'.format(test_host, api_v2, pokemon_species.pokemon_color.pk))
|
|
# shape params
|
|
self.assertEqual(response.data['shape']['name'], pokemon_species.pokemon_shape.name)
|
|
self.assertEqual(response.data['shape']['url'], '{}{}/pokemon-shape/{}/'.format(test_host, api_v2, pokemon_species.pokemon_shape.pk))
|
|
# habitat params
|
|
self.assertEqual(response.data['habitat']['name'], pokemon_species.pokemon_habitat.name)
|
|
self.assertEqual(response.data['habitat']['url'], '{}{}/pokemon-habitat/{}/'.format(test_host, api_v2, pokemon_species.pokemon_habitat.pk))
|
|
# evolves from params
|
|
self.assertEqual(response.data['evolves_from_species']['name'], evolves_from_species.name)
|
|
self.assertEqual(response.data['evolves_from_species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, evolves_from_species.pk))
|
|
# genus params
|
|
self.assertEqual(response.data['genera'][0]['genus'], pokemon_species_name.genus)
|
|
self.assertEqual(response.data['genera'][0]['language']['name'], pokemon_species_name.language.name)
|
|
self.assertEqual(response.data['genera'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokemon_species_name.language.pk))
|
|
# pokemon varieties params
|
|
self.assertEqual(response.data['varieties'][0]['is_default'], pokemon.is_default)
|
|
self.assertEqual(response.data['varieties'][0]['pokemon']['name'], pokemon.name)
|
|
self.assertEqual(response.data['varieties'][0]['pokemon']['url'], '{}{}/pokemon/{}/'.format(test_host, api_v2, pokemon.pk))
|
|
# form descriptions params
|
|
self.assertEqual(response.data['form_descriptions'][0]['description'], pokemon_species_form_description.description)
|
|
self.assertEqual(response.data['form_descriptions'][0]['language']['name'], pokemon_species_form_description.language.name)
|
|
self.assertEqual(response.data['form_descriptions'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokemon_species_form_description.language.pk))
|
|
# flavor text params
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['flavor_text'], pokemon_species_flavor_text.flavor_text)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['name'], pokemon_species_flavor_text.language.name)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pokemon_species_flavor_text.language.pk))
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['version']['name'], pokemon_species_flavor_text.version.name)
|
|
self.assertEqual(response.data['flavor_text_entries'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, pokemon_species_flavor_text.version.pk))
|
|
# pal park params
|
|
self.assertEqual(response.data['pal_park_encounters'][0]['base_score'], pal_park.base_score)
|
|
self.assertEqual(response.data['pal_park_encounters'][0]['rate'], pal_park.rate)
|
|
self.assertEqual(response.data['pal_park_encounters'][0]['area']['name'], pal_park.pal_park_area.name)
|
|
self.assertEqual(response.data['pal_park_encounters'][0]['area']['url'], '{}{}/pal-park-area/{}/'.format(test_host, api_v2, pal_park.pal_park_area.pk))
|
|
|
|
|
|
def test_pokemon_api(self):
|
|
|
|
pokemon_species = self.setup_pokemon_species_data(name='pkmn spcs for base pkmn')
|
|
pokemon = self.setup_pokemon_data(pokemon_species=pokemon_species, name='base pkm')
|
|
pokemon_form = self.setup_pokemon_form_data(pokemon=pokemon, name='pkm form for base pkmn')
|
|
pokemon_ability = self.setup_pokemon_ability_data(pokemon=pokemon)
|
|
pokemon_stat = self.setup_pokemon_stat_data(pokemon=pokemon)
|
|
pokemon_type = self.setup_pokemon_type_data(pokemon=pokemon)
|
|
pokemon_item = self.setup_pokemon_item_data(pokemon=pokemon)
|
|
pokemon_sprites = self.setup_pokemon_sprites_data(pokemon=pokemon)
|
|
pokemon_game_index = self.setup_pokemon_game_index_data(pokemon=pokemon, game_index=10)
|
|
# To test issue #85, we will create one move that has multiple
|
|
# learn levels in different version groups. Later, we'll
|
|
# assert that we only got one move record back.
|
|
pokemon_move = self.setup_move_data(name='mv for pkmn')
|
|
pokemon_moves = []
|
|
for move in range(0,4):
|
|
version_group = self.setup_version_group_data(
|
|
name='ver grp '+str(move)+' for pkmn')
|
|
new_move = self.setup_pokemon_move_data(
|
|
pokemon=pokemon,
|
|
move=pokemon_move,
|
|
version_group=version_group,
|
|
level=move)
|
|
pokemon_moves.append(new_move)
|
|
|
|
encounter_method = self.setup_encounter_method_data(name='encntr mthd for lctn area')
|
|
location_area1 = self.setup_location_area_data(name='lctn1 area for base pkmn')
|
|
encounter_slot1 = self.setup_encounter_slot_data(encounter_method, slot=1, rarity=30)
|
|
encounter1 = self.setup_encounter_data(location_area=location_area1, pokemon=pokemon, encounter_slot=encounter_slot1, min_level=30, max_level=35)
|
|
location_area2 = self.setup_location_area_data(name='lctn2 area for base pkmn')
|
|
encounter_slot2 = self.setup_encounter_slot_data(encounter_method, slot=2, rarity=40)
|
|
encounter2 = self.setup_encounter_data(location_area=location_area2, pokemon=pokemon, encounter_slot=encounter_slot2, min_level=32, max_level=36)
|
|
response = self.client.get('{}/pokemon/{}/'.format(api_v2, pokemon.pk), HTTP_HOST='testserver')
|
|
|
|
sprites_data = json.loads(pokemon_sprites.sprites)
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pokemon.pk)
|
|
self.assertEqual(response.data['name'], pokemon.name)
|
|
self.assertEqual(response.data['order'], pokemon.order)
|
|
self.assertEqual(response.data['is_default'], pokemon.is_default)
|
|
self.assertEqual(response.data['height'], pokemon.height)
|
|
self.assertEqual(response.data['weight'], pokemon.weight)
|
|
self.assertEqual(response.data['base_experience'], pokemon.base_experience)
|
|
# species params
|
|
self.assertEqual(response.data['species']['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
# abilities params
|
|
self.assertEqual(response.data['abilities'][0]['is_hidden'], pokemon_ability.is_hidden)
|
|
self.assertEqual(response.data['abilities'][0]['slot'], pokemon_ability.slot)
|
|
self.assertEqual(response.data['abilities'][0]['ability']['name'], pokemon_ability.ability.name)
|
|
self.assertEqual(response.data['abilities'][0]['ability']['url'], '{}{}/ability/{}/'.format(test_host, api_v2, pokemon_ability.ability.pk))
|
|
# stat params
|
|
self.assertEqual(response.data['stats'][0]['base_stat'], pokemon_stat.base_stat)
|
|
self.assertEqual(response.data['stats'][0]['effort'], pokemon_stat.effort)
|
|
self.assertEqual(response.data['stats'][0]['stat']['name'], pokemon_stat.stat.name)
|
|
self.assertEqual(response.data['stats'][0]['stat']['url'], '{}{}/stat/{}/'.format(test_host, api_v2, pokemon_stat.stat.pk))
|
|
# stat params
|
|
self.assertEqual(response.data['types'][0]['slot'], pokemon_type.slot)
|
|
self.assertEqual(response.data['types'][0]['type']['name'], pokemon_type.type.name)
|
|
self.assertEqual(response.data['types'][0]['type']['url'], '{}{}/type/{}/'.format(test_host, api_v2, pokemon_type.type.pk))
|
|
# items params
|
|
self.assertEqual(response.data['held_items'][0]['item']['name'], pokemon_item.item.name)
|
|
self.assertEqual(response.data['held_items'][0]['item']['url'], '{}{}/item/{}/'.format(test_host, api_v2, pokemon_item.item.pk))
|
|
self.assertEqual(response.data['held_items'][0]['version_details'][0]['rarity'], pokemon_item.rarity)
|
|
self.assertEqual(response.data['held_items'][0]['version_details'][0]['version']['name'], pokemon_item.version.name)
|
|
self.assertEqual(response.data['held_items'][0]['version_details'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, pokemon_item.version.pk))
|
|
# move params -- Make sure that we only got one move back,
|
|
# but that we got all of the distinct version group and learn
|
|
# level values. (See issue #85)
|
|
# Number of Moves
|
|
expected = 1
|
|
actual = len(response.data['moves'])
|
|
self.assertEqual(expected, actual)
|
|
# Move name
|
|
expected = pokemon_moves[0].move.name
|
|
actual = response.data['moves'][0]['move']['name']
|
|
self.assertEqual(expected, actual)
|
|
# Move URL
|
|
expected = '{}{}/move/{}/'.format(
|
|
test_host,
|
|
api_v2,
|
|
pokemon_moves[0].move.pk)
|
|
actual = response.data['moves'][0]['move']['url']
|
|
self.assertEqual(expected, actual)
|
|
# Numbver of version groups
|
|
expected = len(pokemon_moves)
|
|
actual = len(response.data['moves'][0]['version_group_details'])
|
|
self.assertEqual(expected, actual)
|
|
for i in range(0,len(pokemon_moves)):
|
|
version = response.data['moves'][0]['version_group_details'][i]
|
|
# Learn Level
|
|
expected = pokemon_moves[i].level
|
|
actual = version['level_learned_at']
|
|
self.assertEqual(expected ,actual)
|
|
# Version Group Name
|
|
expected = pokemon_moves[i].version_group.name
|
|
actual = version['version_group']['name']
|
|
self.assertEqual(expected, actual)
|
|
# Version Group URL
|
|
expected = '{}{}/version-group/{}/'.format(
|
|
test_host,
|
|
api_v2,
|
|
pokemon_moves[i].version_group.pk)
|
|
actual = version['version_group']['url']
|
|
self.assertEqual(expected, actual)
|
|
# Learn Method Name
|
|
expected = pokemon_moves[i].move_learn_method.name
|
|
actual = version['move_learn_method']['name']
|
|
self.assertEqual(expected, actual)
|
|
# Learn Method URL
|
|
expected = '{}{}/move-learn-method/{}/'.format(
|
|
test_host,
|
|
api_v2,
|
|
pokemon_moves[i].move_learn_method.pk)
|
|
actual = version['move_learn_method']['url']
|
|
self.assertEqual(expected, actual)
|
|
# game indices params
|
|
self.assertEqual(response.data['game_indices'][0]['game_index'], pokemon_game_index.game_index)
|
|
self.assertEqual(response.data['game_indices'][0]['version']['name'], pokemon_game_index.version.name)
|
|
self.assertEqual(response.data['game_indices'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, pokemon_game_index.version.pk))
|
|
# form params
|
|
self.assertEqual(response.data['forms'][0]['name'], pokemon_form.name)
|
|
self.assertEqual(response.data['forms'][0]['url'], '{}{}/pokemon-form/{}/'.format(test_host, api_v2, pokemon_form.pk))
|
|
# encounter params
|
|
self.assertEqual(response.data['location_area_encounters'][0]['location_area']['name'], location_area1.name)
|
|
self.assertEqual(response.data['location_area_encounters'][0]['location_area']['url'], '{}{}/location-area/{}/'.format(test_host, api_v2, location_area1.pk))
|
|
self.assertEqual(response.data['location_area_encounters'][0]['version_details'][0]['max_chance'], encounter_slot1.rarity)
|
|
self.assertEqual(response.data['location_area_encounters'][0]['version_details'][0]['version']['name'], encounter1.version.name)
|
|
self.assertEqual(response.data['location_area_encounters'][0]['version_details'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, encounter1.version.pk))
|
|
self.assertEqual(response.data['location_area_encounters'][0]['version_details'][0]['encounter_details'][0]['chance'], encounter_slot1.rarity)
|
|
self.assertEqual(response.data['location_area_encounters'][0]['version_details'][0]['encounter_details'][0]['method']['name'], encounter_method.name)
|
|
self.assertEqual(response.data['location_area_encounters'][0]['version_details'][0]['encounter_details'][0]['method']['url'], '{}{}/encounter-method/{}/'.format(test_host, api_v2, encounter_method.pk))
|
|
|
|
self.assertEqual(response.data['location_area_encounters'][1]['location_area']['name'], location_area2.name)
|
|
self.assertEqual(response.data['location_area_encounters'][1]['location_area']['url'], '{}{}/location-area/{}/'.format(test_host, api_v2, location_area2.pk))
|
|
self.assertEqual(response.data['location_area_encounters'][1]['version_details'][0]['max_chance'], encounter_slot2.rarity)
|
|
self.assertEqual(response.data['location_area_encounters'][1]['version_details'][0]['version']['name'], encounter2.version.name)
|
|
self.assertEqual(response.data['location_area_encounters'][1]['version_details'][0]['version']['url'], '{}{}/version/{}/'.format(test_host, api_v2, encounter2.version.pk))
|
|
self.assertEqual(response.data['location_area_encounters'][1]['version_details'][0]['encounter_details'][0]['chance'], encounter_slot2.rarity)
|
|
self.assertEqual(response.data['location_area_encounters'][1]['version_details'][0]['encounter_details'][0]['method']['name'], encounter_method.name)
|
|
self.assertEqual(response.data['location_area_encounters'][1]['version_details'][0]['encounter_details'][0]['method']['url'], '{}{}/encounter-method/{}/'.format(test_host, api_v2, encounter_method.pk))
|
|
# sprite params
|
|
self.assertEqual(response.data['sprites']['front_default'], '{}{}'.format(test_host, sprites_data['front_default']))
|
|
self.assertEqual(response.data['sprites']['back_default'], None)
|
|
|
|
|
|
def test_pokemon_form_api(self):
|
|
|
|
pokemon_species = self.setup_pokemon_species_data()
|
|
pokemon = self.setup_pokemon_data(pokemon_species=pokemon_species)
|
|
pokemon_form = self.setup_pokemon_form_data(pokemon=pokemon, name='pkm form for base pkmn')
|
|
pokemon_form_sprites = self.setup_pokemon_form_sprites_data(pokemon_form)
|
|
|
|
sprites_data = json.loads(pokemon_form_sprites.sprites)
|
|
|
|
response = self.client.get('{}/pokemon-form/{}/'.format(api_v2, pokemon_form.pk), HTTP_HOST='testserver')
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pokemon_form.pk)
|
|
self.assertEqual(response.data['name'], pokemon_form.name)
|
|
self.assertEqual(response.data['form_name'], pokemon_form.form_name)
|
|
self.assertEqual(response.data['order'], pokemon_form.order)
|
|
self.assertEqual(response.data['form_order'], pokemon_form.form_order)
|
|
self.assertEqual(response.data['is_default'], pokemon_form.is_default)
|
|
self.assertEqual(response.data['is_battle_only'], pokemon_form.is_battle_only)
|
|
self.assertEqual(response.data['is_mega'], pokemon_form.is_mega)
|
|
# pokemon params
|
|
self.assertEqual(response.data['pokemon']['name'], pokemon.name)
|
|
self.assertEqual(response.data['pokemon']['url'], '{}{}/pokemon/{}/'.format(test_host, api_v2, pokemon.pk))
|
|
# version group params
|
|
self.assertEqual(response.data['version_group']['name'], pokemon_form.version_group.name)
|
|
self.assertEqual(response.data['version_group']['url'], '{}{}/version-group/{}/'.format(test_host, api_v2, pokemon_form.version_group.pk))
|
|
# sprite params
|
|
self.assertEqual(response.data['sprites']['front_default'], '{}{}'.format(test_host, sprites_data['front_default']))
|
|
self.assertEqual(response.data['sprites']['back_default'], None)
|
|
|
|
|
|
"""
|
|
Evolution Tests
|
|
"""
|
|
|
|
def test_evolution_trigger_api(self):
|
|
|
|
evolution_trigger = self.setup_evolution_trigger_data(name='base evltn trgr')
|
|
evolution_trigger_name = self.setup_evolution_trigger_name_data(evolution_trigger, name='base evltn trgr name')
|
|
pokemon_species = self.setup_pokemon_species_data(name='pkmn spcs for base evltn trgr')
|
|
self.setup_pokemon_evolution_data(evolved_species=pokemon_species, evolution_trigger=evolution_trigger)
|
|
|
|
response = self.client.get('{}/evolution-trigger/{}/'.format(api_v2, evolution_trigger.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], evolution_trigger.pk)
|
|
self.assertEqual(response.data['name'], evolution_trigger.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], evolution_trigger_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], evolution_trigger_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, evolution_trigger_name.language.pk))
|
|
# pokemon species params
|
|
self.assertEqual(response.data['pokemon_species'][0]['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_species'][0]['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
|
|
|
|
|
|
def test_evolution_chain_api(self):
|
|
|
|
baby_trigger_item = self.setup_item_data(name="bby itm for evo chn")
|
|
evolution_chain = self.setup_evolution_chain_data(baby_trigger_item=baby_trigger_item)
|
|
|
|
baby = self.setup_pokemon_species_data(name="bby for evo chn", is_baby=True, evolution_chain=evolution_chain)
|
|
|
|
basic = self.setup_pokemon_species_data(name="bsc for evo chn", evolves_from_species=baby, evolution_chain=evolution_chain)
|
|
basic_location = self.setup_location_data(name='lctn for bsc evo chn')
|
|
basic_evolution = self.setup_pokemon_evolution_data(evolved_species=basic, min_level=5, location=basic_location)
|
|
|
|
stage_one = self.setup_pokemon_species_data(name="stg one for evo chn", evolves_from_species=basic, evolution_chain=evolution_chain)
|
|
stage_one_held_item = self.setup_item_data(name='itm for stg one evo chn')
|
|
stage_one_evolution = self.setup_pokemon_evolution_data(evolved_species=stage_one, min_level=18, held_item=stage_one_held_item)
|
|
|
|
stage_two_first = self.setup_pokemon_species_data(name="stg two frst for evo chn", evolves_from_species=stage_one, evolution_chain=evolution_chain)
|
|
stage_two_first_known_move = self.setup_move_data(name="mv for evo chn")
|
|
stage_two_first_evolution = self.setup_pokemon_evolution_data(evolved_species=stage_two_first, min_level=32, known_move=stage_two_first_known_move,)
|
|
|
|
stage_two_second = self.setup_pokemon_species_data(name="stg two scnd for evo chn", evolves_from_species=stage_one, evolution_chain=evolution_chain)
|
|
stage_two_second_party_type = self.setup_type_data(name="tp for evo chn")
|
|
stage_two_second_evolution = self.setup_pokemon_evolution_data(evolved_species=stage_two_second, min_level=32, party_type=stage_two_second_party_type)
|
|
|
|
response = self.client.get('{}/evolution-chain/{}/'.format(api_v2, evolution_chain.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
baby_data = response.data['chain']
|
|
basic_data = baby_data['evolves_to'][0]
|
|
stage_one_data = basic_data['evolves_to'][0]
|
|
stage_two_first_data = stage_one_data['evolves_to'][0]
|
|
stage_two_second_data = stage_one_data['evolves_to'][1]
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], evolution_chain.pk)
|
|
# baby trigger params
|
|
self.assertEqual(response.data['baby_trigger_item']['name'], baby_trigger_item.name)
|
|
self.assertEqual(response.data['baby_trigger_item']['url'], '{}{}/item/{}/'.format(test_host, api_v2, baby_trigger_item.pk))
|
|
# baby params
|
|
self.assertEqual(baby_data['is_baby'], baby.is_baby)
|
|
self.assertEqual(baby_data['species']['name'], baby.name)
|
|
self.assertEqual(baby_data['species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, baby.pk))
|
|
# basic params
|
|
self.assertEqual(basic_data['is_baby'], basic.is_baby)
|
|
self.assertEqual(basic_data['species']['name'], basic.name)
|
|
self.assertEqual(basic_data['species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, basic.pk))
|
|
self.assertEqual(basic_data['evolution_details']['min_level'], basic_evolution.min_level)
|
|
self.assertEqual(basic_data['evolution_details']['location']['name'], basic_location.name)
|
|
self.assertEqual(basic_data['evolution_details']['location']['url'], '{}{}/location/{}/'.format(test_host, api_v2, basic_location.pk))
|
|
# stage one params
|
|
self.assertEqual(stage_one_data['is_baby'], stage_one.is_baby)
|
|
self.assertEqual(stage_one_data['species']['name'], stage_one.name)
|
|
self.assertEqual(stage_one_data['species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, stage_one.pk))
|
|
self.assertEqual(stage_one_data['evolution_details']['min_level'], stage_one_evolution.min_level)
|
|
self.assertEqual(stage_one_data['evolution_details']['held_item']['name'], stage_one_held_item.name)
|
|
self.assertEqual(stage_one_data['evolution_details']['held_item']['url'], '{}{}/item/{}/'.format(test_host, api_v2, stage_one_held_item.pk))
|
|
# stage two first params
|
|
self.assertEqual(stage_two_first_data['is_baby'], stage_two_first.is_baby)
|
|
self.assertEqual(stage_two_first_data['species']['name'], stage_two_first.name)
|
|
self.assertEqual(stage_two_first_data['species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, stage_two_first.pk))
|
|
self.assertEqual(stage_two_first_data['evolution_details']['min_level'], stage_two_first_evolution.min_level)
|
|
self.assertEqual(stage_two_first_data['evolution_details']['known_move']['name'], stage_two_first_known_move.name)
|
|
self.assertEqual(stage_two_first_data['evolution_details']['known_move']['url'], '{}{}/move/{}/'.format(test_host, api_v2, stage_two_first_known_move.pk))
|
|
# stage two second params
|
|
self.assertEqual(stage_two_second_data['is_baby'], stage_two_second.is_baby)
|
|
self.assertEqual(stage_two_second_data['species']['name'], stage_two_second.name)
|
|
self.assertEqual(stage_two_second_data['species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, stage_two_second.pk))
|
|
self.assertEqual(stage_two_second_data['evolution_details']['min_level'], stage_two_second_evolution.min_level)
|
|
self.assertEqual(stage_two_second_data['evolution_details']['party_type']['name'], stage_two_second_party_type.name)
|
|
self.assertEqual(stage_two_second_data['evolution_details']['party_type']['url'], '{}{}/type/{}/'.format(test_host, api_v2, stage_two_second_party_type.pk))
|
|
|
|
|
|
|
|
"""
|
|
Encounter Tests
|
|
"""
|
|
|
|
def test_encounter_method_api(self):
|
|
|
|
encounter_method = self.setup_encounter_method_data(name='base encntr mthd')
|
|
encounter_method_name = self.setup_encounter_method_name_data(encounter_method, name='base encntr mthd name')
|
|
|
|
response = self.client.get('{}/encounter-method/{}/'.format(api_v2, encounter_method.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], encounter_method.pk)
|
|
self.assertEqual(response.data['name'], encounter_method.name)
|
|
self.assertEqual(response.data['order'], encounter_method.order)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], encounter_method_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], encounter_method_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, encounter_method_name.language.pk))
|
|
|
|
def test_encounter_condition_value_api(self):
|
|
|
|
encounter_condition = self.setup_encounter_condition_data(name='encntr cndtn for base encntr cndtn vlu')
|
|
encounter_condition_value = self.setup_encounter_condition_value_data(encounter_condition, name='base encntr cndtn vlu')
|
|
encounter_condition_value_name = self.setup_encounter_condition_value_name_data(encounter_condition_value, name='base encntr cndtn vlu name')
|
|
|
|
response = self.client.get('{}/encounter-condition-value/{}/'.format(api_v2, encounter_condition_value.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], encounter_condition_value.pk)
|
|
self.assertEqual(response.data['name'], encounter_condition_value.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], encounter_condition_value_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], encounter_condition_value_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, encounter_condition_value_name.language.pk))
|
|
# condition params
|
|
self.assertEqual(response.data['condition']['name'], encounter_condition.name)
|
|
self.assertEqual(response.data['condition']['url'], '{}{}/encounter-condition/{}/'.format(test_host, api_v2, encounter_condition.pk))
|
|
|
|
|
|
def test_encounter_condition_api(self):
|
|
|
|
encounter_condition = self.setup_encounter_condition_data(name='base encntr cndtn')
|
|
encounter_condition_name = self.setup_encounter_condition_name_data(encounter_condition, name='base encntr cndtn name')
|
|
encounter_condition_value = self.setup_encounter_condition_value_data(encounter_condition, name='encntr cndtn vlu for base encntr', is_default=True)
|
|
|
|
response = self.client.get('{}/encounter-condition/{}/'.format(api_v2, encounter_condition.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], encounter_condition.pk)
|
|
self.assertEqual(response.data['name'], encounter_condition.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], encounter_condition_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], encounter_condition_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, encounter_condition_name.language.pk))
|
|
# value params
|
|
self.assertEqual(response.data['values'][0]['name'], encounter_condition_value.name)
|
|
self.assertEqual(response.data['values'][0]['url'], '{}{}/encounter-condition-value/{}/'.format(test_host, api_v2, encounter_condition_value.pk))
|
|
|
|
|
|
"""
|
|
Pal Park Tests
|
|
"""
|
|
|
|
def test_pal_park_area_api(self):
|
|
|
|
pal_park_area = self.setup_pal_park_area_data(name='base pl prk area')
|
|
pal_park_area_name = self.setup_pal_park_area_name_data(pal_park_area, name='base pl prk area nm')
|
|
pokemon_species = self.setup_pokemon_species_data(name='pkmn spcs for pl prk')
|
|
pal_park = self.setup_pal_park_data(pal_park_area=pal_park_area, pokemon_species=pokemon_species, base_score=10, rate=20)
|
|
|
|
response = self.client.get('{}/pal-park-area/{}/'.format(api_v2, pal_park_area.pk))
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
# base params
|
|
self.assertEqual(response.data['id'], pal_park_area.pk)
|
|
self.assertEqual(response.data['name'], pal_park_area.name)
|
|
# name params
|
|
self.assertEqual(response.data['names'][0]['name'], pal_park_area_name.name)
|
|
self.assertEqual(response.data['names'][0]['language']['name'], pal_park_area_name.language.name)
|
|
self.assertEqual(response.data['names'][0]['language']['url'], '{}{}/language/{}/'.format(test_host, api_v2, pal_park_area_name.language.pk))
|
|
# encounter params
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['base_score'], pal_park.base_score)
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['rate'], pal_park.rate)
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['pokemon_species']['name'], pokemon_species.name)
|
|
self.assertEqual(response.data['pokemon_encounters'][0]['pokemon_species']['url'], '{}{}/pokemon-species/{}/'.format(test_host, api_v2, pokemon_species.pk))
|
|
|
|
|