Add file name formatter class

2021-02-11 09:08:47 +10:00
parent cd2c511db2
commit 358357590f
2 changed files with 114 additions and 0 deletions
--- a/bulkredditdownloader/tests/test_file_name_formatter.py
+++ b/bulkredditdownloader/tests/test_file_name_formatter.py
@@ -0,0 +1,74 @@
+#!/usr/bin/env python3
+# coding=utf-8
+
+from pathlib import Path
+from unittest.mock import Mock
+
+import praw.models
+import pytest
+
+from bulkredditdownloader.file_name_formatter import FileNameFormatter
+from bulkredditdownloader.resource import Resource
+
+
+@pytest.fixture()
+def submission() -> Mock:
+    test = Mock()
+    test.title = 'name'
+    test.subreddit.display_name = 'randomreddit'
+    test.author.name = 'person'
+    test.id = '12345'
+    test.score = 1000
+    test.link_flair_text = 'test_flair'
+    test.created_utc = 123456789
+    return test
+
+
+@pytest.fixture()
+def reddit_submission() -> praw.models.Submission:
+    rd = praw.Reddit(client_id='U-6gk4ZCh3IeNQ', client_secret='7CZHY6AmKweZME5s50SfDGylaPg', user_agent='test')
+    return rd.submission(id='lgilgt')
+
+
+@pytest.mark.parametrize(('format_string', 'expected'), (('{SUBREDDIT}', 'randomreddit'),
+                                                         ('{REDDITOR}', 'person'),
+                                                         ('{POSTID}', '12345'),
+                                                         ('{UPVOTES}', '1000'),
+                                                         ('{FLAIR}', 'test_flair'),
+                                                         ('{DATE}', '123456789'),
+                                                         ('{REDDITOR}_{TITLE}_{POSTID}', 'person_name_12345')
+                                                         ))
+def test_format_name_mock(format_string: str, expected: str, submission: Mock):
+    result = FileNameFormatter._format_name(submission, format_string)
+    assert result == expected
+
+
+@pytest.mark.parametrize(('format_string', 'expected'),
+                         (('{SUBREDDIT}', 'Mindustry'),
+                          ('{REDDITOR}', 'Gamer_player_boi'),
+                          ('{POSTID}', 'lgilgt'),
+                          ('{FLAIR}', 'Art'),
+                          ('{SUBREDDIT}_{TITLE}', 'Mindustry_Toxopid that is NOT humane >:('),
+                          ('{REDDITOR}_{TITLE}_{POSTID}', 'Gamer_player_boi_Toxopid that is NOT humane >:(_lgilgt')
+                          ))
+def test_format_name_real(format_string: str, expected: str, reddit_submission: praw.models.Submission):
+    result = FileNameFormatter._format_name(reddit_submission, format_string)
+    assert result == expected
+
+
+@pytest.mark.parametrize(('format_string_directory', 'format_string_file', 'expected'),
+                         (('{SUBREDDIT}', '{POSTID}', 'test/Mindustry/lgilgt.png'),
+                          ('{SUBREDDIT}', '{TITLE}_{POSTID}',
+                           'test/Mindustry/Toxopid that is NOT humane >:(_lgilgt.png'),
+                          ('{SUBREDDIT}', '{REDDITOR}_{TITLE}_{POSTID}',
+                           'test/Mindustry/Gamer_player_boi_Toxopid that is NOT humane >:(_lgilgt.png')
+                          ))
+def test_format_full(
+        format_string_directory: str,
+        format_string_file: str,
+        expected: str,
+        reddit_submission: praw.models.Submission):
+    test_resource = Resource(reddit_submission, 'i.reddit.com/blabla.png', b'')
+    test_formatter = FileNameFormatter(format_string_file, format_string_directory)
+    result = test_formatter.format_path(test_resource, Path('test'))
+    assert str(result) == expected