def test_get_parser_for_type_no_parser_error(parser_class_map): parser_class_map.get.return_value = None with pytest.raises(NoParserError) as error: get_parser_for_type('foo') parser_class_map.get.assert_called_once_with('foo') assert error.value.mimetype == 'foo' assert error.value.mimetypes == set()
def test_get_parser_for_type(parser_class_map): class_ = MagicMock() parser_class_map.get.return_value = class_ result = get_parser_for_type('foo') parser_class_map.get.assert_called_once_with('foo') assert result == class_()