mirror of https://github.com/helm/helm
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
322 lines
9.2 KiB
322 lines
9.2 KiB
/*
|
|
Copyright The Helm Authors.
|
|
|
|
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 repo
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"k8s.io/helm/pkg/getter"
|
|
"k8s.io/helm/pkg/helm/environment"
|
|
"k8s.io/helm/pkg/proto/hapi/chart"
|
|
)
|
|
|
|
const (
|
|
testRepository = "testdata/repository"
|
|
testURL = "http://example-charts.com"
|
|
)
|
|
|
|
func TestLoadChartRepository(t *testing.T) {
|
|
r, err := NewChartRepository(&Entry{
|
|
Name: testRepository,
|
|
URL: testURL,
|
|
}, getter.All(environment.EnvSettings{}))
|
|
if err != nil {
|
|
t.Errorf("Problem creating chart repository from %s: %v", testRepository, err)
|
|
}
|
|
|
|
if err := r.Load(); err != nil {
|
|
t.Errorf("Problem loading chart repository from %s: %v", testRepository, err)
|
|
}
|
|
|
|
paths := []string{
|
|
filepath.Join(testRepository, "frobnitz-1.2.3.tgz"),
|
|
filepath.Join(testRepository, "sprocket-1.1.0.tgz"),
|
|
filepath.Join(testRepository, "sprocket-1.2.0.tgz"),
|
|
filepath.Join(testRepository, "universe/zarthal-1.0.0.tgz"),
|
|
}
|
|
|
|
if r.Config.Name != testRepository {
|
|
t.Errorf("Expected %s as Name but got %s", testRepository, r.Config.Name)
|
|
}
|
|
|
|
if !reflect.DeepEqual(r.ChartPaths, paths) {
|
|
t.Errorf("Expected %#v but got %#v\n", paths, r.ChartPaths)
|
|
}
|
|
|
|
if r.Config.URL != testURL {
|
|
t.Errorf("Expected url for chart repository to be %s but got %s", testURL, r.Config.URL)
|
|
}
|
|
}
|
|
|
|
func TestIndex(t *testing.T) {
|
|
r, err := NewChartRepository(&Entry{
|
|
Name: testRepository,
|
|
URL: testURL,
|
|
}, getter.All(environment.EnvSettings{}))
|
|
if err != nil {
|
|
t.Errorf("Problem creating chart repository from %s: %v", testRepository, err)
|
|
}
|
|
|
|
if err := r.Load(); err != nil {
|
|
t.Errorf("Problem loading chart repository from %s: %v", testRepository, err)
|
|
}
|
|
|
|
err = r.Index()
|
|
if err != nil {
|
|
t.Errorf("Error performing index: %v\n", err)
|
|
}
|
|
|
|
tempIndexPath := filepath.Join(testRepository, indexPath)
|
|
actual, err := LoadIndexFile(tempIndexPath)
|
|
defer os.Remove(tempIndexPath) // clean up
|
|
if err != nil {
|
|
t.Errorf("Error loading index file %v", err)
|
|
}
|
|
verifyIndex(t, actual)
|
|
|
|
// Re-index and test again.
|
|
err = r.Index()
|
|
if err != nil {
|
|
t.Errorf("Error performing re-index: %s\n", err)
|
|
}
|
|
second, err := LoadIndexFile(tempIndexPath)
|
|
if err != nil {
|
|
t.Errorf("Error re-loading index file %v", err)
|
|
}
|
|
verifyIndex(t, second)
|
|
}
|
|
|
|
func verifyIndex(t *testing.T, actual *IndexFile) {
|
|
var empty time.Time
|
|
if actual.Generated == empty {
|
|
t.Errorf("Generated should be greater than 0: %s", actual.Generated)
|
|
}
|
|
|
|
if actual.APIVersion != APIVersionV1 {
|
|
t.Error("Expected v1 API")
|
|
}
|
|
|
|
entries := actual.Entries
|
|
if numEntries := len(entries); numEntries != 3 {
|
|
t.Errorf("Expected 3 charts to be listed in index file but got %v", numEntries)
|
|
}
|
|
|
|
expects := map[string]ChartVersions{
|
|
"frobnitz": {
|
|
{
|
|
Metadata: &chart.Metadata{
|
|
Name: "frobnitz",
|
|
Version: "1.2.3",
|
|
},
|
|
},
|
|
},
|
|
"sprocket": {
|
|
{
|
|
Metadata: &chart.Metadata{
|
|
Name: "sprocket",
|
|
Version: "1.2.0",
|
|
},
|
|
},
|
|
{
|
|
Metadata: &chart.Metadata{
|
|
Name: "sprocket",
|
|
Version: "1.1.0",
|
|
},
|
|
},
|
|
},
|
|
"zarthal": {
|
|
{
|
|
Metadata: &chart.Metadata{
|
|
Name: "zarthal",
|
|
Version: "1.0.0",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for name, versions := range expects {
|
|
got, ok := entries[name]
|
|
if !ok {
|
|
t.Errorf("Could not find %q entry", name)
|
|
continue
|
|
}
|
|
if len(versions) != len(got) {
|
|
t.Errorf("Expected %d versions, got %d", len(versions), len(got))
|
|
continue
|
|
}
|
|
for i, e := range versions {
|
|
g := got[i]
|
|
if e.Name != g.Name {
|
|
t.Errorf("Expected %q, got %q", e.Name, g.Name)
|
|
}
|
|
if e.Version != g.Version {
|
|
t.Errorf("Expected %q, got %q", e.Version, g.Version)
|
|
}
|
|
if len(g.Keywords) != 3 {
|
|
t.Error("Expected 3 keyrwords.")
|
|
}
|
|
if len(g.Maintainers) != 2 {
|
|
t.Error("Expected 2 maintainers.")
|
|
}
|
|
if g.Created == empty {
|
|
t.Error("Expected created to be non-empty")
|
|
}
|
|
if g.Description == "" {
|
|
t.Error("Expected description to be non-empty")
|
|
}
|
|
if g.Home == "" {
|
|
t.Error("Expected home to be non-empty")
|
|
}
|
|
if g.Digest == "" {
|
|
t.Error("Expected digest to be non-empty")
|
|
}
|
|
if len(g.URLs) != 1 {
|
|
t.Error("Expected exactly 1 URL")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// startLocalServerForTests Start the local helm server
|
|
func startLocalServerForTests(handler http.Handler) (*httptest.Server, error) {
|
|
if handler == nil {
|
|
fileBytes, err := ioutil.ReadFile("testdata/local-index.yaml")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write(fileBytes)
|
|
})
|
|
}
|
|
|
|
return httptest.NewServer(handler), nil
|
|
}
|
|
|
|
func TestFindChartInRepoURL(t *testing.T) {
|
|
srv, err := startLocalServerForTests(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer srv.Close()
|
|
|
|
chartURL, err := FindChartInRepoURL(srv.URL, "nginx", "", "", "", "", getter.All(environment.EnvSettings{}))
|
|
if err != nil {
|
|
t.Errorf("%s", err)
|
|
}
|
|
if chartURL != "https://kubernetes-charts.storage.googleapis.com/nginx-0.2.0.tgz" {
|
|
t.Errorf("%s is not the valid URL", chartURL)
|
|
}
|
|
|
|
chartURL, err = FindChartInRepoURL(srv.URL, "nginx", "0.1.0", "", "", "", getter.All(environment.EnvSettings{}))
|
|
if err != nil {
|
|
t.Errorf("%s", err)
|
|
}
|
|
if chartURL != "https://kubernetes-charts.storage.googleapis.com/nginx-0.1.0.tgz" {
|
|
t.Errorf("%s is not the valid URL", chartURL)
|
|
}
|
|
}
|
|
|
|
func TestErrorFindChartInRepoURL(t *testing.T) {
|
|
_, err := FindChartInRepoURL("http://someserver/something", "nginx", "", "", "", "", getter.All(environment.EnvSettings{}))
|
|
if err == nil {
|
|
t.Errorf("Expected error for bad chart URL, but did not get any errors")
|
|
}
|
|
if err != nil && !strings.Contains(err.Error(), `Looks like "http://someserver/something" is not a valid chart repository or cannot be reached: Get http://someserver/something/index.yaml`) {
|
|
t.Errorf("Expected error for bad chart URL, but got a different error (%v)", err)
|
|
}
|
|
|
|
srv, err := startLocalServerForTests(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer srv.Close()
|
|
|
|
_, err = FindChartInRepoURL(srv.URL, "nginx1", "", "", "", "", getter.All(environment.EnvSettings{}))
|
|
if err == nil {
|
|
t.Errorf("Expected error for chart not found, but did not get any errors")
|
|
}
|
|
if err != nil && err.Error() != `chart "nginx1" not found in `+srv.URL+` repository` {
|
|
t.Errorf("Expected error for chart not found, but got a different error (%v)", err)
|
|
}
|
|
|
|
_, err = FindChartInRepoURL(srv.URL, "nginx1", "0.1.0", "", "", "", getter.All(environment.EnvSettings{}))
|
|
if err == nil {
|
|
t.Errorf("Expected error for chart not found, but did not get any errors")
|
|
}
|
|
if err != nil && err.Error() != `chart "nginx1" version "0.1.0" not found in `+srv.URL+` repository` {
|
|
t.Errorf("Expected error for chart not found, but got a different error (%v)", err)
|
|
}
|
|
|
|
_, err = FindChartInRepoURL(srv.URL, "chartWithNoURL", "", "", "", "", getter.All(environment.EnvSettings{}))
|
|
if err == nil {
|
|
t.Errorf("Expected error for no chart URLs available, but did not get any errors")
|
|
}
|
|
if err != nil && err.Error() != `chart "chartWithNoURL" has no downloadable URLs` {
|
|
t.Errorf("Expected error for chart not found, but got a different error (%v)", err)
|
|
}
|
|
}
|
|
|
|
func TestResolveReferenceURL(t *testing.T) {
|
|
chartURL, err := ResolveReferenceURL("http://localhost:8123/charts/", "nginx-0.2.0.tgz")
|
|
if err != nil {
|
|
t.Errorf("%s", err)
|
|
}
|
|
if chartURL != "http://localhost:8123/charts/nginx-0.2.0.tgz" {
|
|
t.Errorf("%s", chartURL)
|
|
}
|
|
|
|
chartURL, err = ResolveReferenceURL("http://localhost:8123/charts", "nginx-0.2.0.tgz")
|
|
if err != nil {
|
|
t.Errorf("%s", err)
|
|
}
|
|
if chartURL != "http://localhost:8123/charts/nginx-0.2.0.tgz" {
|
|
t.Errorf("%s", chartURL)
|
|
}
|
|
|
|
chartURL, err = ResolveReferenceURL("http://localhost:8123/charts/?st=2018-08-06T22%3A59%3A04Z&se=2018-08-07T22%3A59%3A04Z&sp=rl&sv=2018-03-28&sr=c&sig=cyqM4%2F5G7HNk%2F3faaHSDMaWxFxefCglvZlYSnmQBwiY%3D", "nginx-0.2.0.tgz")
|
|
if err != nil {
|
|
t.Errorf("%s", err)
|
|
}
|
|
if chartURL != "http://localhost:8123/charts/nginx-0.2.0.tgz?st=2018-08-06T22%3A59%3A04Z&se=2018-08-07T22%3A59%3A04Z&sp=rl&sv=2018-03-28&sr=c&sig=cyqM4%2F5G7HNk%2F3faaHSDMaWxFxefCglvZlYSnmQBwiY%3D" {
|
|
t.Errorf("%s does not contain the query string of the base URL", chartURL)
|
|
}
|
|
|
|
chartURL, err = ResolveReferenceURL("http://localhost:8123", "https://kubernetes-charts.storage.googleapis.com/nginx-0.2.0.tgz")
|
|
if err != nil {
|
|
t.Errorf("%s", err)
|
|
}
|
|
if chartURL != "https://kubernetes-charts.storage.googleapis.com/nginx-0.2.0.tgz" {
|
|
t.Errorf("%s", chartURL)
|
|
}
|
|
|
|
chartURL, err = ResolveReferenceURL("http://localhost:8123/?querystring", "https://kubernetes-charts.storage.googleapis.com/nginx-0.2.0.tgz")
|
|
if err != nil {
|
|
t.Errorf("%s", err)
|
|
}
|
|
if chartURL != "https://kubernetes-charts.storage.googleapis.com/nginx-0.2.0.tgz" {
|
|
t.Errorf("%s contains query string from base URL when it shouldn't", chartURL)
|
|
}
|
|
}
|