diff options
| author | Cody Logan <clpo13@gmail.com> | 2020-01-28 13:50:01 -0800 |
|---|---|---|
| committer | Cody Logan <clpo13@gmail.com> | 2020-01-28 13:50:01 -0800 |
| commit | 21bd3641d69a8558fd75195a76ead8f07101768a (patch) | |
| tree | 6ee9a9c4230f993c9db58212ff5dfdc5b73b8d35 /test/test_wikiget.py | |
| parent | 71f9b13025882512f150d9b167c4dc7f91ab3b5c (diff) | |
| download | wikiget-21bd3641d69a8558fd75195a76ead8f07101768a.tar.gz wikiget-21bd3641d69a8558fd75195a76ead8f07101768a.zip | |
Split main file into separate chunks
Diffstat (limited to 'test/test_wikiget.py')
| -rw-r--r-- | test/test_wikiget.py | 105 |
1 files changed, 0 insertions, 105 deletions
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 <https://www.gnu.org/licenses/>. - -"""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) |
