mirror of
https://github.com/bspeice/metrik
synced 2024-11-05 06:58:12 -05:00
34 lines
917 B
Python
34 lines
917 B
Python
|
from pymongo import MongoClient
|
||
|
|
from random import randint
|
||
|
|
|
||
|
|
from metrik.targets.mongo_target import MongoTarget
|
||
|
|
from metrik.conf import MONGO_DATABASE, MONGO_HOST, MONGO_PORT
|
||
|
|
from test.mongo_test import MongoTest
|
||
|
|
|
||
|
|
|
||
|
|
class MongoTargetTest(MongoTest):
|
||
|
|
|
||
|
|
def test_exists(self):
|
||
|
|
collection = 'test_collection'
|
||
|
|
id = 1234
|
||
|
|
t = MongoTarget(collection, id)
|
||
|
|
assert not t.exists()
|
||
|
|
|
||
|
|
t.persist({'a': 'b'})
|
||
|
|
assert t.exists()
|
||
|
|
|
||
|
|
db = MongoClient(MONGO_HOST, MONGO_PORT)[MONGO_DATABASE]
|
||
|
|
db_collection = db[collection]
|
||
|
|
db_collection.remove(id)
|
||
|
|
assert not t.exists()
|
||
|
|
|
||
|
|
def test_persist_retrieve(self):
|
||
|
|
collection = 'test_collection'
|
||
|
|
id = 1234
|
||
|
|
t = MongoTarget(collection, id)
|
||
|
|
|
||
|
|
d = {str(k): randint(0, 9999) for k in range(20)}
|
||
|
|
t.persist(d)
|
||
|
|
|
||
|
|
u = MongoTarget(collection, id)
|
||
|
|
assert u.retrieve() == d
|