insightconnect-plugin-runtime 5.5.1__tar.gz → 5.5.3__tar.gz

{insightconnect_plugin_runtime-5.5.1/insightconnect_plugin_runtime.egg-info → insightconnect_plugin_runtime-5.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: insightconnect-plugin-runtime
-Version: 5.5.1
+Version: 5.5.3
 Summary: InsightConnect Plugin Runtime
 Home-page: https://github.com/rapid7/komand-plugin-sdk-python
 Author: Rapid7 Integrations Alliance
@@ -211,6 +211,8 @@ contributed. Black is installed as a test dependency and the hook can be initial
 after cloning this repository.
 
 ## Changelog
+* 5.5.3 - Adding in a new endpoint that can be called to run a task connection test
+* 5.5.2 - Address bug with typing for type `List` in Python 3.8
 * 5.5.1 - Address bug with typing for type `Tuple` in Python 3.8
 * 5.5.0 - Updated helper class to add `make_request`, `response_handler`, `extract_json`, and `request_error_handling` for HTTP requests, and `hash_sha1` and `compare_and_dedupe_hashes` to provide support for hash comparisons | Add `METHOD_NOT_ALLOWED`, `CONFLICT`, `REDIRECT_ERROR`, and `CONNECTION_ERROR` to PluginException presets
 * 5.4.9 - Updated aws_client to clean assume role json object to remove any none or empty string values.

{insightconnect_plugin_runtime-5.5.1 → insightconnect_plugin_runtime-5.5.3}/README.md

@@ -182,6 +182,8 @@ contributed. Black is installed as a test dependency and the hook can be initial
 after cloning this repository.
 
 ## Changelog
+* 5.5.3 - Adding in a new endpoint that can be called to run a task connection test
+* 5.5.2 - Address bug with typing for type `List` in Python 3.8
 * 5.5.1 - Address bug with typing for type `Tuple` in Python 3.8
 * 5.5.0 - Updated helper class to add `make_request`, `response_handler`, `extract_json`, and `request_error_handling` for HTTP requests, and `hash_sha1` and `compare_and_dedupe_hashes` to provide support for hash comparisons | Add `METHOD_NOT_ALLOWED`, `CONFLICT`, `REDIRECT_ERROR`, and `CONNECTION_ERROR` to PluginException presets
 * 5.4.9 - Updated aws_client to clean assume role json object to remove any none or empty string values.

{insightconnect_plugin_runtime-5.5.1 → insightconnect_plugin_runtime-5.5.3}/insightconnect-plugin-swagger.json

@@ -415,10 +415,19 @@
         }
       }
     },
-    "/api/v1/connection/test": {
+    "/api/v1/connection/{connection_test_type}": {
       "post": {
         "summary": "Run connection test",
         "description": "Run InsightConnect plugin connection test",
+        "parameters": [
+          {
+            "in": "path",
+            "name": "connection_test_type",
+            "description": "Type of connection test to be run",
+            "type": "string",
+            "required": true
+          }
+        ],
         "responses": {
           "200": {
             "description": "Connection test output to be returned",

{insightconnect_plugin_runtime-5.5.1 → insightconnect_plugin_runtime-5.5.3}/insightconnect_plugin_runtime/api/endpoints.py

@@ -634,14 +634,19 @@ class Endpoints:
             schema = conn.schema
             return jsonify(ConnectionDetailsSchema().dump(schema))
 
-        @v1.route("/connection/test", methods=["POST"])
-        def connection_test():
+        @v1.route("/connection/<string:connection_test_type>", methods=["POST"])
+        def connection_test(connection_test_type: str):
             """
             Run connection test endpoint
             ---
             post:
               summary: Run connection test
               description: Run InsightConnect plugin connection test
+              parameters:
+                - in: path
+                  name: connection_test_type
+                  description: Type of connection test to be run
+                  type: string
               responses:
                 200:
                   description: Connection test output to be returned
@@ -658,12 +663,26 @@ class Endpoints:
             status_code = 200
             output = None
 
+            supported_tests = ["test", "test_task"]
+            if connection_test_type not in supported_tests:
+                return make_response(
+                    jsonify(
+                        {
+                            "error": f"The requested endpoint is not available, only {supported_tests} are supported.",
+                            "code": 404,
+                            "method": "POST",
+                            "name": "Not Found"
+                        }
+                    ),
+                    404,
+                )
+
             input_message = request.get_json(force=True)
             Endpoints.validate_action_trigger_task_empty_input(input_message)
 
             try:
                 output = self.plugin.handle_step(
-                    input_message, is_debug=self.debug, is_test=True
+                    input_message, is_debug=self.debug, is_test=True, connection_test_type=connection_test_type
                 )
                 if output.get("body", {}).get("output") is None:
                     status_code = 204

{insightconnect_plugin_runtime-5.5.1 → insightconnect_plugin_runtime-5.5.3}/insightconnect_plugin_runtime/helper.py

@@ -31,11 +31,11 @@ ENCODE_TYPE = "utf-8"
 DEFAULTS_HOURS_AGO = 24
 
 
-def hash_sha1(log: dict) -> str:
+def hash_sha1(log: Dict) -> str:
     """
     Iterate through a dictionary and hash each value.
     :param log: Dictionary to be hashed.
-    :type dict:
+    :type Dict:
     :return: Hex digest of hash.
     :rtype: str
     """
@@ -80,8 +80,8 @@ def make_request(
     allow_redirects: bool = True,
     exception_custom_configs: Dict[int, Exception]={},
     exception_data_location: str = None,
-    allowed_status_codes: list[str] = [],
-) -> Tuple[requests.Response, dict]:
+    allowed_status_codes: List[str] = [],
+) -> Tuple[requests.Response, Dict]:
     """
     Makes a HTTP request while checking for RequestErrors and JSONDecodeErrors
     Returns the request response and the response JSON if required.
@@ -92,7 +92,7 @@ def make_request(
     :param verify: Whether to verify the server's TLS certificate
     :type bool:
     :param cert: Certificate to include with request, str location or key/value pair
-    :type Union[str, dict]:
+    :type Union[str, Dict]:
     :param stream: Whether to immediately download the response content
     :type bool:
     :param allow_redirects: Set to true by default
@@ -102,10 +102,10 @@ def make_request(
     :param exception_data_location: Where the returned data should be retrieved. Can provide ResponseExceptionData values.
     :type str:
     :param allowed_status_codes: Status codes that will not raise an exception.
-    :type list[str]:
+    :type List[str]:
 
     :return: The request response and the response JSON.
-    :rtype: Tuple[Response, dict]
+    :rtype: Tuple[Response, Dict]
     """
     try:
         with requests.Session() as session:
@@ -139,7 +139,7 @@ def make_request(
     return response
 
 
-def extract_json(response: requests.Response) -> dict:
+def extract_json(response: requests.Response) -> Dict:
     """Extract JSON from a request object while error handling a JSONDecodeError.
     :param response: Response object ot utilize in extract
     :type Response:
@@ -197,7 +197,7 @@ def response_handler(
     response: requests.Response,
     custom_configs: Dict[int, Exception]={},
     data_location: str = None,
-    allowed_status_codes: list[str] = [],
+    allowed_status_codes: List[str] = [],
 ) -> None:
     """
     Check response status codes and return a generic PluginException preset if a HTTPError is raised.
@@ -210,7 +210,7 @@ def response_handler(
     :param data_location: Where the returned data should be retrieved. Can provide ResponseExceptionData values.
     :type str:
     :param allowed_status_codes: Status codes that will not raise an exception.
-    :type list[str]:
+    :type List[str]:
 
     :return: None.
     :rtype: None

{insightconnect_plugin_runtime-5.5.1 → insightconnect_plugin_runtime-5.5.3}/insightconnect_plugin_runtime/plugin.py

@@ -362,7 +362,7 @@ class Plugin(object):
         if "input" not in body:
             body["input"] = {}
 
-    def handle_step(self, input_message, is_test=False, is_debug=False):
+    def handle_step(self, input_message, is_test=False, is_debug=False, connection_test_type="test"):
         """
         Executes a single step, given the input message dictionary.
 
@@ -427,6 +427,7 @@ class Plugin(object):
                     log_stream,
                     is_test,
                     is_debug,
+                    connection_test_type=connection_test_type
                 )
             elif message_type == "trigger_start":
                 out_type = "trigger_event"
@@ -437,6 +438,7 @@ class Plugin(object):
                     log_stream,
                     is_test,
                     is_debug,
+                    connection_test_type=connection_test_type
                 )
             elif message_type == "task_start":
                 out_type = "task_event"
@@ -449,6 +451,7 @@ class Plugin(object):
                         log_stream,
                         is_test,
                         is_debug,
+                        connection_test_type=connection_test_type
                     )
                 else:
                     (
@@ -464,6 +467,7 @@ class Plugin(object):
                         log_stream,
                         is_test,
                         is_debug,
+                        connection_test_type=connection_test_type
                     )
             elif message_type == "connection_test":
                 out_type = "connection_test"
@@ -475,6 +479,7 @@ class Plugin(object):
                     is_test,
                     is_debug,
                     is_connection_test=True,
+                    connection_test_type=connection_test_type
                 )
         except (ClientException, ServerException, PluginException, Exception) as e:
             success = False
@@ -513,6 +518,7 @@ class Plugin(object):
         is_test=False,
         is_debug=False,
         is_connection_test=False,
+        connection_test_type="test"
     ):
         """
         Starts an action.
@@ -523,6 +529,7 @@ class Plugin(object):
         :param is_test: True if the action's test method should execute
         :param is_debug: True if debug is enabled
         :param is_connection_test: True if connection test is running
+        :param connection_test_type: The type of connection test to be run
         :return: An action_event message
         """
         connection = self.connection_cache.get(message_body["connection"], logger)
@@ -534,7 +541,9 @@ class Plugin(object):
                     plugin_version=connection.meta.version,
                 )
             )
-            if hasattr(connection, "test"):
+            if hasattr(connection, "test_task") and connection_test_type == "test_task":
+                func = connection.test_task
+            elif hasattr(connection, "test"):
                 func = connection.test
             else:
                 raise NotImplementedError(
@@ -617,7 +626,10 @@ class Plugin(object):
             if is_test:
                 # Check if connection test func available. If so - use it (preferred). Else fallback to action/trigger test
                 if hasattr(step.connection, "test"):
-                    func = step.connection.test
+                    if hasattr(step.connection, "test") and connection_test_type == "test_task":
+                        func = step.connection.task_test
+                    else:
+                        func = step.connection.test
                 else:
                     func = step.test
             else:

{insightconnect_plugin_runtime-5.5.1 → insightconnect_plugin_runtime-5.5.3/insightconnect_plugin_runtime.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: insightconnect-plugin-runtime
-Version: 5.5.1
+Version: 5.5.3
 Summary: InsightConnect Plugin Runtime
 Home-page: https://github.com/rapid7/komand-plugin-sdk-python
 Author: Rapid7 Integrations Alliance
@@ -211,6 +211,8 @@ contributed. Black is installed as a test dependency and the hook can be initial
 after cloning this repository.
 
 ## Changelog
+* 5.5.3 - Adding in a new endpoint that can be called to run a task connection test
+* 5.5.2 - Address bug with typing for type `List` in Python 3.8
 * 5.5.1 - Address bug with typing for type `Tuple` in Python 3.8
 * 5.5.0 - Updated helper class to add `make_request`, `response_handler`, `extract_json`, and `request_error_handling` for HTTP requests, and `hash_sha1` and `compare_and_dedupe_hashes` to provide support for hash comparisons | Add `METHOD_NOT_ALLOWED`, `CONFLICT`, `REDIRECT_ERROR`, and `CONNECTION_ERROR` to PluginException presets
 * 5.4.9 - Updated aws_client to clean assume role json object to remove any none or empty string values.

{insightconnect_plugin_runtime-5.5.1 → insightconnect_plugin_runtime-5.5.3}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name="insightconnect-plugin-runtime",
-    version="5.5.1",
+    version="5.5.3",
     description="InsightConnect Plugin Runtime",
     long_description=long_description,
     long_description_content_type="text/markdown",