test_mongo_pipeline.py 文件源码

def test_process_item(self):
        normal_item = Item()

        class DummyDocument(Document):
            pass
        DummyDocument.save = MagicMock()
        document_item = document_to_item(DummyDocument)()

        after = self.pipe.process_item(normal_item, None)
        self.assertEqual(normal_item, after)
        after = self.pipe.process_item(document_item, None)
        self.assertIsInstance(after, DummyDocument)