From 71b1f6874b81f2ef9490e60233424e8628626470 Mon Sep 17 00:00:00 2001
From: HSN <hasanlock@gmail.com>
Date: Sun, 29 Dec 2024 10:54:38 +0600
Subject: [PATCH] refactor: validate module test and unused imports

---
 chk/modules/validate/__init__.py               | 13 +++++--------
 tests/modules/validate/validate_module_test.py |  5 +----
 2 files changed, 6 insertions(+), 12 deletions(-)

diff --git a/chk/modules/validate/__init__.py b/chk/modules/validate/__init__.py
index 0f193e4a..3aa2ad8b 100644
--- a/chk/modules/validate/__init__.py
+++ b/chk/modules/validate/__init__.py
@@ -38,7 +38,7 @@
     AssertionEntityProperty,
     get_schema_map,
 )
-from chk.modules.validate.entities import RunReport, ValidationTask
+from chk.modules.validate.entities import ValidationTask
 from chk.modules.validate.services import ValidatePresenter
 
 VERSION_SCOPE = ["validation"]
@@ -136,7 +136,7 @@ def set_data_template(
         )
 
     @staticmethod
-    def  process_data_template(variables: Variables) -> None:
+    def process_data_template(variables: Variables) -> None:
         """process data or template before assertion"""
         data = variables[ValidationConfigNode.VAR_NODE.value]
         tmp_variables = {
@@ -223,16 +223,14 @@ def call(file_ctx: FileContext, exec_ctx: ExecuteContext) -> ExecResponse:
         )
     except Exception as ex:
         error_trace(exception=sys.exc_info()).error(ex)
-        return ExecResponse(file_ctx=file_ctx, exec_ctx=exec_ctx, exception=ex)
+        return ExecResponse(file_ctx=file_ctx, exec_ctx=exec_ctx, exception=ex, report={"is_success": False})
 
     variable_doc = Variables()
     VariableTableManager.handle(variable_doc, validate_doc, exec_ctx)
     debug(variable_doc.data)
 
     # handle passed data in asserts
-    ValidationDocumentSupport.set_data_template(
-        validate_doc, variable_doc, exec_ctx
-    )
+    ValidationDocumentSupport.set_data_template(validate_doc, variable_doc, exec_ctx)
 
     ValidationDocumentSupport.process_data_template(variable_doc)
     debug(variable_doc.data)
@@ -245,7 +243,7 @@ def call(file_ctx: FileContext, exec_ctx: ExecuteContext) -> ExecResponse:
         run_rpt = AssertionEntryListRunner.test_run(assert_list, variable_doc.data)
     except Exception as ex:
         error_trace(exception=sys.exc_info()).error(ex)
-        return ExecResponse(file_ctx=file_ctx, exec_ctx=exec_ctx, exception=ex)
+        return ExecResponse(file_ctx=file_ctx, exec_ctx=exec_ctx, exception=ex, report={"is_success": False})
 
     output_data = Variables(
         {
@@ -276,7 +274,6 @@ def call(file_ctx: FileContext, exec_ctx: ExecuteContext) -> ExecResponse:
         variables=variable_doc,
         extra=run_rpt,
         exposed=exposed_data,
-        exception=None,
         report={
             "is_success": run_rpt.count_fail == 0,
             "count_all": len(validate_doc.asserts),
diff --git a/tests/modules/validate/validate_module_test.py b/tests/modules/validate/validate_module_test.py
index 75442712..f361b007 100644
--- a/tests/modules/validate/validate_module_test.py
+++ b/tests/modules/validate/validate_module_test.py
@@ -65,9 +65,6 @@ def test_fail_for_wrong_key() -> None:
         )
 
         er = call(file_ctx, exec_ctx)
-        import icecream
-        icecream.ic(er)
 
         assert isinstance(er, ExecResponse)
-        icecream.ic(er.exception, type(er.exception))
-        # assert isinstance(er.exception, RuntimeError)
+        assert isinstance(er.exception, KeyError)