예제 #1
0
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()
예제 #2
0
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_()