From 865088207b39427b6b932de4f312d82bd5e05a53 Mon Sep 17 00:00:00 2001 From: Cody Logan Date: Tue, 3 Oct 2023 13:26:09 -0700 Subject: Refactor for better code organization --- tests/test_file_class.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 tests/test_file_class.py (limited to 'tests/test_file_class.py') diff --git a/tests/test_file_class.py b/tests/test_file_class.py new file mode 100644 index 0000000..7ad0b87 --- /dev/null +++ b/tests/test_file_class.py @@ -0,0 +1,31 @@ +# wikiget - CLI tool for downloading files from Wikimedia sites +# Copyright (C) 2023 Cody Logan +# SPDX-License-Identifier: GPL-3.0-or-later +# +# Wikiget is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Wikiget is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with Wikiget. If not, see . + +from wikiget.file import File + + +def test_file_with_name_only(): + file = File("foobar.jpg") + assert file.name == "foobar.jpg" + assert file.dest == file.name + + +def test_file_with_name_and_dest(): + file = File("foobar.jpg", "bazqux.jpg") + assert file.name == "foobar.jpg" + assert file.dest == "bazqux.jpg" + assert file.dest != file.name -- cgit v1.2.3 From 05457af0d73ff3a820c0b465e6607fc5832a6e74 Mon Sep 17 00:00:00 2001 From: Cody Logan Date: Fri, 20 Oct 2023 16:23:28 -0700 Subject: Reorganize File class --- tests/test_file_class.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'tests/test_file_class.py') diff --git a/tests/test_file_class.py b/tests/test_file_class.py index 7ad0b87..dd30207 100644 --- a/tests/test_file_class.py +++ b/tests/test_file_class.py @@ -15,6 +15,7 @@ # You should have received a copy of the GNU General Public License # along with Wikiget. If not, see . +from wikiget import DEFAULT_SITE from wikiget.file import File @@ -22,10 +23,15 @@ def test_file_with_name_only(): file = File("foobar.jpg") assert file.name == "foobar.jpg" assert file.dest == file.name + assert file.site == DEFAULT_SITE def test_file_with_name_and_dest(): - file = File("foobar.jpg", "bazqux.jpg") - assert file.name == "foobar.jpg" + file = File("foobar.jpg", dest="bazqux.jpg") assert file.dest == "bazqux.jpg" assert file.dest != file.name + + +def test_file_with_name_and_site(): + file = File("foobar.jpg", site="en.wikipedia.org") + assert file.site == "en.wikipedia.org" -- cgit v1.2.3