Add more unit tests

ricardobranco777 · Aug 13, 2023 · ced7394 · ced7394
1 parent 04de0d6
commit ced7394
Show file tree

Hide file tree

Showing 5 changed files with 380 additions and 135 deletions.
diff --git a/Makefile b/Makefile
@@ -1,4 +1,4 @@
-FILES=*.py
+FILES=*.py tests/*.py
 
 .PHONY: all
 all: flake8 pylint pytest mypy
@@ -17,7 +17,7 @@ pytest:
 
 .PHONY: mypy
 mypy:
-	@mypy .
+	@mypy $(FILES)
 
 .PHONY: e2e
 e2e:

diff --git a/clean_registry.py b/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)
@@ -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__":

diff --git a/requirements-test.txt b/requirements-test.txt
@@ -3,4 +3,5 @@
 flake8
 pylint
 pytest
+pytest-mock
 mypy
diff --git a/tests/clean_registry_test.py b/tests/clean_registry_test.py