1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
# -*- coding: utf-8 -*-
# wikiget - simple CLI tool for downloading files from Wikimedia sites.
# Copyright (C) 2018-2019 Cody Logan; licensed GPLv3+
# SPDX-License-Identifier: GPL-3.0-or-later
"""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.
:return:
"""
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:
with open(file_name, "w") as dl:
dl.write(file_contents)
except PermissionError:
pytest.skip("need write access to create test file")
assert wikiget.verify_hash(file_name) == file_sha1
os.remove(file_name)
|
