etorres/packaging-scripts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Incorporate hypothesis test framework into tests and streamline testing

Browse Source
This commit is contained in:
Eric Torres 2019-03-27 21:46:33 -07:00
parent c034d48723
commit 660e5870ef
1 changed files with 17 additions and 98 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

115
packaging_scripts/tests/test_pkgfiles.py
View File

@ -2,6 +2,8 @@ import packaging_scripts.pkgfiles as pkgfiles
import re import re
import unittest import unittest
from hypothesis import given
from hypothesis.strategies import iterables, text
from pathlib import Path from pathlib import Path
from types import GeneratorType from types import GeneratorType
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
@ -14,105 +16,21 @@ PKGREGEX = r"^[\w.+/-]+\.pkg\.tar(\.\w+)?$"
# Match any sigfile of any name with the .pkg.tar.*.sig extension # Match any sigfile of any name with the .pkg.tar.*.sig extension
SIGREGEX = r"^[\w.+/-]+\.pkg\.tar(\.\w+)?\.sig$" SIGREGEX = r"^[\w.+/-]+\.pkg\.tar(\.\w+)?\.sig$"
ALL_PKGFILES = [
Path("pkg1.pkg.tar.xz"),
Path("pkg1.pkg.tar.xz.sig"),
Path("pkg2.pkg.tar.xz"),
Path("pkg3.pkg.tar.xz"),
Path("pkg3.pkg.tar.xz.sig"),
Path("notapkg.sig"),
]
# ========== Functions ==========
def get_all_files(*args):
for f in ALL_PKGFILES:
yield f
def filter_files(query):
for f in [Path("pkg3.pkg.tar.xz"), Path("pkg3.pkg.tar.xz.sig")]:
yield f
# ========== Unit Tests ========== # ========== Unit Tests ==========
class TestGetPkgfiles(unittest.TestCase): class TestFilterByRegex(unittest.TestCase):
def setUp(self): @given(iterables(text()))
self.patched_path = patch.object(Path, "glob") def test_pkgregex(self, i):
self.mocked_glob = self.patched_path.start() result = pkgfiles._filter_by_regex(pkgfiles.PKGREGEX, i)
expected = [Path(p) for p in i if re.match(PKGREGEX, str(p))]
self.mocked_glob.side_effect = get_all_files
def test_yield_no_query(self):
result = pkgfiles.get()
expected = [s for s in get_all_files() if re.match(PKGREGEX, str(s))]
self.assertListEqual(list(result), expected)
self.assertIsInstance(result, GeneratorType)
def test_yield_with_query(self):
self.mocked_glob.side_effect = filter_files
result = pkgfiles.get("pkg3")
expected = [Path("pkg3.pkg.tar.xz")]
self.assertListEqual(list(result), expected)
self.assertIsInstance(result, GeneratorType)
def tearDown(self):
self.patched_path.stop()
class TestGetSigfiles(unittest.TestCase):
def setUp(self):
self.patched_path = patch.object(Path, "glob")
self.mocked_glob = self.patched_path.start()
self.mocked_glob.side_effect = get_all_files
def test_yield_no_query(self):
result = pkgfiles.get(signatures_only=True)
expected = [s for s in get_all_files() if re.match(SIGREGEX, str(s))]
self.assertListEqual(list(result), expected)
self.assertIsInstance(result, GeneratorType)
def test_yield_with_query(self):
self.mocked_glob.side_effect = filter_files
result = pkgfiles.get("pkg3", signatures_only=True)
expected = [Path("pkg3.pkg.tar.xz.sig")]
self.assertListEqual(list(result), expected)
self.assertIsInstance(result, GeneratorType)
def tearDown(self):
self.patched_path.stop()
class TestFilterPkgfiles(unittest.TestCase):
def test_yield(self):
result = pkgfiles.filter(ALL_PKGFILES)
expected = [s for s in ALL_PKGFILES if re.match(PKGREGEX, str(s))]
expected.sort()
self.assertListEqual(sorted(result), expected) self.assertListEqual(sorted(result), expected)
self.assertIsInstance(result, GeneratorType) self.assertIsInstance(result, GeneratorType)
def test_yield_duplicate(self): @given(iterables(text()))
duplicate_info = ['package.pkg.tar.xz', 'package.pkg.tar.xz'] def test_sigregex(self, i):
result = pkgfiles._filter_by_regex(pkgfiles.SIGREGEX, i)
result = pkgfiles.filter(duplicate_info) expected = [Path(p) for p in i if re.match(SIGREGEX, str(p))]
expected = [Path('package.pkg.tar.xz')]
self.assertListEqual(sorted(result), expected)
self.assertIsInstance(result, GeneratorType)
class TestFilterSigfiles(unittest.TestCase):
def test_yield(self):
result = pkgfiles.filter(ALL_PKGFILES, signatures_only=True)
expected = [s for s in get_all_files() if re.match(SIGREGEX, str(s))]
self.assertListEqual(list(result), expected) self.assertListEqual(list(result), expected)
self.assertIsInstance(result, GeneratorType) self.assertIsInstance(result, GeneratorType)
@ -123,14 +41,15 @@ class TestAddPkgfile(unittest.TestCase):
self.patched_shutil = patch(f"{TESTING_MODULE}.shutil") self.patched_shutil = patch(f"{TESTING_MODULE}.shutil")
self.mocked_shutil = self.patched_shutil.start() self.mocked_shutil = self.patched_shutil.start()
self.pkgfile = Path("/home/user/pkgfile") @given(text())
self.cachedir = Path("/var/cache/repo") def test_pkgfile_path(self, p):
pkgfile = Path(p)
cachedir = Path("/var/cache/repo")
def test_pkgfile_path(self): pkgfiles.add(pkgfile, cachedir)
pkgfiles.add(self.pkgfile, self.cachedir)
self.mocked_shutil.move.assert_called_with( self.mocked_shutil.move.assert_called_with(
self.pkgfile, Path("/var/cache/repo/pkgfile") pkgfile, Path(f"/var/cache/repo/{pkgfile.name}")
) )
def tearDown(self): def tearDown(self):