added test for nil IPAddress

Azure · Feb 21, 2023 · 7e8d7fb · 7e8d7fb
1 parent 54ed684
commit 7e8d7fb
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/internal/scanners/ci_test.go b/internal/scanners/ci_test.go
@@ -40,6 +40,12 @@ func newContainerInstanceWithPrivateEndpoints(t *testing.T) *armcontainerinstanc
 	return svc
 }
 
+func newContainerInstanceWithoutIPAdress(t *testing.T) *armcontainerinstance.ContainerGroup {
+	svc := newContainerInstance(t)
+	svc.Properties.IPAddress = nil
+	return svc
+}
+
 func newContainerInstanceResult(t *testing.T) AzureServiceResult {
 	return AzureServiceResult{
 		SubscriptionID:     "subscriptionId",
@@ -68,6 +74,13 @@ func newContainerInstancePrivateEndpointResult(t *testing.T) AzureServiceResult
 	return svc
 }
 
+func newContainerInstanceWithoutIPAddressResult(t *testing.T) AzureServiceResult {
+	svc := newContainerInstanceResult(t)
+	svc.PrivateEndpoints = false
+	return svc
+}
+
+
 func TestContainerInstanceScanner_Scan(t *testing.T) {
 	type args struct {
 		resourceGroupName string
@@ -101,6 +114,7 @@ func TestContainerInstanceScanner_Scan(t *testing.T) {
 							newContainerInstance(t),
 							newContainerInstanceWithAvailabilityZones(t),
 							newContainerInstanceWithPrivateEndpoints(t),
+							newContainerInstanceWithoutIPAdress(t),
 						},
 						nil
 				},
@@ -112,6 +126,7 @@ func TestContainerInstanceScanner_Scan(t *testing.T) {
 				newContainerInstanceResult(t),
 				newContainerInstanceAvailabilityZonesResult(t),
 				newContainerInstancePrivateEndpointResult(t),
+				newContainerInstanceWithoutIPAddressResult(t),
 			},
 			wantErr: false,
 		},