From 21bd3641d69a8558fd75195a76ead8f07101768a Mon Sep 17 00:00:00 2001 From: Cody Logan Date: Tue, 28 Jan 2020 13:50:01 -0800 Subject: Split main file into separate chunks --- test/test_validations.py | 105 +++++++++++++++++++++++++++++++++++++++++++++++ test/test_wikiget.py | 105 ----------------------------------------------- 2 files changed, 105 insertions(+), 105 deletions(-) create mode 100644 test/test_validations.py delete mode 100644 test/test_wikiget.py (limited to 'test') diff --git a/test/test_validations.py b/test/test_validations.py new file mode 100644 index 0000000..a0e2628 --- /dev/null +++ b/test/test_validations.py @@ -0,0 +1,105 @@ +# -*- coding: utf-8 -*- +# wikiget - CLI tool for downloading files from Wikimedia sites +# Copyright (C) 2018, 2019 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 . + +"""Defines test functions to run with pytest.""" + +import os + +import pytest + +from wikiget.validations import valid_file, valid_site, verify_hash + + +def test_invalid_site_input(): + """ + Invalid site strings should not return regex match objects. + """ + invalid_input = ['example.com', 'vim.wikia.com', + 'en.wikipedia.com', 'en.wikimpedia.org'] + for i in invalid_input: + site_match = valid_site(i) + assert site_match is None + + +def test_valid_site_input(): + """ + Valid site strings should return regex match objects. + """ + valid_input = ['en.wikipedia.org', 'commons.wikimedia.org', + 'de.wikipedia.org', 'meta.wikimedia.org'] + for i in valid_input: + site_match = valid_site(i) + assert site_match is not None + + +def test_file_regex(): + """ + File regex should return a match object with match groups corresponding + to the file prefix and name. + """ + i = 'File:Example.jpg' + file_match = valid_file(i) + assert file_match is not None + assert file_match.group(0) == 'File:Example.jpg' # entire match + assert file_match.group(1) == 'File:' # first group + assert file_match.group(2) == 'Example.jpg' # second group + + +def test_invalid_file_input(): + """ + Invalid file strings should not return regex match objects. + """ + invalid_input = ['file:example', 'example.jpg', 'Foo Bar.gif', + 'Fil:Example.jpg'] + for i in invalid_input: + file_match = valid_file(i) + assert file_match is None + + +def test_valid_file_input(): + """ + Valid file strings should return regex match objects. + """ + valid_input = ['Image:example.jpg', 'file:example.jpg', + 'File:example.file-01.jpg', 'FILE:FOO.BMP', + 'File:ß handwritten sample.gif', 'File:A (1).jpeg'] + for i in valid_input: + file_match = valid_file(i) + assert file_match is not None + + +def test_verify_hash(): + """ + Confirm that verify_hash returns the proper SHA1 hash. + """ + # TODO: do we need to actually create a file? + file_name = 'testfile' + file_contents = 'foobar' + file_sha1 = '8843d7f92416211de9ebb963ff4ce28125932878' + + try: + dl = open(file_name, 'w') + except PermissionError: + pytest.skip('need write access to create test file') + else: + with dl: + dl.write(file_contents) + + assert verify_hash(file_name) == file_sha1 + + os.remove(file_name) diff --git a/test/test_wikiget.py b/test/test_wikiget.py deleted file mode 100644 index cdc0706..0000000 --- a/test/test_wikiget.py +++ /dev/null @@ -1,105 +0,0 @@ -# -*- coding: utf-8 -*- -# wikiget - CLI tool for downloading files from Wikimedia sites -# Copyright (C) 2018, 2019 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 . - -"""Defines test functions to run with pytest.""" - -import os - -import pytest - -from wikiget import wikiget - - -def test_invalid_site_input(): - """ - Invalid site strings should not return regex match objects. - """ - invalid_input = ['example.com', 'vim.wikia.com', - 'en.wikipedia.com', 'en.wikimpedia.org'] - for i in invalid_input: - site_match = wikiget.valid_site(i) - assert site_match is None - - -def test_valid_site_input(): - """ - Valid site strings should return regex match objects. - """ - valid_input = ['en.wikipedia.org', 'commons.wikimedia.org', - 'de.wikipedia.org', 'meta.wikimedia.org'] - for i in valid_input: - site_match = wikiget.valid_site(i) - assert site_match is not None - - -def test_file_regex(): - """ - File regex should return a match object with match groups corresponding - to the file prefix and name. - """ - i = 'File:Example.jpg' - file_match = wikiget.valid_file(i) - assert file_match is not None - assert file_match.group(0) == 'File:Example.jpg' # entire match - assert file_match.group(1) == 'File:' # first group - assert file_match.group(2) == 'Example.jpg' # second group - - -def test_invalid_file_input(): - """ - Invalid file strings should not return regex match objects. - """ - invalid_input = ['file:example', 'example.jpg', 'Foo Bar.gif', - 'Fil:Example.jpg'] - for i in invalid_input: - file_match = wikiget.valid_file(i) - assert file_match is None - - -def test_valid_file_input(): - """ - Valid file strings should return regex match objects. - """ - valid_input = ['Image:example.jpg', 'file:example.jpg', - 'File:example.file-01.jpg', 'FILE:FOO.BMP', - 'File:ß handwritten sample.gif', 'File:A (1).jpeg'] - for i in valid_input: - file_match = wikiget.valid_file(i) - assert file_match is not None - - -def test_verify_hash(): - """ - Confirm that verify_hash returns the proper SHA1 hash. - """ - # TODO: do we need to actually create a file? - file_name = 'testfile' - file_contents = 'foobar' - file_sha1 = '8843d7f92416211de9ebb963ff4ce28125932878' - - try: - dl = open(file_name, 'w') - except PermissionError: - pytest.skip('need write access to create test file') - else: - with dl: - dl.write(file_contents) - - assert wikiget.verify_hash(file_name) == file_sha1 - - os.remove(file_name) -- cgit v1.2.3