allow to use sets in field choices (#1482)

2017-02-15 14:51:47 +01:00
parent ed34c2ca68
commit b27c7ce11b
6 changed files with 68 additions and 35 deletions
--- a/tests/fields/file_tests.py
+++ b/tests/fields/file_tests.py
@@ -18,15 +18,13 @@ try:
 except ImportError:
    HAS_PIL = False

+from tests.utils import MongoDBTestCase
+
 TEST_IMAGE_PATH = os.path.join(os.path.dirname(__file__), 'mongoengine.png')
 TEST_IMAGE2_PATH = os.path.join(os.path.dirname(__file__), 'mongodb_leaf.png')


-class FileTest(unittest.TestCase):
-
-    def setUp(self):
-        connect(db='mongoenginetest')
-        self.db = get_db()
+class FileTest(MongoDBTestCase):

    def tearDown(self):
        self.db.drop_collection('fs.files')