1 files changed, 105 insertions, 0 deletions
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 <https://www.gnu.org/licenses/>.
+
+"""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)