1 files changed, 68 insertions, 0 deletions
diff --git a/mod/distributorapi/tests/test_transform.py b/mod/distributorapi/tests/test_transform.py
new file mode 100644
index 0000000..f275e63
--- /dev/null
+++ b/mod/distributorapi/tests/test_transform.py
@@ -0,0 +1,68 @@
+# ============LICENSE_START=======================================================
+# Copyright (c) 2019 AT&T Intellectual Property. All rights reserved.
+# ================================================================================
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============LICENSE_END=========================================================
+import os, json
+from distributor import transform as tr
+
+TEST_DIR = os.path.dirname(__file__)
+
+def _load_data(filename):
+    path = os.path.join(TEST_DIR, filename)
+    with open(path) as f:
+        return json.load(f)
+
+def _setup():
+    flow = _load_data("flow.json")
+    components = _load_data("components.json")
+    components = dict([((c["name"], c["version"]), c) for c in components])
+    return (flow, components)
+
+
+def test_get_component():
+    flow, components = _setup()
+    c = tr.get_component(flow, components, "a8134467-b4b4-348f-8a1c-8d732fe4fcad")
+    assert "3fadb641-2079-4ca9-bb07-0df5952967fc" == c["id"]
+
+
+def test_make_fbp_from_flow():
+    flow, components = _setup()
+
+    fbp = tr.make_fbp_from_flow(flow, components)
+    assert len(fbp) == 4
+
+    def check_node(n):
+        n["payload"]["component_id"]
+
+    expected = ["75c9a179-b36b-4985-9445-d44c8768d6eb", "3fadb641-2079-4ca9-bb07-0df5952967fc"]
+    actual = [n["payload"]["component_id"] for n in fbp if n["command"] == "addnode"]
+    assert list(sorted(expected)) == list(sorted(actual))
+
+    # Test processor to processor scenario
+    expected = {'metadata': {'data_type': 'json',
+                           'dmaap_type': 'MR',
+                           'name': 'foo-conn'},
+              'src': {'node': '75c9a179-b36b-4985-9445-d44c8768d6eb',
+                      'port': 'ves-pnfRegistration-secondary'},
+              'tgt': {'node': '3fadb641-2079-4ca9-bb07-0df5952967fc',
+                      'port': 'predict_subscriber'}}
+    actual = [e["payload"] for e in fbp if e["command"] == "addedge"]
+    assert actual[0] == expected or actual[1] == expected
+
+    # Test input port to processor scenario
+    expected = {'metadata': {'data_type': 'json', 'dmaap_type': 'MR',
+        'name': 'ves-data-conn'}, 'src': {},
+        'tgt': {'node': '75c9a179-b36b-4985-9445-d44c8768d6eb',
+            'port': 'ves-notification'}}
+    assert actual[0] == expected or actual[1] == expected