mirror of
https://github.com/veekun/pokedex.git
synced 2024-07-03 02:55:49 +00:00
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
|
|
import pytest
|
|
|
|
from sqlalchemy.orm import aliased, joinedload
|
|
from sqlalchemy.orm.exc import NoResultFound
|
|
|
|
from pokedex.db import connect, tables, util
|
|
|
|
def test_encounter_slots():
|
|
# Encounters have a version, which has a version group; encounters also
|
|
# have an encounter_slot, which has a version group. The two version
|
|
# groups should match, universally.
|
|
session = connect()
|
|
|
|
version_group_a = aliased(tables.VersionGroup)
|
|
version_group_b = aliased(tables.VersionGroup)
|
|
|
|
sanity_q = session.query(tables.Encounter) \
|
|
.join((tables.EncounterSlot, tables.Encounter.slot)) \
|
|
.join((version_group_a, tables.EncounterSlot.version_group)) \
|
|
.join((tables.Version, tables.Encounter.version)) \
|
|
.join((version_group_b, tables.Version.version_group)) \
|
|
.filter(version_group_a.id != version_group_b.id)
|
|
|
|
# Encounter slots all match the encounters they belong to
|
|
assert sanity_q.count() == 0
|
|
|
|
def test_nonzero_autoincrement_ids():
|
|
"""Check that autoincrementing ids don't contain zeroes
|
|
|
|
MySQL doesn't like these, see e.g. bug #580
|
|
"""
|
|
|
|
session = connect()
|
|
for cls in tables.mapped_classes:
|
|
if 'id' in cls.__table__.c:
|
|
if cls.__table__.c.id.autoincrement:
|
|
def nonzero_id(cls):
|
|
with pytest.raises(NoResultFound):
|
|
util.get(session, cls, id=0)
|
|
nonzero_id.description = "No zero id in %s" % cls.__name__
|
|
yield nonzero_id, cls
|
|
|
|
def test_unique_form_order():
|
|
"""Check that tone PokemonForm.order value isn't used for more species
|
|
"""
|
|
|
|
session = connect()
|
|
|
|
species_by_form_order = {}
|
|
|
|
query = session.query(tables.PokemonForm)
|
|
query = query.options(joinedload('pokemon.species'))
|
|
|
|
for form in query:
|
|
print form.name
|
|
try:
|
|
previous_species = species_by_form_order[form.order]
|
|
except KeyError:
|
|
species_by_form_order[form.order] = form.species
|
|
else:
|
|
assert previous_species == form.species, (
|
|
"PokemonForm.order == %s is used for %s and %s" % (
|
|
form.order,
|
|
species_by_form_order[form.order].name,
|
|
form.species.name))
|