chore: import upstream snapshot with attribution

2026-07-13 12:08:39 +08:00
commit a0df89c693
5252 changed files with 523444 additions and 0 deletions
@@ -0,0 +1,117 @@
+package sources
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"strconv"
+	"testing"
+
+	portainer "github.com/portainer/portainer/api"
+	"github.com/portainer/portainer/api/dataservices"
+	"github.com/portainer/portainer/api/datastore"
+	gittypes "github.com/portainer/portainer/api/git/types"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestGetSource_NotFound(t *testing.T) {
+	t.Parallel()
+	_, store := datastore.MustNewTestStore(t, false, true)
+
+	require.NoError(t, store.UpdateTx(func(tx dataservices.DataStoreTx) error {
+		return tx.User().Create(&portainer.User{ID: 1, Role: portainer.AdministratorRole})
+	}))
+
+	h := newTestHandler(t, store)
+	rr := httptest.NewRecorder()
+	h.ServeHTTP(rr, buildGetReq(t, 1, "999"))
+	assert.Equal(t, http.StatusNotFound, rr.Code)
+}
+
+func TestGetSource_ReturnsDetail(t *testing.T) {
+	t.Parallel()
+	_, store := datastore.MustNewTestStore(t, false, true)
+
+	cfg := &gittypes.GitSource{
+		URL:           "https://github.com/org/repo",
+		TLSSkipVerify: true,
+	}
+
+	var srcID portainer.SourceID
+	require.NoError(t, store.UpdateTx(func(tx dataservices.DataStoreTx) error {
+		stack := &portainer.Stack{ID: 1, Name: "my-stack"}
+		srcID = createGitWorkflow(t, tx, stack, cfg)
+		require.NoError(t, tx.Stack().Create(stack))
+		return tx.User().Create(&portainer.User{ID: 1, Role: portainer.AdministratorRole})
+	}))
+
+	h := newTestHandler(t, store)
+	rr := httptest.NewRecorder()
+	h.ServeHTTP(rr, buildGetReq(t, 1, strconv.Itoa(int(srcID))))
+
+	detail := decodeSourceDetail(t, rr)
+	assert.Equal(t, srcID, detail.ID)
+	assert.Equal(t, "repo", detail.Name)
+	assert.Equal(t, 1, detail.UsedBy)
+	assert.True(t, detail.Connection.TLSSkipVerify)
+	require.Len(t, detail.Workflows, 1)
+	assert.Equal(t, "my-stack", detail.Workflows[0].Name)
+}
+
+func TestGetSource_RedactsCredentials(t *testing.T) {
+	t.Parallel()
+	_, store := datastore.MustNewTestStore(t, false, true)
+
+	cfg := &gittypes.GitSource{
+		URL:            "https://github.com/org/secure",
+		Authentication: &gittypes.GitAuthentication{Username: "user", Password: "s3cr3t"},
+	}
+
+	var srcID portainer.SourceID
+	require.NoError(t, store.UpdateTx(func(tx dataservices.DataStoreTx) error {
+		stack := &portainer.Stack{ID: 1, Name: "secure-stack"}
+		srcID = createGitWorkflow(t, tx, stack, cfg)
+		require.NoError(t, tx.Stack().Create(stack))
+		return tx.User().Create(&portainer.User{ID: 1, Role: portainer.AdministratorRole})
+	}))
+
+	h := newTestHandler(t, store)
+	rr := httptest.NewRecorder()
+	h.ServeHTTP(rr, buildGetReq(t, 1, strconv.Itoa(int(srcID))))
+
+	detail := decodeSourceDetail(t, rr)
+	require.Len(t, detail.Workflows, 1)
+	require.NotNil(t, detail.Workflows[0].GitConfig)
+	require.NotNil(t, detail.Workflows[0].GitConfig.Authentication)
+	assert.Equal(t, "user", detail.Workflows[0].GitConfig.Authentication.Username)
+	assert.Empty(t, detail.Workflows[0].GitConfig.Authentication.Password)
+}
+
+func TestGetSource_AutoUpdate(t *testing.T) {
+	t.Parallel()
+	_, store := datastore.MustNewTestStore(t, false, true)
+
+	cfg := gitCfg("https://github.com/org/polled")
+
+	var srcID portainer.SourceID
+	require.NoError(t, store.UpdateTx(func(tx dataservices.DataStoreTx) error {
+		stack := &portainer.Stack{
+			ID:         1,
+			Name:       "polled-stack",
+			AutoUpdate: &portainer.AutoUpdateSettings{Interval: "5m"},
+		}
+		srcID = createGitWorkflow(t, tx, stack, cfg)
+		require.NoError(t, tx.Stack().Create(stack))
+		return tx.User().Create(&portainer.User{ID: 1, Role: portainer.AdministratorRole})
+	}))
+
+	h := newTestHandler(t, store)
+	rr := httptest.NewRecorder()
+	h.ServeHTTP(rr, buildGetReq(t, 1, strconv.Itoa(int(srcID))))
+
+	detail := decodeSourceDetail(t, rr)
+	require.NotNil(t, detail.AutoUpdate)
+	assert.Equal(t, "Interval", detail.AutoUpdate.Mechanism)
+	assert.Equal(t, "5m", detail.AutoUpdate.FetchInterval)
+}