def test_mongo_observer_equality(mongo_obs):
runs = mongo_obs.runs
fs = mock.MagicMock()
m = MongoObserver.create_from(runs, fs)
assert mongo_obs == m
assert not mongo_obs != m
assert not mongo_obs == "foo"
assert mongo_obs != "foo"
def failing_mongo_observer():
db = FailingMongoClient(
max_calls_before_failure=2,
# exception_to_raise=pymongo.errors.AutoReconnect
exception_to_raise=pymongo.errors.ServerSelectionTimeoutError,
).db
runs = db.runs
metrics = db.metrics
fs = mock.MagicMock()
return MongoObserver.create_from(runs, fs, metrics_collection=metrics)
def failing_mongo_observer():
db = FailingMongoClient(
max_calls_before_failure=2,
exception_to_raise=pymongo.errors.ServerSelectionTimeoutError,
).db
runs = db.runs
metrics = db.metrics
fs = gridfs.GridFS(db)
return MongoObserver.create_from(runs, fs, metrics_collection=metrics)
def mongo_obs():
db = mongomock.MongoClient().db
runs = db.runs
metrics = db.metrics
fs = mock.MagicMock()
return MongoObserver.create_from(runs, fs, metrics_collection=metrics)
def mongo_obs():
db = mongomock.MongoClient().db
runs = db.runs
metrics = db.metrics
fs = gridfs.GridFS(db)
return MongoObserver.create_from(runs, fs, metrics_collection=metrics)
