mirror of https://github.com/helm/helm
so it can be used by other projects that use the helm clientreviewable/pr2550/r1
Michelle Noorali
8 years ago
parent
52a3405b1e
commit
70520efac4
@ -0,0 +1,147 @@
|
|||||||
|
/*
|
||||||
|
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 "k8s.io/helm/pkg/helm"
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"k8s.io/helm/pkg/proto/hapi/chart"
|
||||||
|
"k8s.io/helm/pkg/proto/hapi/release"
|
||||||
|
rls "k8s.io/helm/pkg/proto/hapi/services"
|
||||||
|
"k8s.io/helm/pkg/proto/hapi/version"
|
||||||
|
)
|
||||||
|
|
||||||
|
// FakeReleaseClient implements Interface
|
||||||
|
type FakeReleaseClient struct {
|
||||||
|
Rels []*release.Release
|
||||||
|
Responses map[string]release.TestRun_Status
|
||||||
|
Err error
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ Interface = &FakeReleaseClient{}
|
||||||
|
var _ Interface = (*FakeReleaseClient)(nil)
|
||||||
|
|
||||||
|
// Listreleases lists the current releases
|
||||||
|
func (c *FakeReleaseClient) ListReleases(opts ...ReleaseListOption) (*rls.ListReleasesResponse, error) {
|
||||||
|
resp := &rls.ListReleasesResponse{
|
||||||
|
Count: int64(len(c.Rels)),
|
||||||
|
Releases: c.Rels,
|
||||||
|
}
|
||||||
|
return resp, c.Err
|
||||||
|
}
|
||||||
|
|
||||||
|
// InstallRelease returns a response with the first Release on the fake release client
|
||||||
|
func (c *FakeReleaseClient) InstallRelease(chStr, ns string, opts ...InstallOption) (*rls.InstallReleaseResponse, error) {
|
||||||
|
return &rls.InstallReleaseResponse{
|
||||||
|
Release: c.Rels[0],
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// InstallReleaseFromChart returns a response with the first Release on the fake release client
|
||||||
|
func (c *FakeReleaseClient) InstallReleaseFromChart(chart *chart.Chart, ns string, opts ...InstallOption) (*rls.InstallReleaseResponse, error) {
|
||||||
|
return &rls.InstallReleaseResponse{
|
||||||
|
Release: c.Rels[0],
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteRelease returns nil, nil
|
||||||
|
func (c *FakeReleaseClient) DeleteRelease(rlsName string, opts ...DeleteOption) (*rls.UninstallReleaseResponse, error) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateRelease returns nil, nil
|
||||||
|
func (c *FakeReleaseClient) UpdateRelease(rlsName string, chStr string, opts ...UpdateOption) (*rls.UpdateReleaseResponse, error) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetVersion returns a fake version
|
||||||
|
func (c *FakeReleaseClient) GetVersion(opts ...VersionOption) (*rls.GetVersionResponse, error) {
|
||||||
|
return &rls.GetVersionResponse{
|
||||||
|
Version: &version.Version{
|
||||||
|
SemVer: "1.2.3-fakeclient+testonly",
|
||||||
|
},
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateReleaseFromChart returns nil, nil
|
||||||
|
func (c *FakeReleaseClient) UpdateReleaseFromChart(rlsName string, chart *chart.Chart, opts ...UpdateOption) (*rls.UpdateReleaseResponse, error) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RollbackRelease returns nil, nil
|
||||||
|
func (c *FakeReleaseClient) RollbackRelease(rlsName string, opts ...RollbackOption) (*rls.RollbackReleaseResponse, error) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReleaseStatus returns a release status response with info from the first release in the fake
|
||||||
|
// release client
|
||||||
|
func (c *FakeReleaseClient) ReleaseStatus(rlsName string, opts ...StatusOption) (*rls.GetReleaseStatusResponse, error) {
|
||||||
|
if c.Rels[0] != nil {
|
||||||
|
return &rls.GetReleaseStatusResponse{
|
||||||
|
Name: c.Rels[0].Name,
|
||||||
|
Info: c.Rels[0].Info,
|
||||||
|
Namespace: c.Rels[0].Namespace,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("No such release: %s", rlsName)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReleaseContent returns the configuration for the first release in the fake release client
|
||||||
|
func (c *FakeReleaseClient) ReleaseContent(rlsName string, opts ...ContentOption) (resp *rls.GetReleaseContentResponse, err error) {
|
||||||
|
if len(c.Rels) > 0 {
|
||||||
|
resp = &rls.GetReleaseContentResponse{
|
||||||
|
Release: c.Rels[0],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return resp, c.Err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReleaseHistory returns a release's revision history.
|
||||||
|
func (c *FakeReleaseClient) ReleaseHistory(rlsName string, opts ...HistoryOption) (*rls.GetHistoryResponse, error) {
|
||||||
|
return &rls.GetHistoryResponse{Releases: c.Rels}, c.Err
|
||||||
|
}
|
||||||
|
|
||||||
|
// RunReleaseTest executes a pre-defined tests on a release
|
||||||
|
func (c *FakeReleaseClient) RunReleaseTest(rlsName string, opts ...ReleaseTestOption) (<-chan *rls.TestReleaseResponse, <-chan error) {
|
||||||
|
|
||||||
|
results := make(chan *rls.TestReleaseResponse)
|
||||||
|
errc := make(chan error, 1)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
for m, s := range c.Responses {
|
||||||
|
wg.Add(1)
|
||||||
|
|
||||||
|
go func(msg string, status release.TestRun_Status) {
|
||||||
|
defer wg.Done()
|
||||||
|
results <- &rls.TestReleaseResponse{Msg: msg, Status: status}
|
||||||
|
}(m, s)
|
||||||
|
}
|
||||||
|
|
||||||
|
wg.Wait()
|
||||||
|
close(results)
|
||||||
|
close(errc)
|
||||||
|
}()
|
||||||
|
|
||||||
|
return results, errc
|
||||||
|
}
|
||||||
|
|
||||||
|
// Option returns the fake release client
|
||||||
|
func (c *FakeReleaseClient) Option(opt ...Option) Interface {
|
||||||
|
return c
|
||||||
|
}
|
||||||
Loading…
Reference in new issue