Make Pokemon.form an actual relation

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

diff --git a/pokedex/tests/test_lookup.py b/pokedex/tests/test_lookup.py
index 25c3c8a..5d9ff95 100644 (file)
--- a/pokedex/tests/test_lookup.py
+++ b/pokedex/tests/test_lookup.py
@@ -28,7 +28,8 @@ def test_exact_lookup():
 
         # Forms
         (u'Rotom',          'pokemon',      479),
-        (u'Wash Rotom',     'pokemon',      504),
+        (u'Wash Rotom',     'pokemon_forms',10059),
+        (u'East Shellos',   'pokemon_forms',10039),
 
         # Other languages
         (u'イーブイ',       'pokemon',      133),
@@ -146,3 +147,8 @@ def test_random_lookup():
         assert_equal(len(results), 1,           u'Constrained random returns one result')
         assert_equal(results[0].object.__tablename__, table_name,
                                                 u'Constrained random returns result from the right table')
+
+def test_crash_empty_prefix():
+    """Searching for ':foo' used to crash, augh!"""
+    results = lookup.lookup(u':Eevee')
+    assert_equal(results[0].object.name, u'Eevee', u'Empty prefix dun crash')