1
0
mirror of https://github.com/bspeice/metrik synced 2024-12-25 23:18:15 -05:00

Fix one last import

This commit is contained in:
Bradlee Speice 2016-08-24 16:09:54 -04:00
parent af847c16c3
commit 65de3a782a

View File

@ -2,7 +2,7 @@ from pymongo import MongoClient
from random import randint from random import randint
from metrik.targets.mongo import MongoTarget 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 from test.mongo_test import MongoTest
@ -17,8 +17,7 @@ class MongoTargetTest(MongoTest):
t.persist({'a': 'b'}) t.persist({'a': 'b'})
assert t.exists() assert t.exists()
db = MongoClient(MONGO_HOST, MONGO_PORT)[MONGO_DATABASE] db_collection = self.db[collection]
db_collection = db[collection]
db_collection.remove(id) db_collection.remove(id)
assert not t.exists() assert not t.exists()