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.
334 lines
9.5 KiB
334 lines
9.5 KiB
/*
|
|
Copyright 2015 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 repository
|
|
|
|
import (
|
|
"github.com/kubernetes/deployment-manager/manager/manager"
|
|
|
|
"testing"
|
|
)
|
|
|
|
func TestRepositoryListEmpty(t *testing.T) {
|
|
r := NewMapBasedRepository()
|
|
d, err := r.ListDeployments()
|
|
if err != nil {
|
|
t.Fatal("List Deployments failed")
|
|
}
|
|
if len(d) != 0 {
|
|
t.Fatal("Returned non zero list")
|
|
}
|
|
}
|
|
|
|
func TestRepositoryGetFailsWithNonExistentDeployment(t *testing.T) {
|
|
r := NewMapBasedRepository()
|
|
_, err := r.GetDeployment("nothere")
|
|
if err == nil {
|
|
t.Fatal("GetDeployment didn't fail with non-existent deployment")
|
|
}
|
|
if err.Error() != "deployment nothere not found" {
|
|
t.Fatal("Error message doesn't match")
|
|
}
|
|
}
|
|
|
|
func TestRepositoryCreateDeploymentWorks(t *testing.T) {
|
|
var deploymentName = "mydeployment"
|
|
var manifestName = "manifest-0"
|
|
r := NewMapBasedRepository()
|
|
manifest := manager.Manifest{Deployment: deploymentName, Name: manifestName}
|
|
d, err := r.CreateDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("CreateDeployment failed: %v", err)
|
|
}
|
|
l, err := r.ListDeployments()
|
|
if err != nil {
|
|
t.Fatalf("ListDeployments failed: %v", err)
|
|
}
|
|
if len(l) != 1 {
|
|
t.Fatalf("List of deployments is not 1: %d", len(l))
|
|
}
|
|
dNew, err := r.GetDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("GetDeployment failed: %v", err)
|
|
}
|
|
if dNew.Name != d.Name {
|
|
t.Fatalf("Deployment Names don't match, got: %v, expected %v", dNew, d)
|
|
}
|
|
if len(dNew.Manifests) != 0 {
|
|
t.Fatalf("Deployment has non-zero manifest count: %d", len(dNew.Manifests))
|
|
}
|
|
err = r.AddManifest(deploymentName, &manifest)
|
|
if err != nil {
|
|
t.Fatalf("AddManifest failed: %v", err)
|
|
}
|
|
dNew, err = r.GetDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("GetDeployment failed: %v", err)
|
|
}
|
|
if len(dNew.Manifests) != 1 {
|
|
t.Fatalf("Fetched deployment does not have manifest count of 1: %d", len(dNew.Manifests))
|
|
}
|
|
manifestList, err := r.ListManifests(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("ListManifests failed: %v", err)
|
|
}
|
|
if len(manifestList) != 1 {
|
|
t.Fatalf("ManifestList does not have manifest count of 1: %d", len(manifestList))
|
|
}
|
|
m, err := r.GetManifest(deploymentName, manifestName)
|
|
if err != nil {
|
|
t.Fatalf("GetManifest failed: %v", err)
|
|
}
|
|
if m.Name != manifestName {
|
|
t.Fatalf("manifest name doesn't match: %v", m)
|
|
}
|
|
}
|
|
|
|
func TestRepositoryMultipleManifestsWorks(t *testing.T) {
|
|
var deploymentName = "mydeployment"
|
|
var manifestName = "manifest-0"
|
|
var manifestName2 = "manifest-1"
|
|
r := NewMapBasedRepository()
|
|
manifest := manager.Manifest{Deployment: deploymentName, Name: manifestName}
|
|
manifest2 := manager.Manifest{Deployment: deploymentName, Name: manifestName2}
|
|
d, err := r.CreateDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("CreateDeployment failed: %v", err)
|
|
}
|
|
dNew, err := r.GetDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("GetDeployment failed: %v", err)
|
|
}
|
|
if dNew.Name != d.Name {
|
|
t.Fatalf("Deployment Names don't match, got: %v, expected %v", dNew, d)
|
|
}
|
|
if len(dNew.Manifests) != 0 {
|
|
t.Fatalf("Deployment has non-zero manifest count: %d", len(dNew.Manifests))
|
|
}
|
|
err = r.AddManifest(deploymentName, &manifest)
|
|
if err != nil {
|
|
t.Fatalf("AddManifest failed: %v", err)
|
|
}
|
|
dNew, err = r.GetDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("GetDeployment failed: %v", err)
|
|
}
|
|
if len(dNew.Manifests) != 1 {
|
|
t.Fatalf("Fetched deployment does not have manifest count of 1: %d", len(dNew.Manifests))
|
|
}
|
|
manifestList, err := r.ListManifests(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("ListManifests failed: %v", err)
|
|
}
|
|
if len(manifestList) != 1 {
|
|
t.Fatalf("ManifestList does not have manifest count of 1: %d", len(manifestList))
|
|
}
|
|
m, err := r.GetManifest(deploymentName, manifestName)
|
|
if err != nil {
|
|
t.Fatalf("GetManifest failed: %v", err)
|
|
}
|
|
if m.Name != manifestName {
|
|
t.Fatalf("manifest name doesn't match: %v", m)
|
|
}
|
|
err = r.AddManifest(deploymentName, &manifest2)
|
|
if err != nil {
|
|
t.Fatalf("AddManifest failed: %v", err)
|
|
}
|
|
dNew, err = r.GetDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("GetDeployment failed: %v", err)
|
|
}
|
|
if len(dNew.Manifests) != 2 {
|
|
t.Fatalf("Fetched deployment does not have manifest count of 2: %d", len(dNew.Manifests))
|
|
}
|
|
manifestList, err = r.ListManifests(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("ListManifests failed: %v", err)
|
|
}
|
|
if len(manifestList) != 2 {
|
|
t.Fatalf("ManifestList does not have manifest count of 1: %d", len(manifestList))
|
|
}
|
|
m, err = r.GetManifest(deploymentName, manifestName)
|
|
if err != nil {
|
|
t.Fatalf("GetManifest failed: %v", err)
|
|
}
|
|
if m.Name != manifestName {
|
|
t.Fatalf("manifest name doesn't match: %v", m)
|
|
}
|
|
m, err = r.GetManifest(deploymentName, manifestName2)
|
|
if err != nil {
|
|
t.Fatalf("GetManifest failed: %v", err)
|
|
}
|
|
if m.Name != manifestName2 {
|
|
t.Fatalf("manifest name doesn't match: %v", m)
|
|
}
|
|
}
|
|
|
|
func TestRepositoryDeleteFailsWithNonExistentDeployment(t *testing.T) {
|
|
var deploymentName = "mydeployment"
|
|
r := NewMapBasedRepository()
|
|
d, err := r.DeleteDeployment(deploymentName, false)
|
|
if err == nil {
|
|
t.Fatalf("DeleteDeployment didn't fail with non existent deployment")
|
|
}
|
|
if d != nil {
|
|
t.Fatalf("DeleteDeployment returned non-nil for non existent deployment")
|
|
}
|
|
}
|
|
|
|
func TestRepositoryDeleteWorksWithNoLatestManifest(t *testing.T) {
|
|
var deploymentName = "mydeployment"
|
|
r := NewMapBasedRepository()
|
|
_, err := r.CreateDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("CreateDeployment failed: %v", err)
|
|
}
|
|
dDeleted, err := r.DeleteDeployment(deploymentName, false)
|
|
if err != nil {
|
|
t.Fatalf("DeleteDeployment failed: %v", err)
|
|
}
|
|
if dDeleted.Status != manager.DeletedStatus {
|
|
t.Fatalf("Deployment Status is not deleted")
|
|
}
|
|
if len(dDeleted.Manifests) != 0 {
|
|
t.Fatalf("manifests count is not 0, is: %d", len(dDeleted.Manifests))
|
|
}
|
|
}
|
|
|
|
func TestRepositoryDeleteDeploymentWorksNoForget(t *testing.T) {
|
|
var deploymentName = "mydeployment"
|
|
var manifestName = "manifest-0"
|
|
r := NewMapBasedRepository()
|
|
manifest := manager.Manifest{Deployment: deploymentName, Name: manifestName}
|
|
_, err := r.CreateDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("CreateDeployment failed: %v", err)
|
|
}
|
|
err = r.AddManifest(deploymentName, &manifest)
|
|
if err != nil {
|
|
t.Fatalf("AddManifest failed: %v", err)
|
|
}
|
|
dDeleted, err := r.DeleteDeployment(deploymentName, false)
|
|
if err != nil {
|
|
t.Fatalf("DeleteDeployment failed: %v", err)
|
|
}
|
|
if dDeleted.Status != manager.DeletedStatus {
|
|
t.Fatalf("Deployment Status is not deleted")
|
|
}
|
|
}
|
|
|
|
func TestRepositoryDeleteDeploymentWorksForget(t *testing.T) {
|
|
var deploymentName = "mydeployment"
|
|
var manifestName = "manifest-0"
|
|
r := NewMapBasedRepository()
|
|
manifest := manager.Manifest{Deployment: deploymentName, Name: manifestName}
|
|
_, err := r.CreateDeployment(deploymentName)
|
|
if err != nil {
|
|
t.Fatalf("CreateDeployment failed: %v", err)
|
|
}
|
|
err = r.AddManifest(deploymentName, &manifest)
|
|
if err != nil {
|
|
t.Fatalf("AddManifest failed: %v", err)
|
|
}
|
|
dDeleted, err := r.DeleteDeployment(deploymentName, true)
|
|
if err != nil {
|
|
t.Fatalf("DeleteDeployment failed: %v", err)
|
|
}
|
|
if dDeleted.Status != manager.CreatedStatus {
|
|
t.Fatalf("Deployment Status is not created")
|
|
}
|
|
}
|
|
|
|
func TestRepositoryTypeInstances(t *testing.T) {
|
|
r := NewMapBasedRepository()
|
|
|
|
d1Map := map[string][]*manager.TypeInstance{
|
|
"t1": []*manager.TypeInstance{
|
|
&manager.TypeInstance{
|
|
Name: "i1",
|
|
Type: "t1",
|
|
Deployment: "d1",
|
|
Manifest: "m1",
|
|
Path: "p1",
|
|
},
|
|
},
|
|
}
|
|
|
|
d2Map := map[string][]*manager.TypeInstance{
|
|
"t2": []*manager.TypeInstance{
|
|
&manager.TypeInstance{
|
|
Name: "i2",
|
|
Type: "t2",
|
|
Deployment: "d2",
|
|
Manifest: "m2",
|
|
Path: "p2",
|
|
},
|
|
},
|
|
}
|
|
|
|
d3Map := map[string][]*manager.TypeInstance{
|
|
"t2": []*manager.TypeInstance{
|
|
&manager.TypeInstance{
|
|
Name: "i3",
|
|
Type: "t2",
|
|
Deployment: "d3",
|
|
Manifest: "m3",
|
|
Path: "p3",
|
|
},
|
|
},
|
|
}
|
|
|
|
if instances := r.GetTypeInstances("noinstances"); len(instances) != 0 {
|
|
t.Fatalf("expected no instances: %v", instances)
|
|
}
|
|
|
|
if types := r.ListTypes(); len(types) != 0 {
|
|
t.Fatalf("expected no types: %v", types)
|
|
}
|
|
|
|
r.SetTypeInstances("d1", d1Map)
|
|
r.SetTypeInstances("d2", d2Map)
|
|
r.SetTypeInstances("d3", d3Map)
|
|
|
|
if instances := r.GetTypeInstances("unknowntype"); len(instances) != 0 {
|
|
t.Fatalf("expected no instances: %v", instances)
|
|
}
|
|
|
|
if instances := r.GetTypeInstances("t1"); len(instances) != 1 {
|
|
t.Fatalf("expected one instance: %v", instances)
|
|
}
|
|
|
|
if instances := r.GetTypeInstances("t2"); len(instances) != 2 {
|
|
t.Fatalf("expected two instances: %v", instances)
|
|
}
|
|
|
|
if instances := r.GetTypeInstances("all"); len(instances) != 3 {
|
|
t.Fatalf("expected three total instances: %v", instances)
|
|
}
|
|
|
|
if types := r.ListTypes(); len(types) != 2 {
|
|
t.Fatalf("expected two total types: %v", types)
|
|
}
|
|
|
|
r.ClearTypeInstances("d1")
|
|
if instances := r.GetTypeInstances("t1"); len(instances) != 0 {
|
|
t.Fatalf("expected no instances after clear: %v", instances)
|
|
}
|
|
|
|
if types := r.ListTypes(); len(types) != 1 {
|
|
t.Fatalf("expected one total type: %v", types)
|
|
}
|
|
}
|
|
|
|
// TODO(vaikas): Add more tests
|