def test__get_parser_class_n_parsers(self, m, *args): class DummyParser1(object): pass class DummyParser2(object): pass m.return_value = ( (None, { "weight": 0, "parser": DummyParser1, "mime_types": { "application/pdf": ".pdf" } }), (None, { "weight": 1, "parser": DummyParser2, "mime_types": { "application/pdf": ".pdf" } }), ) self.assertEqual(get_parser_class("doc.pdf"), DummyParser2)
def test__get_parser_class_1_parser(self, m, *args): class DummyParser(object): pass m.return_value = ( (None, {"weight": 0, "parser": DummyParser, "mime_types": ["application/pdf"]}), ) self.assertEqual( get_parser_class("doc.pdf"), DummyParser )
def test__get_parser_class_0_parsers(self, m, *args): m.return_value = [] with TemporaryDirectory() as tmpdir: self.assertIsNone(get_parser_class("doc.pdf"))