mirror of
https://github.com/kubescape/kubescape.git
synced 2026-04-04 01:37:24 +00:00
122 lines
3.6 KiB
Go
122 lines
3.6 KiB
Go
package scan
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/kubescape/kubescape/v3/core/cautils"
|
|
"github.com/kubescape/kubescape/v3/core/mocks"
|
|
v1 "github.com/kubescape/opa-utils/httpserver/apis/v1"
|
|
"github.com/kubescape/opa-utils/objectsenvelopes"
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSetWorkloadScanInfo(t *testing.T) {
|
|
test := []struct {
|
|
Description string
|
|
kind string
|
|
name string
|
|
want *cautils.ScanInfo
|
|
}{
|
|
{
|
|
Description: "Set workload scan info",
|
|
kind: "Deployment",
|
|
name: "test",
|
|
want: &cautils.ScanInfo{
|
|
PolicyIdentifier: []cautils.PolicyIdentifier{
|
|
{
|
|
Identifier: "workloadscan",
|
|
Kind: v1.KindFramework,
|
|
},
|
|
{
|
|
Identifier: "allcontrols",
|
|
Kind: v1.KindFramework,
|
|
},
|
|
},
|
|
ScanType: cautils.ScanTypeWorkload,
|
|
ScanObject: &objectsenvelopes.ScanObject{
|
|
Kind: "Deployment",
|
|
Metadata: objectsenvelopes.ScanObjectMetadata{
|
|
Name: "test",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range test {
|
|
t.Run(
|
|
tc.Description,
|
|
func(t *testing.T) {
|
|
scanInfo := &cautils.ScanInfo{}
|
|
setWorkloadScanInfo(scanInfo, tc.kind, tc.name)
|
|
|
|
if scanInfo.ScanType != tc.want.ScanType {
|
|
t.Errorf("got: %v, want: %v", scanInfo.ScanType, tc.want.ScanType)
|
|
}
|
|
|
|
if scanInfo.ScanObject.Kind != tc.want.ScanObject.Kind {
|
|
t.Errorf("got: %v, want: %v", scanInfo.ScanObject.Kind, tc.want.ScanObject.Kind)
|
|
}
|
|
|
|
if scanInfo.ScanObject.Metadata.Name != tc.want.ScanObject.Metadata.Name {
|
|
t.Errorf("got: %v, want: %v", scanInfo.ScanObject.Metadata.Name, tc.want.ScanObject.Metadata.Name)
|
|
}
|
|
|
|
if len(scanInfo.PolicyIdentifier) != len(tc.want.PolicyIdentifier) {
|
|
t.Errorf("got: %v policy identifiers, want: %v", len(scanInfo.PolicyIdentifier), len(tc.want.PolicyIdentifier))
|
|
}
|
|
|
|
for i, wantPolicy := range tc.want.PolicyIdentifier {
|
|
if i < len(scanInfo.PolicyIdentifier) {
|
|
if scanInfo.PolicyIdentifier[i].Identifier != wantPolicy.Identifier {
|
|
t.Errorf("got: %v, want: %v", scanInfo.PolicyIdentifier[i].Identifier, wantPolicy.Identifier)
|
|
}
|
|
if scanInfo.PolicyIdentifier[i].Kind != wantPolicy.Kind {
|
|
t.Errorf("got: %v, want: %v", scanInfo.PolicyIdentifier[i].Kind, wantPolicy.Kind)
|
|
}
|
|
}
|
|
}
|
|
},
|
|
)
|
|
}
|
|
}
|
|
|
|
func TestGetWorkloadCmd_ChartPathAndFilePathEmpty(t *testing.T) {
|
|
// Create a mock Kubescape interface
|
|
mockKubescape := &mocks.MockIKubescape{}
|
|
scanInfo := cautils.ScanInfo{
|
|
ChartPath: "temp",
|
|
FilePath: "",
|
|
}
|
|
|
|
cmd := getWorkloadCmd(mockKubescape, &scanInfo)
|
|
|
|
// Verify the command name and short description
|
|
assert.Equal(t, "workload <kind>/<name> [`<glob pattern>`/`-`] [flags]", cmd.Use)
|
|
assert.Equal(t, "Scan a workload for misconfigurations and image vulnerabilities", cmd.Short)
|
|
assert.Equal(t, workloadExample, cmd.Example)
|
|
|
|
err := cmd.Args(&cobra.Command{}, []string{})
|
|
expectedErrorMessage := "usage: <kind>/<name> [`<glob pattern>`/`-`] [flags]"
|
|
assert.Equal(t, expectedErrorMessage, err.Error())
|
|
|
|
err = cmd.Args(&cobra.Command{}, []string{"nginx"})
|
|
expectedErrorMessage = "invalid workload identifier"
|
|
assert.Equal(t, expectedErrorMessage, err.Error())
|
|
}
|
|
|
|
func Test_parseWorkloadIdentifierString_Empty(t *testing.T) {
|
|
t.Run("empty identifier", func(t *testing.T) {
|
|
_, _, err := parseWorkloadIdentifierString("")
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func Test_parseWorkloadIdentifierString_NoError(t *testing.T) {
|
|
t.Run("valid identifier", func(t *testing.T) {
|
|
_, _, err := parseWorkloadIdentifierString("default/Deployment")
|
|
assert.NoError(t, err)
|
|
})
|
|
}
|