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 - CLI tool for downloading files from Wikimedia sites
# Copyright (C) 2018, 2019, 2020 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/>.
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(tmp_path):
"""
Confirm that verify_hash returns the proper SHA1 hash.
"""
file_name = 'testfile'
file_contents = 'foobar'
file_sha1 = '8843d7f92416211de9ebb963ff4ce28125932878'
tmp_file = tmp_path / file_name
tmp_file.write_text(file_contents)
assert verify_hash(tmp_file) == file_sha1
|
