PrinterValidator

faissaloux · Dec 1, 2023 · d754f91 · d754f91
1 parent d1ee87a
commit d754f91
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 25 deletions.
diff --git a/termspark/termspark.py b/termspark/termspark.py
@@ -7,10 +7,10 @@
 from .exceptions.lenNotSupportedException import LenNotSupportedException
 from .exceptions.minNotReachedException import MinNotReachedException
 from .exceptions.multiplePositionsNotSupported import MultiplePositionsNotSupported
-from .exceptions.printerArgException import PrinterArgException
 from .helpers.existenceChecker import ExistenceChecker
 from .painter.painter import Painter
 from .structurer.structurer import Structurer
+from .validators.printerValidator import PrinterValidator
 
 
 class TermSpark:
@@ -45,14 +45,7 @@ def __init__(self):
     def print_left(
         self, content: str, color: Optional[str] = None, highlight: Optional[str] = None
     ):
-        if any(
-            [
-                isinstance(content, list),
-                isinstance(color, list),
-                isinstance(highlight, list),
-            ]
-        ):
-            raise PrinterArgException("left")
+        PrinterValidator().validate(content, color, highlight)
 
         self.spark_left([content, color, highlight])  # type: ignore
 
@@ -61,14 +54,7 @@ def print_left(
     def print_right(
         self, content: str, color: Optional[str] = None, highlight: Optional[str] = None
     ):
-        if any(
-            [
-                isinstance(content, list),
-                isinstance(color, list),
-                isinstance(highlight, list),
-            ]
-        ):
-            raise PrinterArgException("right")
+        PrinterValidator().validate(content, color, highlight)
 
         self.spark_right([content, color, highlight])  # type: ignore
 
@@ -77,14 +63,7 @@ def print_right(
     def print_center(
         self, content: str, color: Optional[str] = None, highlight: Optional[str] = None
     ):
-        if any(
-            [
-                isinstance(content, list),
-                isinstance(color, list),
-                isinstance(highlight, list),
-            ]
-        ):
-            raise PrinterArgException("center")
+        PrinterValidator().validate(content, color, highlight)
 
         self.spark_center([content, color, highlight])  # type: ignore
 

diff --git a/termspark/validators/__init__.py b/termspark/validators/__init__.py
diff --git a/termspark/validators/printerValidator.py b/termspark/validators/printerValidator.py
@@ -0,0 +1,13 @@
+from ..exceptions.printerArgException import PrinterArgException
+
+
+class PrinterValidator:
+    def validate(self, content, color, highlight):
+        if any(
+            [
+                isinstance(content, list),
+                isinstance(color, list),
+                isinstance(highlight, list),
+            ]
+        ):
+            raise PrinterArgException("left")
diff --git a/tests/printer_validaor_test.py b/tests/printer_validaor_test.py
@@ -0,0 +1,22 @@
+import pytest
+
+from termspark.exceptions.printerArgException import PrinterArgException
+from termspark.validators.printerValidator import PrinterValidator
+
+
+class TestPrinterValidator:
+    def test_validate_content(self):
+        with pytest.raises(PrinterArgException):
+            PrinterValidator().validate(["termspark"], "black", "green")
+
+    def test_validate_color(self):
+        with pytest.raises(PrinterArgException):
+            PrinterValidator().validate("termspark", ["black"], "green")
+
+    def test_validate_highlight(self):
+        with pytest.raises(PrinterArgException):
+            PrinterValidator().validate("termspark", "black", ["green"])
+
+    def test_validate_all(self):
+        with pytest.raises(PrinterArgException):
+            PrinterValidator().validate(["termspark"], ["black"], ["green"])