Added tests for mime parsers.

tests/test_mime_parsers.py

@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+import pytest
+
+from rtv.mime_parsers import parsers
+
+URLS = [
+    ('http://www.example.com/i/image.png',
+     'http://www.example.com/i/image.png', 'image/png'),
+    ('http://www.example.com/v/video.mpeg',
+     'http://www.example.com/v/video.mpeg', 'video/mpeg'),
+    ('http://www.example.com/i/image',
+     'http://www.example.com/i/image', None),
+    ('https://gfycat.com/DeliciousUnfortunateAdouri',
+     'https://giant.gfycat.com/DeliciousUnfortunateAdouri.webm', 'video/webm'),
+    ('https://www.youtube.com/watch?v=FjNdYp2gXRY',
+     'https://www.youtube.com/watch?v=FjNdYp2gXRY', 'video/x-youtube'),
+    ('http://i.imgur.com/i/image.gifv',
+     'http://i.imgur.com/i/image.mp4', 'video/mp4'),
+    ('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'),
+    ('http://imgur.com/yW0kbMi',
+     'https://i.imgur.com/yW0kbMi.jpg', 'image/jpeg'),
+    ('http://imgur.com/yjP1v4B',
+     'https://i.imgur.com/yjP1v4Bh.jpg', 'image/jpeg'),
+    ('http://imgur.com/a/qx9t5',
+     'http://i.imgur.com/uEt0YLI.jpg', 'image/x-imgur-album'),
+]
+
+
+@pytest.mark.parametrize('url,modified_url,mime_type', URLS)
+def test_parser(url, modified_url, mime_type, reddit):
+    # Add the reddit fixture so the cassettes get generated
+
+    for parser in parsers:
+        if parser.pattern.match(url):
+            assert parser.get_mimetype(url) == (modified_url, mime_type)
+            break
+    else:
+        # The base parser should catch all urls before this point
+        assert False