From 693195f70be3b675757f700e40c6a66a74aa4b50 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bastien=20G=C3=A9rard?= Date: Sat, 31 Aug 2019 22:28:31 +0300 Subject: [PATCH] fix test_pickable that was brought back to life recently --- tests/queryset/test_pickable.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/queryset/test_pickable.py b/tests/queryset/test_pickable.py index fbdd1ff0..8c4e3426 100644 --- a/tests/queryset/test_pickable.py +++ b/tests/queryset/test_pickable.py @@ -3,6 +3,7 @@ import unittest from mongoengine import Document, IntField, StringField from mongoengine.connection import connect +from tests.utils import MongoDBTestCase class Person(Document): @@ -10,7 +11,7 @@ class Person(Document): age = IntField() -class TestQuerysetPickable(unittest.TestCase): +class TestQuerysetPickable(MongoDBTestCase): """ Test for adding pickling support for QuerySet instances See issue https://github.com/MongoEngine/mongoengine/issues/442 @@ -18,8 +19,6 @@ class TestQuerysetPickable(unittest.TestCase): def setUp(self): super(TestQuerysetPickable, self).setUp() - connection = connect(db="test") - connection.drop_database("test") self.john = Person.objects.create(name="John", age=21) def test_picke_simple_qs(self):