Merge pull request #422 from armosec/updateriskfacotrs

Add new risk factor mapping for C-0266 and update test expectations

Author: kooomix

armotypes/common.go

@@ -27,6 +27,7 @@ const (
 
 var RiskFactorMapping = map[string]RiskFactor{
 	"C-0256": RiskFactorExternalFacing,
+	"C-0266": RiskFactorExternalFacing,
 	"C-0046": RiskFactorPrivileged,
 	"C-0057": RiskFactorPrivileged,
 	"C-0255": RiskFactorSecretAccess,

armotypes/common_test.go

@@ -1,9 +1,10 @@
 package armotypes
 
 import (
-	"github.com/stretchr/testify/assert"
 	"sort"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func TestGetControlIDsByRiskFactors(t *testing.T) {
@@ -15,7 +16,7 @@ func TestGetControlIDsByRiskFactors(t *testing.T) {
 		{
 			name:     "Single Risk Factor",
 			input:    "External facing",
-			expected: []string{"C-0256"},
+			expected: []string{"C-0256", "C-0266"},
 		},
 		{
 			name:     "Multiple Risk Factors",