mirror of https://github.com/kubernetes/kops.git
A few tests for nodeup tasks
This commit is contained in:
parent
df9efe6591
commit
d6f86ca968
|
@ -0,0 +1,40 @@
|
||||||
|
/*
|
||||||
|
Copyright 2016 The Kubernetes 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 nodetasks
|
||||||
|
|
||||||
|
import (
|
||||||
|
"k8s.io/kops/upup/pkg/fi"
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestLoadImageTask_Deps(t *testing.T) {
|
||||||
|
l := &LoadImageTask{}
|
||||||
|
|
||||||
|
tasks := make(map[string]fi.Task)
|
||||||
|
tasks["LoadImageTask1"] = &LoadImageTask{}
|
||||||
|
tasks["FileTask1"] = &File{}
|
||||||
|
tasks["ServiceDocker"] = &Service{Name: "docker.service"}
|
||||||
|
tasks["Service2"] = &Service{Name: "two.service"}
|
||||||
|
|
||||||
|
deps := l.GetDependencies(tasks)
|
||||||
|
expected := []fi.Task{tasks["ServiceDocker"]}
|
||||||
|
if !reflect.DeepEqual(expected, deps) {
|
||||||
|
t.Fatalf("unexpected deps. expected=%v, actual=%v", expected, deps)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -69,7 +69,7 @@ func (p *Service) GetDependencies(tasks map[string]fi.Task) []fi.Task {
|
||||||
case *Service, *LoadImageTask:
|
case *Service, *LoadImageTask:
|
||||||
// ignore
|
// ignore
|
||||||
default:
|
default:
|
||||||
glog.Warningf("Unhandled type %t in Service::GetDependencies: %v", v, v)
|
glog.Warningf("Unhandled type %T in Service::GetDependencies: %v", v, v)
|
||||||
deps = append(deps, v)
|
deps = append(deps, v)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,56 @@
|
||||||
|
/*
|
||||||
|
Copyright 2016 The Kubernetes 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 nodetasks
|
||||||
|
|
||||||
|
import (
|
||||||
|
"k8s.io/kops/upup/pkg/fi"
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestServiceTask_Deps(t *testing.T) {
|
||||||
|
s := &Service{}
|
||||||
|
|
||||||
|
tasks := make(map[string]fi.Task)
|
||||||
|
tasks["LoadImageTask1"] = &LoadImageTask{}
|
||||||
|
tasks["FileTask1"] = &File{}
|
||||||
|
|
||||||
|
deps := s.GetDependencies(tasks)
|
||||||
|
expected := []fi.Task{tasks["FileTask1"]}
|
||||||
|
if !reflect.DeepEqual(expected, deps) {
|
||||||
|
t.Fatalf("unexpected deps. expected=%v, actual=%v", expected, deps)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type FakeTask struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *FakeTask) Run(*fi.Context) error {
|
||||||
|
panic("not implemented")
|
||||||
|
}
|
||||||
|
func TestServiceTask_UnknownTypes(t *testing.T) {
|
||||||
|
s := &Service{}
|
||||||
|
|
||||||
|
tasks := make(map[string]fi.Task)
|
||||||
|
tasks["FakeTask1"] = &FakeTask{}
|
||||||
|
|
||||||
|
deps := s.GetDependencies(tasks)
|
||||||
|
expected := []fi.Task{tasks["FakeTask1"]}
|
||||||
|
if !reflect.DeepEqual(expected, deps) {
|
||||||
|
t.Fatalf("unexpected deps. expected=%v, actual=%v", expected, deps)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue