hcs-cli 0.1.319__tar.gz → 0.1.321__tar.gz

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hcs-cli
-Version: 0.1.319
+Version: 0.1.321
 Summary: Horizon Cloud Service CLI.
 Project-URL: Homepage, https://github.com/euc-eng/hcs-cli
 Project-URL: Bug Tracker, https://github.com/euc-eng/hcs-cli/issues
@@ -14,7 +14,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.9
-Requires-Dist: hcs-core>=0.1.319
+Requires-Dist: hcs-core>=0.1.321
 Requires-Dist: inquirerpy>=0.3.4
 Requires-Dist: matplotlib>=3.8.0
 Requires-Dist: paho-mqtt>=2.1.0

hcs_cli-0.1.321/hcs_cli/__init__.py

@@ -0,0 +1,3 @@
+__version__ = "0.1.321"
+
+from . import service as service  # noqa: F401

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/dev/fs/clear.py

@@ -13,12 +13,14 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
+import os
 import click
 
 import hcs_cli.cmds.dev.util.log as log
 from hcs_cli.cmds.dev.fs.helper.util import validate_fs_kubeconfig, validate_fs_profile
 from hcs_cli.cmds.dev.fs.init import _resolve_bundled_file_path
 from hcs_cli.support.exec_util import run_cli
+from hcs_core.ctxp import profile
 
 
 @click.command()
@@ -42,6 +44,7 @@ def clear(only_default: bool, **kwargs):
     if not only_default:
         _delete_all()
 
+    os.environ["ORG_ID"] = profile.current().csp.orgId
     akka_plan_path = _resolve_bundled_file_path("provided_files/akka.plan.yml")
     azure_plan_path = _resolve_bundled_file_path("provided_files/azure.plan.yml")
     azsim_plan_path = _resolve_bundled_file_path("provided_files/azsim.plan.yml")
@@ -90,3 +93,8 @@ def _delete_all():
     provider_ids = run_cli("hcs provider list --ids", output_json=True)
     for pid in provider_ids:
         run_cli("hcs provider delete -y " + pid, raise_on_error=False)
+
+    # all lcm internal providers
+    lcm_provider_ids = run_cli("hcs lcm provider list --ids", output_json=True)
+    for lpid in lcm_provider_ids:
+        run_cli("hcs lcm provider delete " + lpid, raise_on_error=False)

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/dev/fs/provided_files/mqtt-secret.yaml

@@ -75,7 +75,7 @@ stringData:
     JcSvTMU2HOuQ/q6vFZq3PRbvroMeXlw2WJrqO3vu04SZwYiP8dcrlHTXQJh+Mp9T
     mQ==
     -----END CERTIFICATE-----
-    mqtt.ca-key: |-
+  mqtt.ca-key: |-
     -----BEGIN PRIVATE KEY-----
     MIIEvAIBADANBgkqhkiG9w0BAQEFAASCBKYwggSiAgEAAoIBAQDUNJx2Cf0noNHc
     DXB9AFPSFMkkpwxb0wtqaCLiB7v5zkF7CVBl5LnecVk7CM8bJY62H90Lv8Zv/2sl
@@ -104,7 +104,7 @@ stringData:
     boPQ+RK1VKj5UFkUKz5LiohLmCBEQ7sh1HBS2Vje8MFtg0P138sRlbMCvdAEARTZ
     uQfAKLns7QFf9W1C60AVAQ==
     -----END PRIVATE KEY-----
-    mqtt.client-cert: |-
+  mqtt.client-cert: |-
     -----BEGIN CERTIFICATE-----
     MIIFKDCCAxCgAwIBAgIFALjj4O4wDQYJKoZIhvcNAQELBQAwdTELMAkGA1UEBhMC
     VVMxCzAJBgNVBAgMAkNBMR8wHQYDVQQKDBZIb3Jpem9uIENsb3VkIFNlcnZpY2Vz
@@ -204,7 +204,7 @@ stringData:
     JcSvTMU2HOuQ/q6vFZq3PRbvroMeXlw2WJrqO3vu04SZwYiP8dcrlHTXQJh+Mp9T
     mQ==
     -----END CERTIFICATE-----
-    mqtt.client-key: |-
+  mqtt.client-key: |-
     -----BEGIN PRIVATE KEY-----
     MIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQC40aC4pyy0+++D
     kD1cPRsMWO4ANZ8L4LaaHcg/e/1NbiExd/QlZESNJnnIyD29qY1h9WGEVDHtInsl

hcs_cli-0.1.321/hcs_cli/cmds/hoc/stats.py

@@ -0,0 +1,72 @@
+import click
+import datetime
+import json
+import time
+
+from hcs_cli.service.hoc import diagnostic
+from hcs_cli.service import tsctl
+
+
+@click.command(name="stats-aggregate-connects")
+@click.argument("from_", type=int)
+@click.argument("to", type=int)
+@click.option("--wait", type=bool, required=False, is_flag=True, default=False)
+@click.option("--verbose", type=bool, required=False, is_flag=True, default=False)
+def aggregate_connects(from_: int, to: int, wait: bool, verbose: bool):
+    """
+    Aggregate connect events between FROM_ and TO unix timestamps.
+    """
+    payload = {"from": from_, "to": to}
+    result = diagnostic.aggregateConnects(payload, verbose=verbose)
+    click.echo(result)
+
+    if not result:
+        return "", 1
+
+    if wait:
+        wait_for_task(result)
+
+
+@click.command(name="stats-aggregate-connects-day-before")
+@click.option("--wait", type=bool, required=False, is_flag=True, default=False)
+@click.option("--verbose", type=bool, required=False, is_flag=True, default=False)
+def aggregate_connects_day_before(wait: bool, verbose: bool):
+    dt = datetime.datetime.utcnow().replace(hour=0, minute=0, second=0, microsecond=0)
+    dt_2_days_ago = dt - datetime.timedelta(days=2)
+    dt_1_day_ago = dt - datetime.timedelta(days=1)
+
+    print(f"Aggregating connect events from {dt_2_days_ago} to {dt_1_day_ago}")
+    from_ = int(dt_2_days_ago.timestamp())
+    to = int(dt_1_day_ago.timestamp())
+    payload = {"from": from_, "to": to}
+    result = diagnostic.aggregateConnects(payload, verbose=verbose)
+    click.echo(result)
+    if not result:
+        return "", 1
+    if wait:
+        wait_for_task(result)
+
+
+def wait_for_task(result: dict):
+    print("Waiting for task to complete...")
+    totalWaitTime = 0
+    while True:
+        time.sleep(10)
+        taskStatus = tsctl.task.lastlog(result.namespace, result.group, result.taskKey)
+        if "state" in taskStatus and taskStatus.state == "Error":
+            print(f"Task is in {taskStatus.state} state - {taskStatus.error} ")
+            print(json.dumps(taskStatus, indent=2))
+            break
+        if "state" in taskStatus and taskStatus.state == "Success":
+            print("Task completed successfully.")
+            print(json.dumps(taskStatus, indent=2))
+            break
+        if ("state" in taskStatus and taskStatus.state == "") or "state" not in taskStatus:
+            print("Task is still in progress...")
+            totalWaitTime += 10
+            if totalWaitTime >= 600:
+                print("Waited for 10 minutes, exiting.")
+                break
+            continue
+        print(json.dumps(taskStatus, indent=2))
+        break

hcs_cli-0.1.321/hcs_cli/cmds/inventory/delete.py

@@ -0,0 +1,48 @@
+"""
+Copyright 2023-2023 VMware Inc.
+SPDX-License-Identifier: Apache-2.0
+
+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.
+"""
+
+import click
+from hcs_core.ctxp import recent
+import hcs_core.sglib.cli_options as cli
+
+from hcs_cli.service import inventory
+from hcs_cli.support.param_util import parse_vm_path
+
+
+@click.command()
+@cli.org_id
+@click.argument("vm_path", type=str, required=False)
+def delete(org: str, vm_path: str):
+    """Delete a VM by path, e.g., template1/vm1, or 'vm1,vm2,vm3'."""
+
+    org_id = cli.get_org_id(org)
+
+    if vm_path.find(",") > 0:
+        vm_ids = [v.strip() for v in vm_path.split(",")]
+        template_id = recent.require("template", None)
+    else:
+        template_id, vm_id = parse_vm_path(vm_path)
+        ret = inventory.get(template_id, vm_id, org_id)
+        if not ret:
+            return "", 1
+        vm_ids = [vm_id]
+
+    ret = {"_requested_ids": vm_ids, "_accepted_ids": [], "_deleted": 0}
+    vms = inventory.begin_deleting_vms_by_id(template_id, org_id, vm_ids)
+    for vm in vms:
+        ret["_accepted_ids"].append(vm["id"])
+    actual_deleted = inventory.finish_deleting_vms(template_id, org_id, vm_ids)
+    ret["_deleted"] = actual_deleted
+    return ret

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/lcm/provider/create.py

@@ -63,6 +63,12 @@ from hcs_cli.support import constant
     required=False,
     help="",
 )
+@click.option(
+    "--subscription-id",
+    type=str,
+    required=False,
+    help="",
+)
 @click.option(
     "--file",
     "-f",
@@ -80,6 +86,7 @@ def create(
     client_id: str,
     client_secret: str,
     tenant_id: str,
+    subscription_id: str,
     file: str,
     org: str,
     **kwargs,
@@ -108,7 +115,7 @@ def create(
             "description": description,
             "credentialId": credential_id,
         }
-    elif client_id or client_secret or tenant_id:
+    elif client_id or client_secret or tenant_id or subscription_id:
         if type != "AZURE":
             raise click.BadParameter("client-id is only supported for Azure provider.")
         if not client_id:
@@ -117,7 +124,8 @@ def create(
             raise click.BadParameter("client-secret is required for client-id.")
         if not tenant_id:
             raise click.BadParameter("tenant-id is required for Azure provider.")
-
+        if not subscription_id:
+            raise click.BadParameter("subscription-id is required for Azure provider.")
         data = {
             "id": id if id else _rand_id(8),
             "type": type,
@@ -126,6 +134,7 @@ def create(
             "description": description,
             "credentialId": credential_id,
             "tenantId": tenant_id,
+            "subscriptionId": subscription_id,
             "credentials": [
                 {
                     "clientId": client_id,

hcs_cli-0.1.321/hcs_cli/cmds/lcm/template/expand.py

@@ -0,0 +1,46 @@
+"""
+Copyright 2023-2023 VMware Inc.
+SPDX-License-Identifier: Apache-2.0
+
+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.
+"""
+
+import click
+import hcs_core.sglib.cli_options as cli
+
+import hcs_cli.service.lcm as lcm
+from hcs_cli.cmds.template.expand import expand_impl
+
+
+@click.command(hidden=True)
+@click.option(
+    "--number",
+    "-n",
+    type=int,
+    required=False,
+    default=0,
+    help="Number of VMs to expand. Use negative number to shrink.",
+)
+@click.option(
+    "--to",
+    "-t",
+    type=int,
+    required=False,
+    default=0,
+    help="Expected size of template.",
+)
+@click.argument("template_id", type=str, required=False)
+@cli.org_id
+@cli.wait
+def expand(number: int, to: int, template_id: str, org: str, wait: str):
+    """Update an existing template"""
+
+    return expand_impl(number, to, template_id, org, wait, lcm)

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/lcm/vm/delete.py

@@ -24,8 +24,9 @@ from hcs_cli.support.param_util import parse_vm_path
 @click.argument("vm_path", type=str, required=False)
 @cli.org_id
 @cli.confirm
+@cli.force
 @cli.wait
-def delete(vm_path: str, org: str, confirm: bool, wait: str, **kwargs):
+def delete(vm_path: str, org: str, confirm: bool, force: bool, wait: str, **kwargs):
     """Delete VM"""
     template_id, vm_id = parse_vm_path(vm_path)
     org_id = cli.get_org_id(org)
@@ -37,7 +38,7 @@ def delete(vm_path: str, org: str, confirm: bool, wait: str, **kwargs):
     if not confirm:
         click.confirm(f"Delete VM {template_id}/{vm_id}?", abort=True)
 
-    vm.delete(template_id, vm_id, org_id, **kwargs)
+    vm.delete(template_id, vm_id, org_id, force=force, **kwargs)
 
     if wait == "0":
         return

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/task.py

@@ -222,6 +222,7 @@ def use(namespace: str, group: str, smart_path: str, reset: bool, **kwargs):
     required=False,
     help="Filter tasks by state, for one-time tasks only. Comma separated values. E.g. Init,Running,Canceled,Error,Success",
 )
+@click.option("--v1", is_flag=True, default=False, required=False, help="Use v1 task API.")
 @click.option("--v1log", is_flag=True, default=False, required=False, help="Fetch last log for v1 tasks.")
 @cli.limit
 @cli.search

hcs_cli-0.1.321/hcs_cli/cmds/template/expand.py

@@ -0,0 +1,110 @@
+"""
+Copyright 2023-2023 VMware Inc.
+SPDX-License-Identifier: Apache-2.0
+
+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.
+"""
+
+import click
+import hcs_core.sglib.cli_options as cli
+import hcs_core.util.duration as duration
+from hcs_core.ctxp import recent
+
+import hcs_cli.service.admin as admin
+
+
+@click.command(hidden=True)
+@click.option(
+    "--number",
+    "-n",
+    type=int,
+    required=False,
+    default=0,
+    help="Number of VMs to expand. Use negative number to shrink.",
+)
+@click.option(
+    "--to",
+    "-t",
+    type=int,
+    required=False,
+    default=0,
+    help="Expected size of template.",
+)
+@click.argument("template_id", type=str, required=False)
+@cli.org_id
+@cli.wait
+def expand(number: int, to: int, template_id: str, org: str, wait: str):
+    """Update an existing template"""
+    return expand_impl(number, to, template_id, org, wait, admin)
+
+
+def expand_impl(number: int, to: int, template_id: str, org: str, wait: str, service):
+    if to != 0 and number != 0:
+        return "Specify either --to or --number, not both.", 1
+    if to < 0:
+        return "--to must be non-negative.", 1
+    if to > 5000:
+        return "--to exceeds maximum limit of 5000.", 1
+
+    org_id = cli.get_org_id(org)
+
+    template_id = recent.require("template", template_id)
+    template = service.template.get(template_id, org_id)
+
+    if not template:
+        return "Template not found: " + template_id, 1
+
+    spare_policy = template.get("sparePolicy", {})
+    new_spare_policy = {
+        "min": spare_policy.get("min", 0),
+        "max": spare_policy.get("max", 0),
+        "limit": spare_policy.get("limit", 0),
+    }
+
+    if to != 0:
+        new_spare_policy["min"] = to
+        new_spare_policy["max"] = to
+        new_spare_policy["limit"] = to
+    else:
+        if number == 0:
+            number = 1  # default to expand by 1
+        new_spare_policy["min"] += number
+        new_spare_policy["max"] += number
+        new_spare_policy["limit"] += number
+
+        if new_spare_policy["limit"] < 0:
+            new_spare_policy["limit"] = 0
+        elif new_spare_policy["limit"] > 5000:
+            new_spare_policy["limit"] = 5000
+
+        if new_spare_policy["min"] < 0:
+            new_spare_policy["min"] = 0
+        elif new_spare_policy["min"] > new_spare_policy["limit"]:
+            new_spare_policy["min"] = new_spare_policy["limit"]
+        if new_spare_policy["max"] < 0:
+            new_spare_policy["max"] = 0
+        elif new_spare_policy["max"] > new_spare_policy["limit"]:
+            new_spare_policy["max"] = new_spare_policy["limit"]
+
+    if (
+        new_spare_policy["min"] == spare_policy.get("min", 0)
+        and new_spare_policy["max"] == spare_policy.get("max", 0)
+        and new_spare_policy["limit"] == spare_policy.get("limit", 0)
+    ):
+        # no change
+        ret = template
+    else:
+        patch = {"sparePolicy": new_spare_policy}
+        ret = service.template.patch(template_id, org_id, patch)
+
+    if wait != "0":
+        ret = service.template.wait_for_ready(template_id, org_id, duration.to_seconds(wait))
+    return ret

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/template/update.py

@@ -43,7 +43,7 @@ def update(template_id: str, update, org: str, wait: str, **kwargs):
     template = admin.template.get(template_id, org_id)
 
     if not template:
-        return "Template not found: " + template_id
+        return "Template not found: " + template_id, 1
 
     allowed_fields = ["name", "description", "powerPolicy", "sparePolicy", "applicationProperties", "flags"]
 
@@ -52,7 +52,7 @@ def update(template_id: str, update, org: str, wait: str, **kwargs):
     if not patch:
         return template
 
-    ret = admin.template.update(template_id, org_id, patch)
+    ret = admin.template.patch(template_id, org_id, patch)
     if wait != "0":
         ret = admin.template.wait_for_ready(template_id, org_id, duration.to_seconds(wait))
     return ret

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/vm/delete.py

@@ -23,10 +23,11 @@ from hcs_cli.support.param_util import parse_vm_path
 
 @click.command()
 @click.argument("vm_path", type=str, required=False)
+@cli.force
 @cli.org_id
 @cli.confirm
 @cli.wait
-def delete(vm_path: str, org: str, confirm: bool, wait: str, **kwargs):
+def delete(vm_path: str, org: str, confirm: bool, force: bool, wait: str, **kwargs):
     """Delete a VM by ID"""
     org_id = cli.get_org_id(org)
     template_id, vm_id = parse_vm_path(vm_path)
@@ -41,7 +42,7 @@ def delete(vm_path: str, org: str, confirm: bool, wait: str, **kwargs):
     if not confirm:
         click.confirm(f"Delete vm {existing['id']} from template {template['name']} ({template['id']})?", abort=True)
 
-    vm.delete()
+    vm.delete(force=force)
 
     if wait == "0":
         return

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/vm/list.py

@@ -47,47 +47,25 @@ from hcs_cli.support.vm_table import format_vm_table
 @click.option(
     "--agent",
     "-a",
-    type=click.Choice(
-        ["UNAVAILABLE", "ERROR", "AVAILABLE", "INIT", "UNKNOWN", "DOMAIN_ERR", "CUSTOMIZATION_FAILURE"],
-        case_sensitive=False,
-    ),
     required=False,
-    multiple=True,
-    help="Filter VMs by agent status.",
+    help="Filter VMs by agent status, in comma separated values: UNAVAILABLE,ERROR,AVAILABLE,INIT,UNKNOWN,DOMAIN_ERR,CUSTOMIZATION_FAILURE",
 )
 @click.option(
     "--power",
     "-p",
-    type=click.Choice(["PoweredOn", "PoweredOff", "PoweringOn", "PoweringOff", "Unknown"], case_sensitive=False),
     required=False,
-    multiple=True,
-    help="Filter VMs by power state.",
+    help="Filter VMs by power state, in comman separated values: PoweredOn,PoweredOff,PoweringOn,PoweringOff,Unknown",
 )
 @click.option(
     "--lifecycle",
-    type=click.Choice(
-        [
-            "PROVISIONING",
-            "PROVISIONED",
-            "MAINTENANCE",
-            "DELETING",
-            "ERROR",
-            "CUSTOMIZING",
-            "AGENT_UPDATING",
-            "AGENT_REINSTALLING",
-        ],
-        case_sensitive=False,
-    ),
+    type=str,
     required=False,
-    multiple=True,
-    help="Filter VMs by lifecycle status.",
+    help="Filter VMs by lifecycle status, in comma separated values: PROVISIONING,PROVISIONED,MAINTENANCE,DELETING,ERROR,CUSTOMIZING,AGENT_UPDATING,AGENT_REINSTALLING",
 )
 @click.option(
     "--session",
-    type=click.Choice(["AVAILABLE", "UNAVAILABLE", "DRAINING", "QUIESCING", "REPRISTINING"], case_sensitive=False),
     required=False,
-    multiple=False,
-    help="Filter VMs by power state.",
+    help="Filter VMs by session placement status, in comma separated values: AVAILABLE,UNAVAILABLE,DRAINING,QUIESCING,REPRISTINING",
 )
 @cli.formatter(format_vm_table)
 def list_vms(
@@ -97,33 +75,34 @@ def list_vms(
     type: list,
     agent: list,
     power: list,
-    lifecycle: list,
+    lifecycle: str,
     session: str,
     **kwargs,
 ):
     """List template VMs"""
     org_id = cli.get_org_id(org)
 
-    agent = _to_lower(agent)
-    power = _to_lower(power)
-    lifecycle = _to_lower(lifecycle)
+    filttered_agent_states = _parse_multi_state(agent, "agent")
+    filtered_power_states = _parse_multi_state(power, "power")
+    filtered_lifecycle_states = _parse_multi_state(lifecycle, "lifecycle")
+    filtered_session_states = _parse_multi_state(session, "session")
 
     def filter_vm(vm):
-        if agent:
+        if filttered_agent_states:
             s = vm.get("agentStatus")
-            if not s or s.lower() not in agent:
+            if not s or s.upper() not in filttered_agent_states:
                 return False
-        if power:
+        if filtered_power_states:
             s = vm.get("powerState")
-            if not s or s.lower() not in power:
+            if not s or s.upper() not in filtered_power_states:
                 return False
-        if lifecycle:
+        if filtered_lifecycle_states:
             s = vm.get("lifecycleStatus")
-            if not s or s.lower() not in lifecycle:
+            if not s or s.upper() not in filtered_lifecycle_states:
                 return False
-        if session:
+        if filtered_session_states:
             s = vm.get("sessionPlacementStatus")
-            if not s or s.lower() != session.lower():
+            if not s or s.upper() not in filtered_session_states:
                 return False
         return True
 
@@ -178,6 +157,39 @@ def list_vms(
     return vms
 
 
+def _parse_multi_state(param_list: str, name: str):
+    if not param_list:
+        return None
+    param_list = param_list.upper()
+    values = []
+
+    valid_map = {
+        "lifecycle": [
+            "PROVISIONING",
+            "PROVISIONED",
+            "MAINTENANCE",
+            "DELETING",
+            "ERROR",
+            "CUSTOMIZING",
+            "AGENT_UPDATING",
+            "AGENT_REINSTALLING",
+        ],
+        "agent": ["UNAVAILABLE", "ERROR", "AVAILABLE", "INIT", "UNKNOWN", "DOMAIN_ERR", "CUSTOMIZATION_FAILURE"],
+        "power": ["POWEREDON", "POWEREDOFF", "POWERINGON", "POWERINGOFF", "UNKNOWN"],
+        "session": ["AVAILABLE", "UNAVAILABLE", "DRAINING", "QUIESCING", "REPRISTINING"],
+    }
+    if name not in valid_map:
+        raise Exception(f"Unknown multi-state filter name: {name}")
+
+    valid = valid_map[name]
+    for s in param_list.split(","):
+        s = s.strip()
+        if s not in valid:
+            raise Exception(f"Invalid {name} value: {s}. Valid values are: {', '.join(valid)}")
+        values.append(s)
+    return values
+
+
 def _to_lower(values):
     if values:
         return [v.lower() for v in values]

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/cmds/vmm/rootca_migrate.py

@@ -556,7 +556,7 @@ def power_policy_on_template(template: dict, enable: bool, verbose: bool):
     patch = calculate_patch(template, allowed_fields, [field])
     for i in range(3):
         try:
-            admin.template.update(template_id, target_org_id, patch)
+            admin.template.patch(template_id, target_org_id, patch)
             admin.template.wait_for_ready(template_id, target_org_id, 60)
             break
         except Exception as ex:

hcs_cli-0.1.319/hcs_cli/payload/lcm/azure-dummy-nt.json → hcs_cli-0.1.321/hcs_cli/payload/lcm/azure-dummy.json

@@ -43,9 +43,9 @@
   },
   "image": {
     "imageReference": {
-      "publisher": "Canonical",
-      "offer": "UbuntuServer",
-      "sku": "18.04-LTS",
+      "publisher": "canonical",
+      "offer": "0001-com-ubuntu-server-jammy",
+      "sku": "22_04-lts-gen2",
       "version": "latest"
     },
     "os": "linux",
@@ -62,6 +62,7 @@
   "properties": {
     "azureCreateResourceGroup": true,
     "new-task-engine": true,
-    "simulateAgentStatus": "AVAILABLE"
+    "simulateAgentStatus": "AVAILABLE",
+    "countinuousErrorThreshold": 3
   }
 }

{hcs_cli-0.1.319 → hcs_cli-0.1.321}/hcs_cli/payload/lcm/zero.json

@@ -56,8 +56,9 @@
     }
   },
   "properties": {
-    "new-task-engine": false,
+    "new-task-engine": true,
     "sg-lcm-enforced-provision": true,
-    "autonomous": true
+    "autonomous": true,
+    "countinuousErrorThreshold": 2
   }
 }