aboutsummaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorgabrielgio <gabriel.giovanini@pm.me>2021-07-14 21:23:44 +0200
committergabrielgio <gabriel.giovanini@pm.me>2021-07-14 21:23:44 +0200
commit4d43e402b2b4e27bbbbfe557216d95963a27af72 (patch)
tree46f3013be10a1062280d64052bcb654d67084f91 /test
parent6887fcc0e42b1a64ed80f8565fe3099aaa818930 (diff)
downloadreddit-nextcloud-importer-4d43e402b2b4e27bbbbfe557216d95963a27af72.tar.gz
reddit-nextcloud-importer-4d43e402b2b4e27bbbbfe557216d95963a27af72.tar.bz2
reddit-nextcloud-importer-4d43e402b2b4e27bbbbfe557216d95963a27af72.zip
feat: Add nextcloud path param
Add a new param to input the nextcloud root path.
Diffstat (limited to 'test')
-rw-r--r--test/test_download.py79
-rw-r--r--test/test_uploader.py20
2 files changed, 99 insertions, 0 deletions
diff --git a/test/test_download.py b/test/test_download.py
new file mode 100644
index 0000000..9bafc1c
--- /dev/null
+++ b/test/test_download.py
@@ -0,0 +1,79 @@
+import os
+
+import pytest
+
+from src.downloader import SourceType, Downloader
+
+reddit_env = pytest.mark.skipif(
+ os.environ.get('CLIENT_ID', '') == '' or
+ os.environ.get('CLIENT_SECRET', '') == '' or
+ os.environ.get('USERNAME', '') == '' or
+ os.environ.get('PASSWORD', '') == ''
+ , reason="Require reddit env variables to be set."
+)
+
+
+@pytest.fixture
+def mock_ydl_download(mocker):
+ # this function is responsible for downloading the file
+ return mocker.patch('downloader.youtube_dl.YoutubeDL.process_info')
+
+
+@pytest.mark.parametrize('url,source_type', [
+ ("https://i.redd.it/pjj1ll1b2rr41.jpg", SourceType.IREDDIT),
+ ("https://gfycat.com/presentdangerousdromedary", SourceType.GFYCAT),
+ ("https://i.imgur.com/fXLMjfp.jpg", SourceType.IMAGURJPG),
+ ("https://redgifs.com/watch/ripesnivelingfiddlercrab", SourceType.REDGIFS),
+ ("https://www.youtube.com/watch?v=oLkdqptmfng", SourceType.YOUTUBE),
+ ("https://v.redd.it/42j6r7i8z7151", SourceType.VREDDIT),
+ ("https://www.reddit.com/gallery/mik7c9", SourceType.GREDDIT),
+ ("https://duckduckgo.com", SourceType.UNKNOWN),
+])
+def test_source_type(url, source_type):
+ with Downloader(url, "1-A") as d:
+ assert d.source_type == source_type
+
+
+@pytest.mark.parametrize('url,paths', [
+ ("https://gfycat.com/presentdangerousdromedary", ["source_presentdangerousdromedary.mp4"]),
+ ("https://redgifs.com/watch/ripesnivelingfiddlercrab", ["source_RipeSnivelingFiddlercrab.mp4", 'source_RipeSnivelingFiddlercrab-mobile.mp4']),
+ ("https://www.youtube.com/watch?v=oLkdqptmfng", ["source_oLkdqptmfng.mp4"]),
+ ("https://v.redd.it/42j6r7i8z7151", ["source_42j6r7i8z7151.mp4"]),
+])
+def test_download_youtube_dl(url, paths, mock_ydl_download):
+ with Downloader(url, "1-A") as d:
+ assert d.downloaded is False
+ d.download()
+ assert d.downloaded is True
+ assert d.paths == paths
+ mock_ydl_download.assert_called()
+
+
+@pytest.mark.parametrize('url,path', [
+ ("https://i.redd.it/pjj1ll1b2rr41.jpg", "source_pjj1ll1b2rr41.jpg"),
+ ("https://i.imgur.com/fXLMjfp.jpg", "source_fXLMjfp.jpg"),
+])
+def test_download_raw_data(url, path):
+ with Downloader(url, "1-A") as d:
+ assert d.downloaded is False
+ d.download()
+ assert d.paths == [path]
+ assert d.downloaded is True
+
+
+@reddit_env
+def test_praw_download():
+ client_id = os.environ.get('CLIENT_ID', '')
+ client_secret = os.environ.get('CLIENT_SECRET', '')
+ username = os.environ.get('USERNAME', '')
+ password = os.environ.get('PASSWORD', '')
+ files = {'source_hlokpsyhgrq61.jpg', 'source_n31c2y7igrq61.jpg', 'source_7eg0o76igrq61.jpg',
+ 'source_whl12jbigrq61.jpg', 'source_4uok762igrq61.jpg', 'source_t3pgm64igrq61.jpg',
+ 'source_ymc4hv9igrq61.jpg'}
+
+ with Downloader("https://www.reddit.com/gallery/mik7c9", "1-A", client_id=client_id, client_secret=client_secret,
+ password=password, user_agent="hcrawler", username=username) as d:
+ assert d.downloaded is False
+ d.download()
+ assert d.downloaded is True
+ assert set(d.paths) == files
diff --git a/test/test_uploader.py b/test/test_uploader.py
new file mode 100644
index 0000000..bcdf04b
--- /dev/null
+++ b/test/test_uploader.py
@@ -0,0 +1,20 @@
+import pytest
+
+from uploader import create_folders
+
+
+@pytest.fixture
+def nxc(mocker):
+ return mocker.MagicMock()
+
+
+@pytest.mark.parametrize("path,folders", [
+ ("folder1", ["/folder1"]),
+ ("folder1/folder2", ["/folder1", "/folder1/folder2"]),
+ ("/folder1/folder2", ["/folder1", "/folder1/folder2"])
+])
+def test_create_folders(nxc, path, folders):
+ create_folders(path, nxc)
+
+ for call, folder in zip(nxc.method_calls, folders):
+ assert folder == call.args[0]