mirror of https://github.com/helm/helm
Merge pull request #2550 from michelleN/export-fake-release-client
ref(helm): make FakeReleaseClient publicreviewable/pr2612/r1
Michelle Noorali
7 years ago
committed by
GitHub
commit
29c4709b5e
@ -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"
|
||||
)
|
||||
|
||||
// FakeClient implements Interface
|
||||
type FakeClient struct {
|
||||
Rels []*release.Release
|
||||
Responses map[string]release.TestRun_Status
|
||||
Err error
|
||||
}
|
||||
|
||||
var _ Interface = &FakeClient{}
|
||||
var _ Interface = (*FakeClient)(nil)
|
||||
|
||||
// ListReleases lists the current releases
|
||||
func (c *FakeClient) 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 *FakeClient) 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 *FakeClient) 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 *FakeClient) DeleteRelease(rlsName string, opts ...DeleteOption) (*rls.UninstallReleaseResponse, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
// UpdateRelease returns nil, nil
|
||||
func (c *FakeClient) UpdateRelease(rlsName string, chStr string, opts ...UpdateOption) (*rls.UpdateReleaseResponse, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
// GetVersion returns a fake version
|
||||
func (c *FakeClient) 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 *FakeClient) UpdateReleaseFromChart(rlsName string, chart *chart.Chart, opts ...UpdateOption) (*rls.UpdateReleaseResponse, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
// RollbackRelease returns nil, nil
|
||||
func (c *FakeClient) 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 *FakeClient) 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 *FakeClient) 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 *FakeClient) 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 *FakeClient) 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 *FakeClient) Option(opt ...Option) Interface {
|
||||
return c
|
||||
}
|
||||
Loading…
Reference in new issue