Make EmailTestModel globally available in tests

limpyd · Jan 26, 2018 · 9b059b4 · 9b059b4
1 parent c06a2ba
commit 9b059b4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/tests/fields/hash.py b/tests/fields/hash.py
@@ -3,18 +3,12 @@
 
 from limpyd import fields
 from limpyd.exceptions import ImplementationError
-
-from ..model import TestRedisModel, BaseModelTest
-
-
-class EmailTestModel(TestRedisModel):
-    headers = fields.HashField(indexable=True)
-    raw_headers = fields.HashField()
+from ..model import TestRedisModel, BaseModelTest, Email
 
 
 class HashFieldTest(BaseModelTest):
 
-    model = EmailTestModel
+    model = Email
 
     def test_hashfield_can_be_set_at_init(self):
         headers = {

diff --git a/tests/model.py b/tests/model.py
@@ -9,7 +9,7 @@
 import time
 import unittest
 
-from limpyd import model
+from limpyd import model, fields
 from limpyd import fields
 from limpyd.exceptions import *
 
@@ -46,6 +46,10 @@ class Boat(TestRedisModel):
     length = fields.StringField()
 
 
+class Email(TestRedisModel):
+    headers = fields.HashField(indexable=True)
+
+
 class BaseModelTest(LimpydBaseTest):
 
     model = None