From 10cbc378ad0daf0e80f5ceed92d70fdbf573df88 Mon Sep 17 00:00:00 2001
From: gabrielgio <gabriel.giovanini@pm.me>
Date: Sun, 18 Jul 2021 19:56:59 +0200
Subject: ref: Move to OO implementation

Heavily inspired by the `youtube-dl` implementation I moved to OO
implementation where now every source type has its own class, making
easy to add new providers.

Also new it has a fallback back, where if no provider is chose it will
try to download with `YoutubeDlProvideBase`.

Add `_TEST` to each class to make it easy to add test to new providers.
---
 test/test_providers.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 test/test_providers.py

(limited to 'test/test_providers.py')

diff --git a/test/test_providers.py b/test/test_providers.py
new file mode 100644
index 0000000..9a5084e
--- /dev/null
+++ b/test/test_providers.py
@@ -0,0 +1,23 @@
+import praw
+import pytest
+
+from importer.downloader import Downloader
+import importer.providers as providers
+from importer.providers import ProviderBase
+
+
+@pytest.mark.parametrize("provider",
+                         [
+                             providers.IReddit,
+                             providers.Imgur,
+                             providers.RawImageProviderBase,
+                             providers.RedGifs,
+                             providers.Youtube,
+                             providers.YoutubeDlProviderBase
+                         ])
+def test_provider(provider):
+    for test in provider._TEST:
+        with provider(url=test['url']) as p:
+            p.download()
+            assert p.downloaded
+            assert p.paths == test['paths']
-- 
cgit v1.2.3