From 2267b7e7d740409dd1b9f648d1fe5b9e1cfdb7a4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bastien=20G=C3=A9rard?= <bastien.gerard@cluepoints.com>
Date: Fri, 30 Aug 2019 16:27:56 +0300
Subject: [PATCH] rename remaining files for pytest migration

---
 tests/queryset/{modify.py => test_modify.py}       |  6 ++----
 tests/queryset/{pickable.py => test_pickable.py}   | 11 +++--------
 tests/queryset/{transform.py => test_transform.py} |  4 +---
 tests/queryset/{visitor.py => test_visitor.py}     |  4 +---
 4 files changed, 7 insertions(+), 18 deletions(-)
 rename tests/queryset/{modify.py => test_modify.py} (96%)
 rename tests/queryset/{pickable.py => test_pickable.py} (87%)
 rename tests/queryset/{transform.py => test_transform.py} (99%)
 rename tests/queryset/{visitor.py => test_visitor.py} (99%)

diff --git a/tests/queryset/modify.py b/tests/queryset/test_modify.py
similarity index 96%
rename from tests/queryset/modify.py
rename to tests/queryset/test_modify.py
index e092d11c..60f4884c 100644
--- a/tests/queryset/modify.py
+++ b/tests/queryset/test_modify.py
@@ -1,8 +1,6 @@
 import unittest
 
-from mongoengine import connect, Document, IntField, StringField, ListField
-
-__all__ = ("FindAndModifyTest",)
+from mongoengine import Document, IntField, ListField, StringField, connect
 
 
 class Doc(Document):
@@ -10,7 +8,7 @@ class Doc(Document):
     value = IntField()
 
 
-class FindAndModifyTest(unittest.TestCase):
+class TestFindAndModify(unittest.TestCase):
     def setUp(self):
         connect(db="mongoenginetest")
         Doc.drop_collection()
diff --git a/tests/queryset/pickable.py b/tests/queryset/test_pickable.py
similarity index 87%
rename from tests/queryset/pickable.py
rename to tests/queryset/test_pickable.py
index 0945fcbc..fbdd1ff0 100644
--- a/tests/queryset/pickable.py
+++ b/tests/queryset/test_pickable.py
@@ -1,10 +1,8 @@
 import pickle
 import unittest
-from pymongo.mongo_client import MongoClient
-from mongoengine import Document, StringField, IntField
-from mongoengine.connection import connect
 
-__author__ = "stas"
+from mongoengine import Document, IntField, StringField
+from mongoengine.connection import connect
 
 
 class Person(Document):
@@ -20,11 +18,8 @@ class TestQuerysetPickable(unittest.TestCase):
 
     def setUp(self):
         super(TestQuerysetPickable, self).setUp()
-
-        connection = connect(db="test")  # type: pymongo.mongo_client.MongoClient
-
+        connection = connect(db="test")
         connection.drop_database("test")
-
         self.john = Person.objects.create(name="John", age=21)
 
     def test_picke_simple_qs(self):
diff --git a/tests/queryset/transform.py b/tests/queryset/test_transform.py
similarity index 99%
rename from tests/queryset/transform.py
rename to tests/queryset/test_transform.py
index cfcd8c22..8207351d 100644
--- a/tests/queryset/transform.py
+++ b/tests/queryset/test_transform.py
@@ -5,10 +5,8 @@ from bson.son import SON
 from mongoengine import *
 from mongoengine.queryset import Q, transform
 
-__all__ = ("TransformTest",)
 
-
-class TransformTest(unittest.TestCase):
+class TestTransform(unittest.TestCase):
     def setUp(self):
         connect(db="mongoenginetest")
 
diff --git a/tests/queryset/visitor.py b/tests/queryset/test_visitor.py
similarity index 99%
rename from tests/queryset/visitor.py
rename to tests/queryset/test_visitor.py
index 0a22416f..acadabd4 100644
--- a/tests/queryset/visitor.py
+++ b/tests/queryset/test_visitor.py
@@ -8,10 +8,8 @@ from mongoengine import *
 from mongoengine.errors import InvalidQueryError
 from mongoengine.queryset import Q
 
-__all__ = ("QTest",)
 
-
-class QTest(unittest.TestCase):
+class TestQ(unittest.TestCase):
     def setUp(self):
         connect(db="mongoenginetest")