Skip to content

Commit

Permalink
test: add functional tests
Browse files Browse the repository at this point in the history 
Signed-off-by: Nikita Pivkin <[email protected]>
  • Loading branch information
@nikpivkin
nikpivkin committed Aug 19, 2024
1 parent 4801e54 commit 3de02c0
Show file tree
Hide file tree
Showing 5 changed files with 652 additions and 0 deletions.
190 changes: 190 additions & 0 deletions test/rego/azure_monitor_test.go
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,190 @@
package test

import (
"github.com/aquasecurity/trivy/pkg/iac/providers/azure"
"github.com/aquasecurity/trivy/pkg/iac/providers/azure/monitor"
"github.com/aquasecurity/trivy/pkg/iac/state"
trivyTypes "github.com/aquasecurity/trivy/pkg/iac/types"
)

var azureMonitorTestCases = testCases{
"AVD-AZU-0031": {
{
name: "Log retention policy disabled",
input: state.State{Azure: azure.Azure{Monitor: monitor.Monitor{
LogProfiles: []monitor.LogProfile{
{
Metadata: trivyTypes.NewTestMetadata(),
RetentionPolicy: monitor.RetentionPolicy{
Metadata: trivyTypes.NewTestMetadata(),
Enabled: trivyTypes.Bool(false, trivyTypes.NewTestMetadata()),
Days: trivyTypes.Int(365, trivyTypes.NewTestMetadata()),
},
},
},
}}},
expected: true,
},
{
name: "Log retention policy enabled for 90 days",
input: state.State{Azure: azure.Azure{Monitor: monitor.Monitor{
LogProfiles: []monitor.LogProfile{
{
Metadata: trivyTypes.NewTestMetadata(),
RetentionPolicy: monitor.RetentionPolicy{
Metadata: trivyTypes.NewTestMetadata(),
Enabled: trivyTypes.Bool(true, trivyTypes.NewTestMetadata()),
Days: trivyTypes.Int(90, trivyTypes.NewTestMetadata()),
},
},
},
}}},
expected: true,
},
{
name: "Log retention policy enabled for 365 days",
input: state.State{Azure: azure.Azure{Monitor: monitor.Monitor{
LogProfiles: []monitor.LogProfile{
{
Metadata: trivyTypes.NewTestMetadata(),
RetentionPolicy: monitor.RetentionPolicy{
Metadata: trivyTypes.NewTestMetadata(),
Enabled: trivyTypes.Bool(true, trivyTypes.NewTestMetadata()),
Days: trivyTypes.Int(365, trivyTypes.NewTestMetadata()),
},
},
},
}}},
expected: false,
},
},
"AVD-AZU-0033": {
{
name: "Log profile captures only write activities",
input: state.State{Azure: azure.Azure{Monitor: monitor.Monitor{
LogProfiles: []monitor.LogProfile{
{
Metadata: trivyTypes.NewTestMetadata(),
Categories: []trivyTypes.StringValue{
trivyTypes.String("Write", trivyTypes.NewTestMetadata()),
},
},
},
}}},
expected: true,
},
{
name: "Log profile captures action, write, delete activities",
input: state.State{Azure: azure.Azure{Monitor: monitor.Monitor{
LogProfiles: []monitor.LogProfile{
{
Metadata: trivyTypes.NewTestMetadata(),
Categories: []trivyTypes.StringValue{
trivyTypes.String("Action", trivyTypes.NewTestMetadata()),
trivyTypes.String("Write", trivyTypes.NewTestMetadata()),
trivyTypes.String("Delete", trivyTypes.NewTestMetadata()),
},
},
},
}}},
expected: false,
},
},
"AVD-AZU-0032": {
{
name: "Log profile captures only eastern US region",
input: state.State{Azure: azure.Azure{Monitor: monitor.Monitor{
LogProfiles: []monitor.LogProfile{
{
Metadata: trivyTypes.NewTestMetadata(),
Locations: []trivyTypes.StringValue{
trivyTypes.String("eastus", trivyTypes.NewTestMetadata()),
},
},
},
}}},
expected: true,
},
{
name: "Log profile captures all regions",
input: state.State{Azure: azure.Azure{Monitor: monitor.Monitor{
LogProfiles: []monitor.LogProfile{
{
Metadata: trivyTypes.NewTestMetadata(),
Locations: []trivyTypes.StringValue{
trivyTypes.String("eastus", trivyTypes.NewTestMetadata()),
trivyTypes.String("eastus2", trivyTypes.NewTestMetadata()),
trivyTypes.String("southcentralus", trivyTypes.NewTestMetadata()),
trivyTypes.String("westus2", trivyTypes.NewTestMetadata()),
trivyTypes.String("westus3", trivyTypes.NewTestMetadata()),
trivyTypes.String("australiaeast", trivyTypes.NewTestMetadata()),
trivyTypes.String("southeastasia", trivyTypes.NewTestMetadata()),
trivyTypes.String("northeurope", trivyTypes.NewTestMetadata()),
trivyTypes.String("swedencentral", trivyTypes.NewTestMetadata()),
trivyTypes.String("uksouth", trivyTypes.NewTestMetadata()),
trivyTypes.String("westeurope", trivyTypes.NewTestMetadata()),
trivyTypes.String("centralus", trivyTypes.NewTestMetadata()),
trivyTypes.String("northcentralus", trivyTypes.NewTestMetadata()),
trivyTypes.String("westus", trivyTypes.NewTestMetadata()),
trivyTypes.String("southafricanorth", trivyTypes.NewTestMetadata()),
trivyTypes.String("centralindia", trivyTypes.NewTestMetadata()),
trivyTypes.String("eastasia", trivyTypes.NewTestMetadata()),
trivyTypes.String("japaneast", trivyTypes.NewTestMetadata()),
trivyTypes.String("jioindiawest", trivyTypes.NewTestMetadata()),
trivyTypes.String("koreacentral", trivyTypes.NewTestMetadata()),
trivyTypes.String("canadacentral", trivyTypes.NewTestMetadata()),
trivyTypes.String("francecentral", trivyTypes.NewTestMetadata()),
trivyTypes.String("germanywestcentral", trivyTypes.NewTestMetadata()),
trivyTypes.String("norwayeast", trivyTypes.NewTestMetadata()),
trivyTypes.String("switzerlandnorth", trivyTypes.NewTestMetadata()),
trivyTypes.String("uaenorth", trivyTypes.NewTestMetadata()),
trivyTypes.String("brazilsouth", trivyTypes.NewTestMetadata()),
trivyTypes.String("centralusstage", trivyTypes.NewTestMetadata()),
trivyTypes.String("eastusstage", trivyTypes.NewTestMetadata()),
trivyTypes.String("eastus2stage", trivyTypes.NewTestMetadata()),
trivyTypes.String("northcentralusstage", trivyTypes.NewTestMetadata()),
trivyTypes.String("southcentralusstage", trivyTypes.NewTestMetadata()),
trivyTypes.String("westusstage", trivyTypes.NewTestMetadata()),
trivyTypes.String("westus2stage", trivyTypes.NewTestMetadata()),
trivyTypes.String("asia", trivyTypes.NewTestMetadata()),
trivyTypes.String("asiapacific", trivyTypes.NewTestMetadata()),
trivyTypes.String("australia", trivyTypes.NewTestMetadata()),
trivyTypes.String("brazil", trivyTypes.NewTestMetadata()),
trivyTypes.String("canada", trivyTypes.NewTestMetadata()),
trivyTypes.String("europe", trivyTypes.NewTestMetadata()),
trivyTypes.String("global", trivyTypes.NewTestMetadata()),
trivyTypes.String("india", trivyTypes.NewTestMetadata()),
trivyTypes.String("japan", trivyTypes.NewTestMetadata()),
trivyTypes.String("uk", trivyTypes.NewTestMetadata()),
trivyTypes.String("unitedstates", trivyTypes.NewTestMetadata()),
trivyTypes.String("eastasiastage", trivyTypes.NewTestMetadata()),
trivyTypes.String("southeastasiastage", trivyTypes.NewTestMetadata()),
trivyTypes.String("centraluseuap", trivyTypes.NewTestMetadata()),
trivyTypes.String("eastus2euap", trivyTypes.NewTestMetadata()),
trivyTypes.String("westcentralus", trivyTypes.NewTestMetadata()),
trivyTypes.String("southafricawest", trivyTypes.NewTestMetadata()),
trivyTypes.String("australiacentral", trivyTypes.NewTestMetadata()),
trivyTypes.String("australiacentral2", trivyTypes.NewTestMetadata()),
trivyTypes.String("australiasoutheast", trivyTypes.NewTestMetadata()),
trivyTypes.String("japanwest", trivyTypes.NewTestMetadata()),
trivyTypes.String("jioindiacentral", trivyTypes.NewTestMetadata()),
trivyTypes.String("koreasouth", trivyTypes.NewTestMetadata()),
trivyTypes.String("southindia", trivyTypes.NewTestMetadata()),
trivyTypes.String("westindia", trivyTypes.NewTestMetadata()),
trivyTypes.String("canadaeast", trivyTypes.NewTestMetadata()),
trivyTypes.String("francesouth", trivyTypes.NewTestMetadata()),
trivyTypes.String("germanynorth", trivyTypes.NewTestMetadata()),
trivyTypes.String("norwaywest", trivyTypes.NewTestMetadata()),
trivyTypes.String("swedensouth", trivyTypes.NewTestMetadata()),
trivyTypes.String("switzerlandwest", trivyTypes.NewTestMetadata()),
trivyTypes.String("ukwest", trivyTypes.NewTestMetadata()),
trivyTypes.String("uaecentral", trivyTypes.NewTestMetadata()),
trivyTypes.String("brazilsoutheast", trivyTypes.NewTestMetadata()),
},
},
},
}}},
expected: false,
},
},
}
Loading

0 comments on commit 3de02c0

Please sign in to comment.