mirror of
https://github.com/portainer/portainer.git
synced 2025-07-19 13:29:41 +02:00
377326085d
* backport migration EE code structure * filesystem copy function * set db status to updating before migration - reset on completion * support for auto-backup on version upgrade * - rollback cli flag support (with confirmation) - rollback implementation backport from EE * removed edition as it is not required in CE * migrated test datastore from bolttest to bolt package to make it usable for testing * backported failsafe migration * - backported tests from EE - refactored tests to use test datastore * test store implementing datastore interface * addressed PR issues/improvements * refactor test * added backup file removal error logging * resolved conflicts, updated code * fixed missing bolttest package - migrated to bolt * feat(migration): wrap migration errors to provide context for failure EE-1742 (#5711) * feat(migrator): wrap errors to provide more context to failures EE-1742 * add overall failure back in. diff log file * updated helm tests pointing to correct teststore Co-authored-by: Matt Hook <hookenz@gmail.com>
65 lines
2.2 KiB
Go
65 lines
2.2 KiB
Go
package helm
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/portainer/libhelm/binary/test"
|
|
"github.com/portainer/libhelm/options"
|
|
"github.com/portainer/libhelm/release"
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/bolt"
|
|
"github.com/portainer/portainer/api/http/security"
|
|
helper "github.com/portainer/portainer/api/internal/testhelpers"
|
|
"github.com/portainer/portainer/api/kubernetes"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_helmInstall(t *testing.T) {
|
|
is := assert.New(t)
|
|
|
|
store, teardown := bolt.MustNewTestStore(true)
|
|
defer teardown()
|
|
|
|
err := store.Endpoint().CreateEndpoint(&portainer.Endpoint{ID: 1})
|
|
is.NoError(err, "error creating environment")
|
|
|
|
err = store.User().CreateUser(&portainer.User{Username: "admin", Role: portainer.AdministratorRole})
|
|
is.NoError(err, "error creating a user")
|
|
|
|
helmPackageManager := test.NewMockHelmBinaryPackageManager("")
|
|
kubeConfigService := kubernetes.NewKubeConfigCAService("", "")
|
|
h := NewHandler(helper.NewTestRequestBouncer(), store, helmPackageManager, kubeConfigService)
|
|
|
|
is.NotNil(h, "Handler should not fail")
|
|
|
|
// Install a single chart. We expect to get these values back
|
|
options := options.InstallOptions{Name: "nginx-1", Chart: "nginx", Namespace: "default", Repo: "https://charts.bitnami.com/bitnami"}
|
|
optdata, err := json.Marshal(options)
|
|
is.NoError(err)
|
|
|
|
t.Run("helmInstall succeeds with admin user", func(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodPost, "/1/kubernetes/helm", bytes.NewBuffer(optdata))
|
|
ctx := security.StoreTokenData(req, &portainer.TokenData{ID: 1, Username: "admin", Role: 1})
|
|
req = req.WithContext(ctx)
|
|
req.Header.Add("Authorization", "Bearer dummytoken")
|
|
|
|
rr := httptest.NewRecorder()
|
|
h.ServeHTTP(rr, req)
|
|
|
|
is.Equal(http.StatusCreated, rr.Code, "Status should be 201")
|
|
|
|
body, err := io.ReadAll(rr.Body)
|
|
is.NoError(err, "ReadAll should not return error")
|
|
|
|
resp := release.Release{}
|
|
err = json.Unmarshal(body, &resp)
|
|
is.NoError(err, "response should be json")
|
|
is.EqualValues(options.Name, resp.Name, "Name doesn't match")
|
|
is.EqualValues(options.Namespace, resp.Namespace, "Namespace doesn't match")
|
|
})
|
|
}
|