aboutsummaryrefslogtreecommitdiff
path: root/test/test_providers.py
diff options
context:
space:
mode:
authorgabrielgio <gabriel.giovanini@pm.me>2021-07-18 19:56:59 +0200
committergabrielgio <gabriel.giovanini@pm.me>2021-07-18 19:56:59 +0200
commit10cbc378ad0daf0e80f5ceed92d70fdbf573df88 (patch)
treea4217e75f591632ed383e334ed8e61935cd2b096 /test/test_providers.py
parentb453f05d18c261d3ce3b20bb5aaa2504da562756 (diff)
downloadreddit-nextcloud-importer-10cbc378ad0daf0e80f5ceed92d70fdbf573df88.tar.gz
reddit-nextcloud-importer-10cbc378ad0daf0e80f5ceed92d70fdbf573df88.tar.bz2
reddit-nextcloud-importer-10cbc378ad0daf0e80f5ceed92d70fdbf573df88.zip
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.
Diffstat (limited to 'test/test_providers.py')
-rw-r--r--test/test_providers.py23
1 files changed, 23 insertions, 0 deletions
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']