PyExecutor ResourceResoluton helper class.

Create a class to call workflow execution requests to gRPC server.
Create an interceptor to use header authorization for gRPC calls.

Issue-ID: CCSDK-1989
Signed-off-by: Michal Jagiello <michal.jagiello@t-mobile.pl>
Change-Id: Ia449a089e02e7a12e31bee5e3b7debee506d8426

2 files changed, 155 insertions, 0 deletions

diff --git a/ms/py-executor/resource_resolution/tests/authorization_interceptor_test.py b/ms/py-executor/resource_resolution/tests/authorization_interceptor_test.py
new file mode 100644
index 000000000..4b03f0b36
--- /dev/null
+++ b/ms/py-executor/resource_resolution/tests/authorization_interceptor_test.py
@@ -0,0 +1,50 @@
+"""Copyright 2020 Deutsche Telekom.
+
+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.
+"""
+
+from unittest.mock import MagicMock, _Call
+
+import pytest
+
+from resource_resolution.authorization import AuthTokenInterceptor, NewClientCallDetails
+
+
+def test_resource_resolution_auth_token_interceptor():
+    """Test AuthTokenInterceptor class.
+
+     - Checks if it's correctly set default value.
+     - Checks if it's correctly set passed values.
+     - Checks if it's correctly checked if all header characters are lowercase.
+     - Checks if continuation function is called with headers setted 
+    """
+    interceptor: AuthTokenInterceptor = AuthTokenInterceptor("test_token", header="header")
+    assert interceptor.token == "test_token"
+    assert interceptor.header == "header"
+
+    interceptor: AuthTokenInterceptor = AuthTokenInterceptor("test_token")
+    assert interceptor.token == "test_token"
+    assert interceptor.header == "authorization"
+
+    with pytest.raises(ValueError):
+        AuthTokenInterceptor("test_token", header="Auth")
+
+    continuation_mock: MagicMock = MagicMock()
+    client_call_details: MagicMock = MagicMock()
+    request_iterator: MagicMock = MagicMock()
+
+    interceptor.intercept_stream_stream(continuation_mock, client_call_details, request_iterator)
+    continuation_mock.assert_called_once()
+    client_call_details_argument: _Call = continuation_mock.call_args_list[0][0][0]  # Get NewClientCallDetails instance
+    assert isinstance(client_call_details_argument, NewClientCallDetails)
+    assert client_call_details_argument.metadata[0] == (interceptor.header, interceptor.token)
diff --git a/ms/py-executor/resource_resolution/tests/resource_resolution_test.py b/ms/py-executor/resource_resolution/tests/resource_resolution_test.py
new file mode 100644
index 000000000..8a41357e6
--- /dev/null
+++ b/ms/py-executor/resource_resolution/tests/resource_resolution_test.py
@@ -0,0 +1,105 @@
+"""Copyright 2020 Deutsche Telekom.
+
+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.
+"""
+
+from google.protobuf import json_format
+from pytest import raises
+
+from resource_resolution.resource_resolution import (
+    ExecutionServiceInput,
+    ExecutionServiceOutput,
+    WorkflowExecution,
+    WorkflowExecutionResult,
+    WorkflowMode,
+)
+
+
+def test_workflow_execution_class():
+    """Workflow execution class tests.
+
+    - Test initialization and default values
+    - Test request message formatting
+    """
+    # Without inputs
+    workflow_execution: WorkflowExecution = WorkflowExecution("test blueprint", "test version", "test workflow")
+    assert workflow_execution.blueprint_name == "test blueprint"
+    assert workflow_execution.blueprint_version == "test version"
+    assert workflow_execution.workflow_name == "test workflow"
+    assert workflow_execution.workflow_inputs == {}
+    assert workflow_execution.workflow_mode == WorkflowMode.SYNC
+
+    msg: ExecutionServiceInput = workflow_execution.message
+    msg_dict: dict = json_format.MessageToDict(msg)
+    assert msg_dict["actionIdentifiers"]["blueprintName"] == "test blueprint"
+    assert msg_dict["actionIdentifiers"]["blueprintVersion"] == "test version"
+    assert msg_dict["actionIdentifiers"]["actionName"] == "test workflow"
+    assert msg_dict["actionIdentifiers"]["mode"] == "sync"
+    assert list(msg_dict["payload"].keys())[0] == "test workflow-request"
+    assert msg_dict["payload"]["test workflow-request"] == {}
+
+    # With inputs
+    workflow_execution: WorkflowExecution = WorkflowExecution(
+        "test blueprint2",
+        "test version2",
+        "test workflow2",
+        workflow_inputs={"test": "test"},
+        workflow_mode=WorkflowMode.ASYNC,
+    )
+    assert workflow_execution.blueprint_name == "test blueprint2"
+    assert workflow_execution.blueprint_version == "test version2"
+    assert workflow_execution.workflow_name == "test workflow2"
+    assert workflow_execution.workflow_inputs == {"test": "test"}
+    assert workflow_execution.workflow_mode == WorkflowMode.ASYNC
+
+    msg: ExecutionServiceInput = workflow_execution.message
+    msg_dict: dict = json_format.MessageToDict(msg)
+    assert msg_dict["actionIdentifiers"]["blueprintName"] == "test blueprint2"
+    assert msg_dict["actionIdentifiers"]["blueprintVersion"] == "test version2"
+    assert msg_dict["actionIdentifiers"]["actionName"] == "test workflow2"
+    assert msg_dict["actionIdentifiers"]["mode"] == "async"
+    assert list(msg_dict["payload"].keys())[0] == "test workflow2-request"
+    assert msg_dict["payload"]["test workflow2-request"] == {"test": "test"}
+
+
+def test_workflow_execution_result_class():
+    """Workflow execution result class tests.
+
+    - Test initizalization and default values
+    - Test `has_error` property
+    - Test `error_message` property
+    - Test payload formatting
+    """
+    workflow_execution: WorkflowExecution = WorkflowExecution("test blueprint", "test version", "test workflow")
+    execution_output: ExecutionServiceOutput = ExecutionServiceOutput()
+    execution_output.actionIdentifiers.blueprintName = "test blueprint"
+    execution_output.actionIdentifiers.blueprintVersion = "test version"
+    execution_output.actionIdentifiers.actionName = "test workflow"
+    execution_output.status.code = 200
+
+    execution_result: WorkflowExecutionResult = WorkflowExecutionResult(workflow_execution, execution_output)
+    assert not execution_result.has_error
+    with raises(AttributeError):
+        execution_result.error_message
+    assert execution_result.payload == {}
+    assert execution_result.blueprint_name == "test blueprint"
+    assert execution_result.blueprint_version == "test version"
+    assert execution_result.workflow_name == "test workflow"
+
+    execution_output.payload.update({"test_key": "test_value"})
+    execution_result: WorkflowExecutionResult = WorkflowExecutionResult(workflow_execution, execution_output)
+    assert execution_result.payload == {"test_key": "test_value"}
+
+    execution_output.status.code = 500
+    assert execution_result.has_error
+    assert execution_result.error_message == ""