From 65de3a782aee3818a9fd927a087647be4880aaad Mon Sep 17 00:00:00 2001 From: Bradlee Speice Date: Wed, 24 Aug 2016 16:09:54 -0400 Subject: [PATCH] Fix one last import --- test/targets/test_mongo_target.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/test/targets/test_mongo_target.py b/test/targets/test_mongo_target.py index faa50f5..9c00ea0 100644 --- a/test/targets/test_mongo_target.py +++ b/test/targets/test_mongo_target.py @@ -2,7 +2,7 @@ from pymongo import MongoClient from random import randint from metrik.targets.mongo import MongoTarget -from metrik.conf import MONGO_DATABASE, MONGO_HOST, MONGO_PORT +from metrik.conf import get_config from test.mongo_test import MongoTest @@ -17,8 +17,7 @@ class MongoTargetTest(MongoTest): t.persist({'a': 'b'}) assert t.exists() - db = MongoClient(MONGO_HOST, MONGO_PORT)[MONGO_DATABASE] - db_collection = db[collection] + db_collection = self.db[collection] db_collection.remove(id) assert not t.exists()