mirror of https://github.com/helm/helm
devdinu
5 years ago
parent
87abc3b47f
commit
cacf215d6c
@ -0,0 +1,98 @@
|
||||
package api_test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/mock"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/stretchr/testify/suite"
|
||||
"gotest.tools/assert"
|
||||
"helm.sh/helm/v3/pkg/chart"
|
||||
"helm.sh/helm/v3/pkg/cli"
|
||||
"helm.sh/helm/v3/pkg/http/api"
|
||||
"helm.sh/helm/v3/pkg/http/api/logger"
|
||||
"helm.sh/helm/v3/pkg/release"
|
||||
)
|
||||
|
||||
type InstallerTestSuite struct {
|
||||
suite.Suite
|
||||
recorder *httptest.ResponseRecorder
|
||||
server *httptest.Server
|
||||
mockInstaller *mockInstaller
|
||||
mockChartLoader *mockChartLoader
|
||||
appConfig *cli.EnvSettings
|
||||
}
|
||||
|
||||
func (s *InstallerTestSuite) SetupSuite() {
|
||||
logger.Setup("default")
|
||||
}
|
||||
|
||||
func (s *InstallerTestSuite) SetupTest() {
|
||||
s.recorder = httptest.NewRecorder()
|
||||
s.mockInstaller = new(mockInstaller)
|
||||
s.mockChartLoader = new(mockChartLoader)
|
||||
s.appConfig = &cli.EnvSettings{
|
||||
RepositoryConfig: "./testdata/helm",
|
||||
PluginsDirectory: "./testdata/helm/plugin",
|
||||
}
|
||||
service := api.NewService(s.appConfig, s.mockChartLoader, s.mockInstaller)
|
||||
handler := api.Install(service)
|
||||
s.server = httptest.NewServer(handler)
|
||||
}
|
||||
|
||||
func (s *InstallerTestSuite) TestShouldReturnDeployedStatusOnSuccessfulInstall() {
|
||||
chartName := "stable/redis-ha"
|
||||
body := fmt.Sprintf(`{
|
||||
"chart":"%s",
|
||||
"name": "redis-v5",
|
||||
"namespace": "something"}`, chartName)
|
||||
req, _ := http.NewRequest("POST", fmt.Sprintf("%s/install", s.server.URL), strings.NewReader(body))
|
||||
s.mockChartLoader.On("LocateChart", chartName, s.appConfig).Return("./testdata/albatross", nil)
|
||||
icfg := api.InstallConfig{ChartName: chartName, Name: "redis-v5", Namespace: "something"}
|
||||
s.mockInstaller.On("SetConfig", icfg)
|
||||
release := &release.Release{Info: &release.Info{Status: release.StatusDeployed}}
|
||||
var vals map[string]interface{}
|
||||
//TODO: pass chart object and verify values present testdata chart yml
|
||||
s.mockInstaller.On("Run", mock.AnythingOfType("*chart.Chart"), vals).Return(release, nil)
|
||||
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
|
||||
assert.Equal(s.T(), 200, resp.StatusCode)
|
||||
expectedResponse := `{"status":"deployed"}` + "\n"
|
||||
respBody, _ := ioutil.ReadAll(resp.Body)
|
||||
assert.Equal(s.T(), expectedResponse, string(respBody))
|
||||
require.NoError(s.T(), err)
|
||||
s.mockInstaller.AssertExpectations(s.T())
|
||||
s.mockChartLoader.AssertExpectations(s.T())
|
||||
}
|
||||
|
||||
func (s *InstallerTestSuite) TearDownTest() {
|
||||
s.server.Close()
|
||||
}
|
||||
|
||||
func TestInstallAPI(t *testing.T) {
|
||||
suite.Run(t, new(InstallerTestSuite))
|
||||
}
|
||||
|
||||
type mockInstaller struct{ mock.Mock }
|
||||
|
||||
func (m *mockInstaller) SetConfig(cfg api.InstallConfig) {
|
||||
m.Called(cfg)
|
||||
}
|
||||
|
||||
func (m *mockInstaller) Run(c *chart.Chart, vals map[string]interface{}) (*release.Release, error) {
|
||||
args := m.Called(c, vals)
|
||||
return args.Get(0).(*release.Release), args.Error(1)
|
||||
}
|
||||
|
||||
type mockChartLoader struct{ mock.Mock }
|
||||
|
||||
func (m *mockChartLoader) LocateChart(name string, settings *cli.EnvSettings) (string, error) {
|
||||
args := m.Called(name, settings)
|
||||
return args.String(0), args.Error(1)
|
||||
}
|
||||
@ -0,0 +1,24 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"helm.sh/helm/v3/pkg/action"
|
||||
"helm.sh/helm/v3/pkg/chart"
|
||||
"helm.sh/helm/v3/pkg/release"
|
||||
)
|
||||
|
||||
type Installer struct {
|
||||
*action.Install
|
||||
}
|
||||
|
||||
type runner interface {
|
||||
Run(*chart.Chart, map[string]interface{}) (*release.Release, error)
|
||||
}
|
||||
|
||||
func (i *Installer) SetConfig(cfg InstallConfig) {
|
||||
i.ReleaseName = cfg.Name
|
||||
i.Namespace = cfg.Namespace
|
||||
}
|
||||
|
||||
func NewInstaller(ai *action.Install) *Installer {
|
||||
return &Installer{ai}
|
||||
}
|
||||
@ -0,0 +1,4 @@
|
||||
name: albatross
|
||||
description: A Helm chart for Kubernetes
|
||||
version: 0.1.0
|
||||
home: ""
|
||||
@ -0,0 +1,4 @@
|
||||
albatross: "true"
|
||||
|
||||
global:
|
||||
author: Coleridge
|
||||
Loading…
Reference in new issue