Added parameter ids to tests.

This commit is contained in:
Michael Lazar
2016-07-21 00:40:51 -07:00
parent 738a46e6dd
commit d3f00cdbc6

View File

@@ -5,35 +5,56 @@ import pytest
from rtv.mime_parsers import parsers from rtv.mime_parsers import parsers
URLS = [ ARGS = 'url,modified_url,mime_type'
('http://www.example.com/i/image.png', URLS = {
'http://www.example.com/i/image.png', 'image/png'), 'simple_png': (
('http://www.example.com/v/video.mpeg', 'http://www.example.com/i/image.png',
'http://www.example.com/v/video.mpeg', 'video/mpeg'), 'http://www.example.com/i/image.png',
('http://www.example.com/i/image', 'image/png'),
'http://www.example.com/i/image', None), 'simple_mpeg': (
('https://gfycat.com/DeliciousUnfortunateAdouri', 'http://www.example.com/v/video.mpeg',
'https://giant.gfycat.com/DeliciousUnfortunateAdouri.webm', 'video/webm'), 'http://www.example.com/v/video.mpeg',
('https://www.youtube.com/watch?v=FjNdYp2gXRY', 'video/mpeg'),
'https://www.youtube.com/watch?v=FjNdYp2gXRY', 'video/x-youtube'), 'simple_unknown': (
('http://i.imgur.com/i/image.gifv', 'http://www.example.com/i/image',
'http://i.imgur.com/i/image.mp4', 'video/mp4'), 'http://www.example.com/i/image',
('https://i.reddituploads.com/a065472e47a4405da159189ee48bff46?fit=max&h=' None),
'1536&w=1536&s=5639918a0c696b9bb3ec694dc3cf59ac', 'gfycat': (
'https://i.reddituploads.com/a065472e47a4405da159189ee48bff46?fit=max&h=' 'https://gfycat.com/DeliciousUnfortunateAdouri',
'1536&w=1536&s=5639918a0c696b9bb3ec694dc3cf59ac', 'image/jpeg'), 'https://giant.gfycat.com/DeliciousUnfortunateAdouri.webm',
('http://imgur.com/yW0kbMi', 'video/webm'),
'https://i.imgur.com/yW0kbMi.jpg', 'image/jpeg'), 'youtube': (
('http://imgur.com/yjP1v4B', 'https://www.youtube.com/watch?v=FjNdYp2gXRY',
'https://i.imgur.com/yjP1v4Bh.jpg', 'image/jpeg'), 'https://www.youtube.com/watch?v=FjNdYp2gXRY',
('http://imgur.com/a/qx9t5', 'video/x-youtube'),
'http://i.imgur.com/uEt0YLI.jpg', 'image/x-imgur-album'), 'gifv': (
] 'http://i.imgur.com/i/image.gifv',
'http://i.imgur.com/i/image.mp4',
'video/mp4'),
'reddit_uploads': (
'https://i.reddituploads.com/a065472e47a4405da159189ee48bff46?fit=max'
'&h=1536&w=1536&s=5639918a0c696b9bb3ec694dc3cf59ac',
'https://i.reddituploads.com/a065472e47a4405da159189ee48bff46?fit=max'
'&h=1536&w=1536&s=5639918a0c696b9bb3ec694dc3cf59ac',
'image/jpeg'),
'imgur_1': (
'http://imgur.com/yW0kbMi',
'https://i.imgur.com/yW0kbMi.jpg',
'image/jpeg'),
'imgur_2': (
'http://imgur.com/yjP1v4B',
'https://i.imgur.com/yjP1v4Bh.jpg',
'image/jpeg'),
'imgur_album': (
'http://imgur.com/a/qx9t5',
'http://i.imgur.com/uEt0YLI.jpg',
'image/x-imgur-album'),
}
@pytest.mark.parametrize('url,modified_url,mime_type', URLS) @pytest.mark.parametrize(ARGS, URLS.values(), ids=URLS.keys())
def test_parser(url, modified_url, mime_type, reddit): def test_parser(url, modified_url, mime_type, reddit):
# Add the reddit fixture so the cassettes get generated # Include the reddit fixture so the cassettes get generated
for parser in parsers: for parser in parsers:
if parser.pattern.match(url): if parser.pattern.match(url):