mirror of
https://github.com/kubescape/kubescape.git
synced 2026-02-14 18:09:55 +00:00
156 lines
3.4 KiB
Go
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)
|
|
}
|
|
})
|
|
}
|
|
}
|