deltafi 2.1.0__tar.gz → 2.2.0__tar.gz

{deltafi-2.1.0 → deltafi-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: deltafi
-Version: 2.1.0
+Version: 2.2.0
 Summary: SDK for DeltaFi plugins and actions
 License: Apache License, Version 2.0
 Keywords: deltafi

{deltafi-2.1.0 → deltafi-2.2.0}/deltafi/result.py

@@ -95,6 +95,7 @@ class IngressResultItem:
         self._did = str(uuid.uuid4())
         self.content = []
         self.metadata = {}
+        self.annotations = {}
         self.delta_file_name = delta_file_name
 
     @property
@@ -139,12 +140,17 @@ class IngressResultItem:
             segment_names.update(c.get_segment_names())
         return segment_names
 
+    def annotate(self, key: str, value: str):
+                    self.annotations[key] = value
+                    return self
+
     def response(self):
         return {
             'did': self._did,
             'deltaFileName': self.delta_file_name,
             'metadata': self.metadata,
-            'content': [content.json() for content in self.content]
+            'content': [content.json() for content in self.content],
+            'annotations': self.annotations
         }
 
 

{deltafi-2.1.0 → deltafi-2.2.0}/deltafi/test_kit/assertions.py

@@ -29,10 +29,18 @@ def assert_equal_with_label(e, a, l):
     assert e == a, f"{l}. Expected:\n<<{e}>>\nBut was:\n<<{a}>>"
 
 
+def assert_equal_short(e, a, l):
+    assert e == a, f"{l}. E:{e}, A::{a}"
+
+
 def assert_equal_len(e, a):
     assert len(e) == len(a), f"{len(e)} != {len(a)}"
 
 
+def assert_equal_len_with_label(e, a, l):
+    assert len(e) == len(a), f"{l}. {len(e)} != {len(a)}"
+
+
 def assert_key_in(k, m):
     assert k in m, f"{k} not found"
 
@@ -45,4 +53,4 @@ def assert_keys_and_values(expected: Dict, actual: Dict):
     for key in expected:
         assert_key_in(key, actual)
         if expected[key] != IGNORE_VALUE:
-            assert_equal(expected[key], actual[key])
+            assert_equal_short(expected[key], actual[key], f"invalid value for key {key}")

{deltafi-2.1.0 → deltafi-2.2.0}/deltafi/test_kit/framework.py

@@ -350,13 +350,13 @@ class ActionTest(ABC):
 
     @staticmethod
     def compare_one_metric(expected: Metric, result: Metric):
-        assert expected.name == result.name
-        assert_equal_with_label(expected.value, result.value, expected.name)
+        assert_equal_short(expected.name, result.name, "invalid metric name")
+        assert_equal_short(expected.value, result.value, f"invalid metric value for {expected.name}")
         assert_keys_and_values(expected.tags, result.tags)
 
     def compare_metrics(self, expected_metrics: List[Metric], results: List[Metric]):
         if len(expected_metrics) > 0:
-            assert_equal_len(expected_metrics, results)
+            assert_equal_len_with_label(expected_metrics, results, "invalid metrics count")
             for index, expected in enumerate(expected_metrics):
                 self.compare_one_metric(expected, results[index])
 

{deltafi-2.1.0 → deltafi-2.2.0}/deltafi/test_kit/timed_ingress.py

@@ -19,7 +19,6 @@
 from typing import List
 
 from deltafi.result import IngressResult, IngressResultItem, IngressStatusEnum
-
 from .assertions import *
 from .framework import TestCaseBase, ActionTest, IOContent
 
@@ -44,12 +43,15 @@ class TimedIngressTestCase(TestCaseBase):
         self.status = status
         self.status_message = status_message
 
-    def add_ingress_result_item(self, content: List[IOContent], metadata: Dict, name: str = None):
+    def add_ingress_result_item(self, content: List[IOContent], metadata: Dict, name: str = None, annotations: Dict = None):
+        if annotations is None:
+            annotations = {}
         self.results.append(
             {
                 'content': content,
                 'metadata': metadata,
-                'name': name
+                'name': name,
+                'annotations': annotations
             }
         )
 
@@ -74,12 +76,12 @@ class TimedIngressActionTest(ActionTest):
         self.assert_ingress_result(test_case, result)
 
     def assert_ingress_result(self, test_case: TimedIngressTestCase, result: IngressResult):
-        assert_equal(test_case.memo, result.memo)
-        assert_equal(test_case.execute_immediate, result.execute_immediate)
-        assert_equal(test_case.status, result.status)
-        assert_equal(test_case.status_message, result.status_message)
+        assert_equal_short(test_case.memo, result.memo, "invalid memo")
+        assert_equal_short(test_case.execute_immediate, result.execute_immediate, "invalid execute_immediate")
+        assert_equal_short(test_case.status, result.status, "invalid status")
+        assert_equal_with_label(test_case.status_message, result.status_message, "invalid status_message")
 
-        assert_equal_len(test_case.results, result.ingress_result_items)
+        assert_equal_len_with_label(test_case.results, result.ingress_result_items, "item count mismatch")
         for index, ingress_item in enumerate(result.ingress_result_items):
             self.compare_one_ingress_item(test_case, ingress_item, index)
             expected = test_case.results[index]
@@ -94,3 +96,6 @@ class TimedIngressActionTest(ActionTest):
 
         # Check metadata
         assert_keys_and_values(expected['metadata'], result.metadata)
+
+        # Check annotations
+        assert_keys_and_values(expected['annotations'], result.annotations)

{deltafi-2.1.0 → deltafi-2.2.0}/deltafi/test_kit/transform.py

@@ -19,7 +19,6 @@
 from typing import List
 
 from deltafi.result import TransformResult, TransformResults
-
 from .assertions import *
 from .framework import TestCaseBase, ActionTest, IOContent
 
@@ -74,7 +73,7 @@ class TransformActionTest(ActionTest):
         self.assert_transform_results(test_case, result)
 
     def assert_transform_results(self, test_case: TransformTestCase, result: TransformResults):
-        assert_equal_len(test_case.results, result.child_results)
+        assert_equal_len_with_label(test_case.results, result.child_results, "invalid child count")
         for index, child_result in enumerate(result.child_results):
             self.compare_one_transform_result(test_case, child_result, index)
             expected = test_case.results[index]

{deltafi-2.1.0 → deltafi-2.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "deltafi"
-version = "2.1.0"
+version = "2.2.0"
 description = "SDK for DeltaFi plugins and actions"
 authors = ["DeltaFi <deltafi@systolic.com>"]
 license = "Apache License, Version 2.0"
@@ -36,7 +36,7 @@ optional = true
 [tool.poetry.group.test.dependencies]
 pytest = "==8.3.3"
 pytest-mock = "==3.14.0"
-mockito = "==1.5.1"
+mockito = "==1.5.3"
 
 [tool.poetry.urls]
 'Source Code' = "https://gitlab.com/deltafi/deltafi"