diff --git a/src/ytdl_sub/entries/base_entry.py b/src/ytdl_sub/entries/base_entry.py index f31bbc6f..88d4e19c 100644 --- a/src/ytdl_sub/entries/base_entry.py +++ b/src/ytdl_sub/entries/base_entry.py @@ -1,7 +1,6 @@ from abc import ABC from typing import Any from typing import Dict -from typing import Optional class BaseEntry(ABC): @@ -9,7 +8,7 @@ class BaseEntry(ABC): Abstract entry object to represent anything download from ytdl (playlist metadata, media, etc). """ - def __init__(self, entry_dict: Dict, working_directory: Optional[str] = None): + def __init__(self, entry_dict: Dict, working_directory: str): """ Initialize the entry using ytdl metadata @@ -38,14 +37,5 @@ class BaseEntry(ABC): Returns ------- The working directory - - Raises - ------ - ValueError - The working directory was never defined in the init """ - if self._working_directory is None: - raise ValueError( - "Entry working directory is not set when trying to access its download file path" - ) return self._working_directory diff --git a/tests/unit/entries/conftest.py b/tests/unit/entries/conftest.py index 31e1a0d8..101aa81f 100644 --- a/tests/unit/entries/conftest.py +++ b/tests/unit/entries/conftest.py @@ -119,7 +119,7 @@ def mock_entry_kwargs( @pytest.fixture def mock_entry(mock_entry_kwargs): - return Entry(entry_dict=mock_entry_kwargs) + return Entry(entry_dict=mock_entry_kwargs, working_directory=".") @pytest.fixture diff --git a/tests/unit/entries/test_soundcloud_entries.py b/tests/unit/entries/test_soundcloud_entries.py index c33d4a51..118b6f3d 100644 --- a/tests/unit/entries/test_soundcloud_entries.py +++ b/tests/unit/entries/test_soundcloud_entries.py @@ -58,7 +58,7 @@ def mock_soundcloud_track_kwargs(mock_entry_kwargs, url): @pytest.fixture def mock_soundcloud_track(mock_soundcloud_track_kwargs): - return SoundcloudTrack(entry_dict=mock_soundcloud_track_kwargs) + return SoundcloudTrack(entry_dict=mock_soundcloud_track_kwargs, working_directory=".") @pytest.fixture diff --git a/tests/unit/entries/test_youtube_entries.py b/tests/unit/entries/test_youtube_entries.py index 5335d619..67f7ea79 100644 --- a/tests/unit/entries/test_youtube_entries.py +++ b/tests/unit/entries/test_youtube_entries.py @@ -66,7 +66,7 @@ def mock_youtube_video_kwargs( @pytest.fixture def mock_youtube_video(mock_youtube_video_kwargs): - return YoutubeVideo(entry_dict=mock_youtube_video_kwargs) + return YoutubeVideo(entry_dict=mock_youtube_video_kwargs, working_directory=".") class TestYoutubeVideo(object):