Lots of encounters were using slots that claimed to be for other games. #297 #281

[zzz-pokedex.git] / pokedex / tests / test_database_sanity.py

diff --git a/pokedex/tests/test_database_sanity.py b/pokedex/tests/test_database_sanity.py
new file mode 100644 (file)
index 0000000..2fff6d7
--- /dev/null
+++ b/pokedex/tests/test_database_sanity.py
@@ -0,0 +1,24 @@
+from nose.tools import *
+import unittest
+from sqlalchemy.orm import aliased
+
+from pokedex.db import connect, tables
+
+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)
+
+    assert_equal(sanity_q.count(), 0,
+        "Encounter slots all match the encounters they belong to")