11from conftest import roundtrip_check , connect_resources
22from 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
44from fix_plugin_azure .resource .compute import AzureDiskEncryptionSet , AzureVirtualMachineScaleSet
55from 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
2930def 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
0 commit comments