def setUp(self): mongo = MongoDriver(MongoClient(HOST, PORT)[DB_NAME][BUCKET]) self.repo = Repository(mongo) self.mongo_collection = MongoClient(HOST, PORT)[DB_NAME][BUCKET] self.mongo_collection.drop()
def setUp(self): self.mongo = Mock() self.repo = Repository(self.mongo)