refactoring and requirements added

requirements.txt

+pymongo >= 3.2

setup.py

@@ -6,4 +6,8 @@ setup(
     name="common_helper_mongo",
     version=subprocess.check_output(['git', 'describe', '--always'], cwd=os.path.dirname(os.path.abspath(__file__))).strip().decode('utf-8'),
     packages=find_packages(),
+    install_requires=[
+        'pymongo >= 3.2'
+    ],
+    description="MongoDB helper functions"
 )

tests/base_class_database_test.py

+import unittest
+from pymongo import MongoClient
+
+
+class MongoDbTest(unittest.TestCase):
+
+    def setUp(self):
+        self.mongo_client = MongoClient()
+        self.db = self.mongo_client["common_code_test"]
+
+    def tearDown(self):
+        self.mongo_client.drop_database(self.db)
+        self.mongo_client.close()
+
+if __name__ == "__main__":
+    unittest.main()

tests/test_gridfs.py

 import unittest
-from pymongo import MongoClient
 import gridfs
 
 from common_helper_mongo.gridfs import overwrite_file
+from tests.base_class_database_test import MongoDbTest
 
 
-class TestGridFS(unittest.TestCase):
+class TestGridFS(MongoDbTest):
 
     def setUp(self):
-        self.mongo_client = MongoClient()
-        self.db = self.mongo_client["common_code_test"]
+        super().setUp()
         self.fs = gridfs.GridFS(self.db)
 
-    def tearDown(self):
-        self.mongo_client.drop_database(self.db)
-        self.mongo_client.close()
-
     def testOverwriteFile(self):
         self.fs.put(b'original', filename="test_file")
         original_content = self.fs.find_one({'filename': "test_file"}).read()
@@ -27,5 +22,4 @@ class TestGridFS(unittest.TestCase):
 
 
 if __name__ == "__main__":
-    # import sys;sys.argv = ['', 'Test.testName']
     unittest.main()