utf-queue-client 1.12.4__tar.gz → 1.13.0b1__tar.gz

{utf-queue-client-1.12.4 → utf-queue-client-1.13.0b1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.2
 Name: utf-queue-client
-Version: 1.12.4
+Version: 1.13.0b1
 Summary: UNKNOWN
 Home-page: UNKNOWN
 Author: Joe Savage

{utf-queue-client-1.12.4 → utf-queue-client-1.13.0b1}/setup.py

@@ -6,7 +6,7 @@
 from setuptools import setup, find_packages  # noqa: H301
 
 NAME = "utf-queue-client"
-VERSION = "1.12.4"
+VERSION = "1.13.0b1"
 # To install the library, run the following
 #
 # python setup.py install
@@ -53,6 +53,7 @@ setup(
             "ubai_search_cli = utf_queue_client.scripts.ubai_search_cli:cli_entrypoint",
             "publish_test_results_cli = utf_queue_client.scripts.publish_test_results_cli:cli_entrypoint",
             "ubai_update_retention_cli = utf_queue_client.scripts.ubai_update_retention_cli:cli_entrypoint",
+            "publish_release_info_cli = utf_queue_client.scripts.publish_release_info_cli:cli_entrypoint"
         ],
     },
     include_package_data=True,

utf-queue-client-1.13.0b1/test/test_publish_release_info.py

@@ -0,0 +1,72 @@
+import pytest
+from otel_extensions import inject_context_to_env, instrumented
+import os
+import subprocess
+import sys
+from utf_queue_client.scripts.publish_release_info_cli import publish_release_information, update_artifacts_retention_days_to_keep_forever
+import requests
+from unittest import mock
+from unittest.mock import MagicMock, patch
+
+
+@pytest.fixture
+def publish_release_info():
+    yield [
+        ("releaseDate", "2023-06-16"),
+        ("branchName", "release/gecko_sdk_3.2"),
+        ("releaseType", "GA"),
+        ("releaseName", "23Q2-GA"),
+        ("SDKBuildNum", "407"),
+        ("installerURLSDK", "https://jenkins-cbs-gecko-sdk.silabs.net/job/Gecko_SDK_Suite_super/job/release%252F21q4/284/"),
+        ("IARVersionNum", "9.20.4"),
+        ("GCCVersionNum", "10.3.1"),
+        ("CommanderVersionNum", "v1.8.1"),
+        ("SEVersionNum", "1.2.4"),
+        ("WSTKFirmwareVersionNum", "v1.4.4")
+    ]
+
+
+@instrumented
+def test_publish_release_info(request, publish_release_info):
+    args = ["--username", "svc_sqabot", "--password", "password"]
+    for k, v in publish_release_info:
+        args += ["--data", k, v]
+
+    base_dir = os.path.dirname(os.path.dirname(__file__))
+    @inject_context_to_env
+    def call_cli_script():
+        assert "TRACEPARENT" in os.environ
+        process = subprocess.Popen(
+            [
+                sys.executable,
+                os.path.join(
+                    base_dir, "utf_queue_client", "scripts", "publish_release_info_cli.py"
+                ),
+            ]
+            + args,
+        )
+        process.communicate()
+        assert process.poll() == 0
+
+    call_cli_script()
+
+
+def test_publish_release_information():
+    requests_post = requests.post
+    requests.post = MagicMock()
+    publish_release_information("", "", {})
+    assert requests.post.call_count == 2
+    requests.post = requests_post
+
+
+def test_update_artifacts_retention_days_to_keep_forever():
+    requests_patch = requests.patch
+    requests.patch = MagicMock()
+    data_dict = {"branchName": "release/gec", "SDKBuildNum": "407"}
+    update_artifacts_retention_days_to_keep_forever(data_dict)
+    assert requests.patch.call_count == 1
+    requests.patch = requests_patch
+    patch_response = MagicMock()
+    mocked_patch =mock.patch.object(requests, "patch", return_value=patch_response)
+    patch_response.status_code = 200
+    update_artifacts_retention_days_to_keep_forever(data_dict)

utf-queue-client-1.13.0b1/utf_queue_client/scripts/publish_release_info_cli.py

@@ -0,0 +1,122 @@
+import click
+from utf_queue_client.scripts import setup_telemetry
+from retry import retry
+from typing import Iterable, Tuple
+from otel_extensions import instrumented
+import requests
+
+NUM_RETRIES = 6
+
+
+@click.command()
+@click.option(
+    "--username",
+    type=str,
+    required=True,
+    help="username",
+)
+@click.option(
+    "--password",
+    type=str,
+    required=True,
+    help="password",
+)
+@click.option(
+    "--data",
+    multiple=True,
+    type=(str, str),
+    required=True,
+    help="data in dictionary format",
+)
+def cli_entrypoint(username, password, data):
+    cli(username, password, data)
+
+
+def cli(username: str, password: str, data: Iterable[Tuple[str, str]]):
+    with setup_telemetry():
+        data_dict = {}
+        for key, value in data:
+            data_dict[key] = value
+
+        @retry(Exception, delay=5, backoff=2, max_delay=30, tries=NUM_RETRIES + 1)
+        def retry_wrapper():
+            response = publish_release_information(username, password, data_dict)
+            if response is not None:
+                update_artifacts_retention_days_to_keep_forever(data_dict)
+        retry_wrapper()
+
+
+@instrumented
+def publish_release_information(username: str, password: str, data_dict: dict):
+    url = 'https://iot-reports.silabs.net/api/v1/auth'
+
+    # Create a dictionary with the username and password
+    credentials = {
+        'username': username,
+        'password': password
+    }
+
+    try:
+        response = requests.post(url, json=credentials, verify=False)
+        response.raise_for_status()  # Raise an exception for 4xx/5xx status codes
+
+        # Extract the JWT token from the response
+        jwt_token = response.json()['access_token']
+
+    except requests.exceptions.RequestException as e:
+        print(f"Error: {e}")
+        return None
+
+    url = 'https://releaseinfo.silabs.net/api/v1/releases'
+
+    # Create the Authorization header with the Bearer token
+    headers = {
+        'Authorization': f'JWT {jwt_token}'
+    }
+
+    # Send the POST request with the headers
+    response = requests.post(url, json=data_dict, headers=headers, verify=False)
+
+    # Check the response status code
+    if response.status_code == 201:
+        # Request successful
+        print('Publish release information successful!')
+        print('Response:', response.json())
+        return True
+    else:
+        # Request failed
+        print('Publish release information failed.')
+        print('Status Code:', response.status_code)
+        return None
+
+
+@instrumented
+def update_artifacts_retention_days_to_keep_forever(data_dict: dict):
+    url = 'https://ubai.silabs.net/api/v1/artifacts'
+
+    artifact_search_spec = {
+        "metadata": {
+            "branch": data_dict["branchName"],
+            "build_number": data_dict["SDKBuildNum"]
+        },
+        "retentionDays": -1
+    }
+
+    # Send the PATCH request with the headers
+    response = requests.patch(url, json=artifact_search_spec, verify=False)
+
+    # Check the response status code
+    if response.status_code == 200:
+        # Request successful
+        print("Updated Retention days in UBAI for release artifact to keep forever!")
+        print("Response:", response.json())
+    else:
+        # Request failed
+        print('Update Retention days in UBAI for release artifact to keep forever failed.')
+        print('Status Code:', response.status_code)
+        print('Response:', response.json())
+        return None
+
+
+if __name__ == "__main__":
+    cli_entrypoint()

{utf-queue-client-1.12.4 → utf-queue-client-1.13.0b1}/utf_queue_client/scripts/publish_test_results_cli.py

@@ -137,6 +137,7 @@ def prepare_session_data(data_dict: dict):
         jenkinsTestResultsUrl=data_dict.get("jenkinsTestResultsUrl"),
         traceId=data_dict.get("traceId"),
         testFramework=data_dict.get("testFramework"),
+        SDKVersion=data_dict.get("SDKVersion")
     )
 
 

{utf-queue-client-1.12.4 → utf-queue-client-1.13.0b1}/utf_queue_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.2
 Name: utf-queue-client
-Version: 1.12.4
+Version: 1.13.0b1
 Summary: UNKNOWN
 Home-page: UNKNOWN
 Author: Joe Savage

{utf-queue-client-1.12.4 → utf-queue-client-1.13.0b1}/utf_queue_client.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ test/test_data_lake_data_producer.py
 test/test_legacy_base_model.py
 test/test_models.py
 test/test_opentelemetry_data_producer.py
+test/test_publish_release_info.py
 test/test_publish_test_results.py
 test/test_sqa_test_result_producer.py
 test/test_ubai_search.py
@@ -31,6 +32,7 @@ utf_queue_client/models/base_model.py
 utf_queue_client/models/model_factory.py
 utf_queue_client/models/schemas/__init__.py
 utf_queue_client/scripts/__init__.py
+utf_queue_client/scripts/publish_release_info_cli.py
 utf_queue_client/scripts/publish_test_results_cli.py
 utf_queue_client/scripts/ubai_search_cli.py
 utf_queue_client/scripts/ubai_update_retention_cli.py

{utf-queue-client-1.12.4 → utf-queue-client-1.13.0b1}/utf_queue_client.egg-info/entry_points.txt

@@ -1,4 +1,5 @@
 [console_scripts]
+publish_release_info_cli = utf_queue_client.scripts.publish_release_info_cli:cli_entrypoint
 publish_test_results_cli = utf_queue_client.scripts.publish_test_results_cli:cli_entrypoint
 ubai_search_cli = utf_queue_client.scripts.ubai_search_cli:cli_entrypoint
 ubai_update_retention_cli = utf_queue_client.scripts.ubai_update_retention_cli:cli_entrypoint