mirror of https://github.com/helm/helm
Michelle Noorali
8 years ago
parent
36699cc22d
commit
f74720613b
@ -0,0 +1,73 @@
|
||||
/*
|
||||
Copyright 2016 The Kubernetes Authors All rights reserved.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
|
||||
"k8s.io/helm/pkg/chartutil"
|
||||
"k8s.io/helm/pkg/proto/hapi/chart"
|
||||
)
|
||||
|
||||
func TestUpgradeCmd(t *testing.T) {
|
||||
tmpChart, _ := ioutil.TempDir("testdata", "tmp")
|
||||
defer os.RemoveAll(tmpChart)
|
||||
cfile := &chart.Metadata{
|
||||
Name: "testUpgradeChart",
|
||||
Description: "A Helm chart for Kubernetes",
|
||||
Version: "0.1.0",
|
||||
}
|
||||
chartPath, err := chartutil.Create(cfile, tmpChart)
|
||||
if err != nil {
|
||||
t.Errorf("Error creating chart for upgrade: %v", err)
|
||||
}
|
||||
ch, _ := chartutil.Load(chartPath)
|
||||
_ = releaseMock(&releaseOptions{
|
||||
name: "funny-bunny",
|
||||
chart: ch,
|
||||
})
|
||||
|
||||
// update chart version
|
||||
cfile = &chart.Metadata{
|
||||
Name: "testUpgradeChart",
|
||||
Description: "A Helm chart for Kubernetes",
|
||||
Version: "0.1.2",
|
||||
}
|
||||
chartPath, err = chartutil.Create(cfile, tmpChart)
|
||||
ch, _ = chartutil.Load(chartPath)
|
||||
|
||||
tests := []releaseCase{
|
||||
{
|
||||
name: "upgrade a release",
|
||||
args: []string{"funny-bunny", chartPath},
|
||||
resp: releaseMock(&releaseOptions{name: "funny-bunny", version: 2, chart: ch}),
|
||||
expected: "It's not you. It's me\nYour upgrade looks valid but this command is still under active development.\nHang tight.\n",
|
||||
},
|
||||
}
|
||||
|
||||
cmd := func(c *fakeReleaseClient, out io.Writer) *cobra.Command {
|
||||
return newUpgradeCmd(c, out)
|
||||
}
|
||||
|
||||
runReleaseCases(t, tests, cmd)
|
||||
|
||||
}
|
||||
@ -1,82 +0,0 @@
|
||||
/*
|
||||
Copyright 2016 The Kubernetes Authors All rights reserved.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
*/
|
||||
|
||||
package helm
|
||||
|
||||
import (
|
||||
rls "k8s.io/helm/pkg/proto/hapi/services"
|
||||
)
|
||||
|
||||
// These APIs are a temporary abstraction layer that captures the interaction between the current cmd/helm and old
|
||||
// pkg/helm implementations. Post refactor the cmd/helm package will use the APIs exposed on helm.Client directly.
|
||||
|
||||
// Config is the base configuration
|
||||
var Config struct {
|
||||
ServAddr string
|
||||
}
|
||||
|
||||
// ListReleases lists releases. DEPRECATED.
|
||||
//
|
||||
// Soon to be deprecated helm ListReleases API.
|
||||
func ListReleases(limit int, offset string, sort rls.ListSort_SortBy, order rls.ListSort_SortOrder, filter string) (*rls.ListReleasesResponse, error) {
|
||||
opts := []ReleaseListOption{
|
||||
ReleaseListLimit(limit),
|
||||
ReleaseListOffset(offset),
|
||||
ReleaseListFilter(filter),
|
||||
ReleaseListSort(int32(sort)),
|
||||
ReleaseListOrder(int32(order)),
|
||||
}
|
||||
return NewClient(Host(Config.ServAddr)).ListReleases(opts...)
|
||||
}
|
||||
|
||||
// GetReleaseStatus gets a release status. DEPRECATED
|
||||
//
|
||||
// Soon to be deprecated helm GetReleaseStatus API.
|
||||
func GetReleaseStatus(rlsName string) (*rls.GetReleaseStatusResponse, error) {
|
||||
return NewClient(Host(Config.ServAddr)).ReleaseStatus(rlsName)
|
||||
}
|
||||
|
||||
// GetReleaseContent gets the content of a release.
|
||||
// Soon to be deprecated helm GetReleaseContent API.
|
||||
func GetReleaseContent(rlsName string) (*rls.GetReleaseContentResponse, error) {
|
||||
return NewClient(Host(Config.ServAddr)).ReleaseContent(rlsName)
|
||||
}
|
||||
|
||||
// UpdateRelease updates a release.
|
||||
// Soon to be deprecated helm UpdateRelease API.
|
||||
func UpdateRelease(rlsName, chStr string, vals []byte, dryRun bool) (*rls.UpdateReleaseResponse, error) {
|
||||
return NewClient(Host(Config.ServAddr)).UpdateRelease(rlsName, chStr)
|
||||
}
|
||||
|
||||
// InstallRelease runs an install for a release.
|
||||
// Soon to be deprecated helm InstallRelease API.
|
||||
func InstallRelease(vals []byte, rlsName, chStr string, dryRun bool) (*rls.InstallReleaseResponse, error) {
|
||||
client := NewClient(Host(Config.ServAddr))
|
||||
if dryRun {
|
||||
client.Option(DryRun())
|
||||
}
|
||||
return client.InstallRelease(chStr, ValueOverrides(vals), ReleaseName(rlsName))
|
||||
}
|
||||
|
||||
// UninstallRelease destroys an existing release.
|
||||
// Soon to be deprecated helm UninstallRelease API.
|
||||
func UninstallRelease(rlsName string, dryRun bool) (*rls.UninstallReleaseResponse, error) {
|
||||
client := NewClient(Host(Config.ServAddr))
|
||||
if dryRun {
|
||||
client.Option(DryRun())
|
||||
}
|
||||
return client.DeleteRelease(rlsName)
|
||||
}
|
||||
Loading…
Reference in new issue