Add more unit tests

.coveragerc

@@ -0,0 +1,3 @@
+[report]
+omit =
+    */tests/*

.gitignore

@@ -1,2 +1,3 @@
 .venv/
 *.pyc
+.coverage

Makefile

@@ -1,4 +1,4 @@
-FILES=*.py
+FILES=*.py tests/*.py
 
 .PHONY: all
 all: flake8 pylint pytest mypy
@@ -13,11 +13,11 @@ pylint:
 
 .PHONY: pytest
 pytest:
-	@pytest -v
+	@pytest --capture=sys -v --cov --cov-report term-missing
 
 .PHONY: mypy
 mypy:
-	@mypy .
+	@mypy $(FILES)
 
 .PHONY: e2e
 e2e:

README.md

@@ -1,5 +1,7 @@
 ![Build Status](https://github.com/ricardobranco777/clean_registry/actions/workflows/ci.yml/badge.svg)
 
+[![codecov](https://codecov.io/gh/ricardobranco777/linux-procfs/branch/master/graph/badge.svg)](https://codecov.io/gh/ricardobranco777/linux-procfs)
+
 # clean_registry
 
 Docker Registry cleanup and image removal tool.

clean_registry.py

@@ -61,7 +61,7 @@ def run_command(command: list) -> int:
         return proc.wait()
 
 
-def clean_registry(images: list[str], dry_run: bool = False) -> None:
+def clean_registrydir(images: list[str], dry_run: bool = False) -> None:
     '''Clean registry'''
     registry_dir = os.environ.get("REGISTRY_STORAGE_FILESYSTEM_ROOTDIRECTORY", "/var/lib/registry")
     logging.debug("registry directory: %s", registry_dir)
@@ -114,7 +114,13 @@ def clean_repo(basedir: str, image: str, dry_run: bool = False) -> None:
         clean_tag(basedir, repo, tag, dry_run)
 
 
-def parse_args():
+def get_os_release():
+    '''Get OS release'''
+    with open("/etc/os-release", encoding="utf-8") as file:
+        return {k: v.strip('"') for k, v in [line.split('=') for line in file.read().splitlines()]}
+
+
+def parse_args():  # pragma: no cover
     """Parse args"""
     parser = ArgumentParser()
     parser.add_argument(
@@ -131,24 +137,18 @@ def parse_args():
     return parser.parse_args()
 
 
-def print_versions():
-    '''Print useful information for debugging'''
-    print(f'{os.path.basename(sys.argv[0])} {VERSION}')
-    print(f'Python {sys.version}')
-    print(subprocess.check_output(shlex.split("/bin/registry --version")).decode("utf-8").strip())
-    with open("/etc/os-release", encoding="utf-8") as file:
-        osrel = {k: v.strip('"') for k, v in [line.split('=') for line in file.read().splitlines()]}
-    print(osrel['NAME'], osrel['VERSION_ID'])
-
-
 def main():
     '''Main function'''
     if not is_container() or not os.path.isfile("/bin/registry"):
         sys.exit("ERROR: This script should run inside a registry:2 container!")
 
     args = parse_args()
     if args.version:
-        print_versions()
+        print(f'{os.path.basename(sys.argv[0])} {VERSION}')
+        print(f'Python {sys.version}')
+        print(subprocess.check_output(shlex.split("/bin/registry --version")).decode("utf-8").strip())
+        osrel = get_os_release()
+        print(osrel['NAME'], osrel['VERSION_ID'])
         sys.exit(0)
 
     for image in args.images:
@@ -158,7 +158,7 @@ def main():
     fmt = "%(asctime)s %(levelname)-8s %(message)s"
     logging.basicConfig(format=fmt, stream=sys.stderr, level=args.log.upper())
 
-    clean_registry(images=args.images, dry_run=args.dry_run)
+    clean_registrydir(images=args.images, dry_run=args.dry_run)
 
 
 if __name__ == "__main__":

requirements-test.txt

@@ -3,4 +3,6 @@
 flake8
 pylint
 pytest
+pytest-cov
+pytest-mock
 mypy