Skip to content

Commit 043ad3a

Browse files
authored
[azure][feat] Align names of the azure resources (#2079)
1 parent 8fb295a commit 043ad3a

File tree

7 files changed

+23
-22
lines changed

7 files changed

+23
-22
lines changed

plugins/azure/fix_plugin_azure/collector.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
AzureNetworkUsage,
2525
resources as network_resources,
2626
)
27-
from fix_plugin_azure.resource.storage import AzureStorageUsage, AzureStorageSku, resources as storage_resources
27+
from fix_plugin_azure.resource.storage import AzureStorageAccountUsage, AzureStorageSku, resources as storage_resources
2828
from fix_plugin_azure.resource.containerservice import resources as aks_resources
2929
from fixlib.baseresources import Cloud, GraphRoot
3030
from fixlib.core.actions import CoreFeedback, ErrorAccumulator
@@ -175,7 +175,7 @@ def rm_nodes(cls, ignore_kinds: Optional[Type[Any]] = None) -> None: # type: ig
175175

176176
def remove_usage_zero_value() -> None:
177177
for node in self.graph.nodes:
178-
if not isinstance(node, (AzureNetworkUsage, AzureStorageUsage)):
178+
if not isinstance(node, (AzureNetworkUsage, AzureStorageAccountUsage)):
179179
continue
180180
# Azure Usage just keep info about how many kind of resources on account exists
181181
# Check if the current usage value of the Azure Usage node is 0

plugins/azure/fix_plugin_azure/resource/compute.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2113,8 +2113,8 @@ class AzureCopyCompletionError:
21132113

21142114

21152115
@define(eq=False, slots=False)
2116-
class AzureSnapshot(AzureResource, BaseSnapshot):
2117-
kind: ClassVar[str] = "azure_snapshot"
2116+
class AzureVirtualMachineSnapshot(AzureResource, BaseSnapshot):
2117+
kind: ClassVar[str] = "azure_virtual_machine_snapshot"
21182118
api_spec: ClassVar[AzureApiSpec] = AzureApiSpec(
21192119
service="compute",
21202120
version="2023-01-02",
@@ -3741,7 +3741,7 @@ class AzureVirtualMachineScaleSetInstance(AzureVirtualMachineBase):
37413741
AzureImage,
37423742
AzureProximityPlacementGroup,
37433743
AzureRestorePointCollection,
3744-
AzureSnapshot,
3744+
AzureVirtualMachineSnapshot,
37453745
AzureSshPublicKeyResource,
37463746
AzureVirtualMachine,
37473747
AzureVirtualMachineScaleSet,

plugins/azure/fix_plugin_azure/resource/containerservice.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -994,8 +994,8 @@ class AzureOSOptionProperty:
994994

995995

996996
@define(eq=False, slots=False)
997-
class AzureKubernetesSnapshot(AzureResource):
998-
kind: ClassVar[str] = "azure_kubernetes_snapshot"
997+
class AzureManagedClusterSnapshot(AzureResource):
998+
kind: ClassVar[str] = "azure_managed_cluster_snapshot"
999999
api_spec: ClassVar[AzureApiSpec] = AzureApiSpec(
10001000
service="containerservice",
10011001
version="2023-08-01",
@@ -1037,4 +1037,4 @@ def connect_in_graph(self, builder: GraphBuilder, source: Json) -> None:
10371037
builder.add_edge(self, edge_type=EdgeType.default, reverse=True, clazz=AzureManagedCluster, id=cluster_id)
10381038

10391039

1040-
resources: List[Type[AzureResource]] = [AzureManagedCluster, AzureFleet, AzureKubernetesSnapshot]
1040+
resources: List[Type[AzureResource]] = [AzureManagedCluster, AzureFleet, AzureManagedClusterSnapshot]

plugins/azure/fix_plugin_azure/resource/storage.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -174,8 +174,8 @@ class AzureBlobContainer(AzureResource, BaseBucket):
174174

175175

176176
@define(eq=False, slots=False)
177-
class AzureDeletedAccount(AzureResource):
178-
kind: ClassVar[str] = "azure_deleted_account"
177+
class AzureStorageAccountDeleted(AzureResource):
178+
kind: ClassVar[str] = "azure_storage_account_deleted"
179179
api_spec: ClassVar[AzureApiSpec] = AzureApiSpec(
180180
service="storage",
181181
version="2023-01-01",
@@ -1150,8 +1150,8 @@ def collect_usage_metrics(
11501150

11511151

11521152
@define(eq=False, slots=False)
1153-
class AzureStorageUsage(AzureResource, AzureBaseUsage):
1154-
kind: ClassVar[str] = "azure_storage_usage"
1153+
class AzureStorageAccountUsage(AzureResource, AzureBaseUsage):
1154+
kind: ClassVar[str] = "azure_storage_account_usage"
11551155
api_spec: ClassVar[AzureApiSpec] = AzureApiSpec(
11561156
service="storage",
11571157
version="2023-01-01",
@@ -1208,12 +1208,12 @@ class AzureTable(AzureResource):
12081208

12091209

12101210
resources: List[Type[AzureResource]] = [
1211-
AzureDeletedAccount,
1211+
AzureStorageAccountDeleted,
12121212
AzureStorageSku,
12131213
AzureFileShare,
12141214
AzureQueue,
12151215
AzureBlobContainer,
12161216
AzureTable,
12171217
AzureStorageAccount,
1218-
AzureStorageUsage,
1218+
AzureStorageAccountUsage,
12191219
]

plugins/azure/test/compute_test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -180,17 +180,17 @@ def test_virtual_machine_size_resources(builder: GraphBuilder) -> None:
180180

181181

182182
def test_snapshot(builder: GraphBuilder) -> None:
183-
collected = roundtrip_check(AzureSnapshot, builder)
183+
collected = roundtrip_check(AzureVirtualMachineSnapshot, builder)
184184
assert len(collected) == 2
185185

186186
resource_type: List[Type[AzureResource]] = [AzureDisk]
187187
connect_resources(builder, resource_type)
188188

189-
assert len(builder.edges_of(AzureDisk, AzureSnapshot)) == 1
189+
assert len(builder.edges_of(AzureDisk, AzureVirtualMachineSnapshot)) == 1
190190

191191

192192
def test_snapshot_resources(builder: GraphBuilder) -> None:
193-
collected = roundtrip_check(AzureSnapshot, builder)[1]
193+
collected = roundtrip_check(AzureVirtualMachineSnapshot, builder)[1]
194194
assert collected.snapshot_status == "None"
195195
assert (
196196
collected.volume_id

plugins/azure/test/containerservice_test.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from conftest import roundtrip_check, connect_resources
22
from fix_plugin_azure.resource.base import GraphBuilder, AzureResource
3-
from fix_plugin_azure.resource.containerservice import AzureFleet, AzureManagedCluster, AzureKubernetesSnapshot
3+
from fix_plugin_azure.resource.containerservice import AzureFleet, AzureManagedCluster, AzureManagedClusterSnapshot
44
from fix_plugin_azure.resource.compute import AzureDiskEncryptionSet, AzureVirtualMachineScaleSet
55
from typing import List, Type
66

@@ -10,8 +10,9 @@ def test_fleet(builder: GraphBuilder) -> None:
1010
assert len(collected) == 1
1111

1212
resource_types: List[Type[AzureResource]] = [AzureManagedCluster]
13-
connect_resources(builder, resource_types)
13+
roundtrip_check(AzureManagedCluster, builder)
1414

15+
connect_resources(builder, resource_types)
1516
assert len(builder.edges_of(AzureFleet, AzureManagedCluster)) == 1
1617

1718

@@ -27,10 +28,10 @@ def test_managed_cluster(builder: GraphBuilder) -> None:
2728

2829

2930
def test_kub_snapshot(builder: GraphBuilder) -> None:
30-
collected = roundtrip_check(AzureKubernetesSnapshot, builder)
31+
collected = roundtrip_check(AzureManagedClusterSnapshot, builder)
3132
assert len(collected) == 1
3233

3334
resource_types: List[Type[AzureResource]] = [AzureManagedCluster]
3435
connect_resources(builder, resource_types)
3536

36-
assert len(builder.edges_of(AzureManagedCluster, AzureKubernetesSnapshot)) == 1
37+
assert len(builder.edges_of(AzureManagedCluster, AzureManagedClusterSnapshot)) == 1

plugins/azure/test/storage_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55

66
def test_deleted_account(builder: GraphBuilder) -> None:
7-
collected = roundtrip_check(AzureDeletedAccount, builder)
7+
collected = roundtrip_check(AzureStorageAccountDeleted, builder)
88
assert len(collected) == 2
99

1010

0 commit comments

Comments
 (0)