fix

Author: olasaadi99

pkg/vpcmodel/subnetsConnectivity.go

@@ -241,10 +241,21 @@ func updateSubnetsConnectivityByTransitGateway(src, dst VPCResourceIntf,
 	c *VPCConfig) (
 	*netset.TransportSet, error) {
 	// assuming a single router representing the tgw for a "MultipleVPCsConfig"
-	if len(c.RoutingResources) != 2 { // expecting tgw and sgw (virtual gateway)
-		return nil, fmt.Errorf("unexpected number of RoutingResources for MultipleVPCsConfig, expecting only TGW")
+	resourceTypeTGW := "TGW"
+	var tgw RoutingResource
+	tgwRouterFound := false
+	for _, router := range c.RoutingResources {
+		if router.Kind() == resourceTypeTGW {
+			if tgwRouterFound {
+				return nil, fmt.Errorf("unexpected number of RoutingResources for MultipleVPCsConfig, expecting only one TGW")
+			}
+			tgw = router
+			tgwRouterFound = true
+		}
+	}
+	if !tgwRouterFound {
+		return nil, fmt.Errorf("unexpected number of RoutingResources for MultipleVPCsConfig, expecting TGW")
 	}
-	tgw := c.RoutingResources[0]
 	connections, err := tgw.AllowedConnectivity(src, dst)
 	if err != nil {
 		return nil, err

pkg/vpcmodel/textOutput.go

@@ -15,11 +15,21 @@ type TextOutputFormatter struct {
 }
 
 func multipleVPCsConfigHeader(c *VPCConfig) (string, error) {
-	if len(c.RoutingResources) != 2 { // expecting tgw and sgw (virtual gateway)
-		fmt.Printf("c.RoutingResources: %v\n", c.RoutingResources)
-		return "", errors.New("unexpected config of multiple VPCs connected by TGW, missing TGW resource")
+	resourceTypeTGW := "TGW"
+	var tgw RoutingResource
+	tgwRouterFound := false
+	for _, router := range c.RoutingResources {
+		if router.Kind() == resourceTypeTGW {
+			if tgwRouterFound {
+				return "", errors.New("unexpected number of RoutingResources for MultipleVPCsConfig, expecting only one TGW")
+			}
+			tgw = router
+			tgwRouterFound = true
+		}
+	}
+	if !tgwRouterFound {
+		return "", errors.New("unexpected number of RoutingResources for MultipleVPCsConfig, expecting TGW")
 	}
-	tgw := c.RoutingResources[0]
 	return fmt.Sprintf("Connectivity between VPCs connected by TGW %s (UID: %s)\n", tgw.NameForAnalyzerOut(c), tgw.UID()), nil
 }