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"))
