From 67e431d27e713b9a14f3378f452951b079054d3f Mon Sep 17 00:00:00 2001
From: Carlos Mendible <266546+cmendible@users.noreply.github.com>
Date: Thu, 9 May 2024 00:21:41 +0000
Subject: [PATCH] fixing vm az check #235

---
 internal/scanners/vm/rules.go      |  2 +-
 internal/scanners/vm/rules_test.go | 16 +++++++++++++++-
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/internal/scanners/vm/rules.go b/internal/scanners/vm/rules.go
index 8dde435e..e6ddc2a3 100644
--- a/internal/scanners/vm/rules.go
+++ b/internal/scanners/vm/rules.go
@@ -20,7 +20,7 @@ func (a *VirtualMachineScanner) GetRules() map[string]scanners.AzureRule {
 			Impact:         scanners.ImpactHigh,
 			Eval: func(target interface{}, scanContext *scanners.ScanContext) (bool, string) {
 				v := target.(*armcompute.VirtualMachine)
-				hasZones := v.Zones != nil && len(v.Zones) > 1
+				hasZones := v.Zones != nil && len(v.Zones) >= 1
 				return !hasZones, ""
 			},
 			Url: "https://learn.microsoft.com/en-us/azure/virtual-machines/availability#availability-zones",
diff --git a/internal/scanners/vm/rules_test.go b/internal/scanners/vm/rules_test.go
index c112c026..73c2177a 100644
--- a/internal/scanners/vm/rules_test.go
+++ b/internal/scanners/vm/rules_test.go
@@ -28,7 +28,7 @@ func TestVirtualMachineScanner_Rules(t *testing.T) {
 		want   want
 	}{
 		{
-			name: "VirtualMachineScanner Availability Zones",
+			name: "VirtualMachineScanner No Availability Zones",
 			fields: fields{
 				rule:        "vm-002",
 				target:      &armcompute.VirtualMachine{},
@@ -39,6 +39,20 @@ func TestVirtualMachineScanner_Rules(t *testing.T) {
 				result: "",
 			},
 		},
+		{
+			name: "VirtualMachineScanner Availability Zones",
+			fields: fields{
+				rule:        "vm-002",
+				target:      &armcompute.VirtualMachine{
+					Zones: []*string{to.Ptr("1")},
+				},
+				scanContext: &scanners.ScanContext{},
+			},
+			want: want{
+				broken: false,
+				result: "",
+			},
+		},
 		{
 			name: "VirtualMachineScanner SLA 99.9%",
 			fields: fields{