Files
kubescape/core/cautils/remotegitutils_test.go
David Wertenteil 3cbd2c458d fix scanning repo
Signed-off-by: David Wertenteil <dwertent@armosec.io>
2024-05-06 16:22:38 +03:00

156 lines
3.4 KiB
Go

package cautils
import (
"errors"
"fmt"
"os"
"testing"
giturl "github.com/kubescape/go-git-url"
"github.com/stretchr/testify/assert"
)
func TestIsGitRepoPublic(t *testing.T) {
tests := []struct {
url string
want bool
}{
{
url: "https://github.com/kubescape/kubescape/",
want: true,
},
{
url: "http://invalidurl",
want: false,
},
{
url: "",
want: false,
},
}
for _, tt := range tests {
t.Run(tt.url, func(t *testing.T) {
assert.Equal(t, tt.want, isGitRepoPublic(tt.url))
})
}
}
func TestGetProviderError(t *testing.T) {
tests := []struct {
url string
err error
}{
{
url: "https://github.com/kubescape/kubescape/",
err: fmt.Errorf("%w", errors.New("GITHUB_TOKEN is not present")),
},
{
url: "https://gitlab.com/kubescape/kubescape/",
err: fmt.Errorf("%w", errors.New("GITLAB_TOKEN is not present")),
},
{
url: "https://dev.azure.com/kubescape/kubescape/",
err: fmt.Errorf("%w", errors.New("AZURE_TOKEN is not present")),
},
{
url: "https://bitbucket.org/kubescape/kubescape/",
err: fmt.Errorf("%w", errors.New("BITBUCKET_TOKEN is not present")),
},
}
for _, tt := range tests {
t.Run(tt.url, func(t *testing.T) {
gitURL, _ := giturl.NewGitAPI(tt.url)
assert.Equal(t, tt.err, getProviderError(gitURL))
})
}
}
func TestCloneRepo(t *testing.T) {
tests := []struct {
url string
err error
}{
{
url: "https://github.com/kubescape/kubescape/",
err: nil,
},
}
for _, tt := range tests {
t.Run(tt.url, func(t *testing.T) {
// Create a temporary directory
tmpDir, err := os.MkdirTemp("", "")
if err != nil {
t.Fatalf("failed to create temporary directory: %v", err)
}
gitURL, _ := giturl.NewGitAPI(tt.url)
tempDir, err := cloneRepo(gitURL)
assert.NotEqual(t, tmpDir, tempDir)
assert.Equal(t, tt.err, err)
})
}
}
func TestGetClonedPath(t *testing.T) {
testCases := []struct {
name string
path string
expected string
}{
{
name: "Valid Git URL",
path: "https://github.com/kubescape/kubescape.git",
expected: "/path/to/cloned/repo", // replace with the expected path
},
{
name: "Invalid Git URL",
path: "invalid",
expected: "",
},
}
tmpDirPaths = make(map[string]string)
tmpDirPaths[hashRepoURL("https://github.com/kubescape/kubescape.git")] = "/path/to/cloned/repo" // replace with the actual path
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
result := GetClonedPath(tc.path)
if result != tc.expected {
t.Errorf("Expected %q, got %q", tc.expected, result)
}
})
}
}
func TestGetDirPath(t *testing.T) {
testCases := []struct {
name string
repoURL string
expected string
}{
{
name: "Existing Repo URL",
repoURL: "https://github.com/user/repo.git",
expected: "/path/to/cloned/repo", // replace with the expected path
},
{
name: "Non-Existing Repo URL",
repoURL: "https://github.com/user/nonexistentrepo.git",
expected: "",
},
}
// Initialize tmpDirPaths
tmpDirPaths = make(map[string]string)
tmpDirPaths[hashRepoURL("https://github.com/user/repo.git")] = "/path/to/cloned/repo" // replace with the actual path
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
result := getDirPath(tc.repoURL)
if result != tc.expected {
t.Errorf("Expected %q, got %q", tc.expected, result)
}
})
}
}