Skip to content

Commit 4f4c670

Browse files
committed
Update unit tests
1 parent b6f6b2e commit 4f4c670

File tree

3 files changed

+17
-12
lines changed

3 files changed

+17
-12
lines changed

cli/src/pcluster/api/controllers/cluster_operations_controller.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -461,7 +461,9 @@ def _analyze_changes(changes):
461461
message = _create_message(row[key_indexes["reason"]], row[key_indexes["action_needed"]])
462462
if not _cluster_update_change_succeded(check_result):
463463
errors.append(
464-
UpdateError(parameter=parameter, requested_value=str(new_value), message=message, current_value=str(old_value))
464+
UpdateError(
465+
parameter=parameter, requested_value=str(new_value), message=message, current_value=str(old_value)
466+
)
465467
)
466468
change_set.append(Change(parameter=parameter, requested_value=str(new_value), current_value=str(old_value)))
467469
return change_set, errors

cli/tests/pcluster/api/controllers/test_cluster_operations_controller.py

Lines changed: 14 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1969,9 +1969,9 @@ def test_dryrun(self, mocker, client, errors):
19691969
"message": "Request would have succeeded, but DryRun flag is set.",
19701970
"changeSet": [
19711971
{
1972-
"currentValue": 10,
1972+
"currentValue": "10",
19731973
"parameter": "Scheduling.SlurmQueues[queue0].ComputeResources[queue0-i0].MaxCount",
1974-
"requestedValue": 11,
1974+
"requestedValue": "11",
19751975
}
19761976
],
19771977
}
@@ -2460,10 +2460,12 @@ def test_cluster_update_change_succeded(check_result):
24602460
],
24612461
],
24622462
"-",
2463-
{
2464-
"ComputeResources": [{"InstanceType": "c5.9xlarge", "MinCount": 0, "Name": "compute2"}],
2465-
"Name": "queue2",
2466-
},
2463+
str(
2464+
{
2465+
"Name": "queue2",
2466+
"ComputeResources": [{"Name": "compute2", "InstanceType": "c5.9xlarge", "MinCount": 0}],
2467+
}
2468+
),
24672469
),
24682470
(
24692471
[
@@ -2491,10 +2493,12 @@ def test_cluster_update_change_succeded(check_result):
24912493
"COMPUTE_FLEET_STOP_ON_REMOVE",
24922494
],
24932495
],
2494-
{
2495-
"ComputeResources": [{"InstanceType": "c5.xlarge", "MinCount": 0, "Name": "compute1"}],
2496-
"Name": "queue1",
2497-
},
2496+
str(
2497+
{
2498+
"Name": "queue1",
2499+
"ComputeResources": [{"Name": "compute1", "InstanceType": "c5.xlarge", "MinCount": 0}],
2500+
}
2501+
),
24982502
"-",
24992503
),
25002504
],

tests/integration-tests/tests/pcluster_api/test_api.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
from botocore.config import Config
2424
from cfn_stacks_factory import CfnStack
2525
from clusters_factory import Cluster, ClustersFactory
26-
from pcluster_client import ApiException
2726
from pcluster_client.api import (
2827
cluster_compute_fleet_api,
2928
cluster_instances_api,

0 commit comments

Comments
 (0)