Ejemplo n.º 1
0
    def create_mongo(self):
        # def insert_one(self, *args, **kwargs):
        #     x = 0
        #     while True:
        #         x += 1
        #         yield x

        mongo_client = MagicMock()
        mongo_client.tpbmeta = MagicMock()
        mongo_client.tpbmeta.pages.insert_one = MagicMock()
        mongo_client.tpbmeta.pages.insert_one.return_value = 'test_value'

        mongo = Mongo(mongo_client)
        return mongo