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:
|
||||
// ignore
|
||||
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)
|
||||
}
|
||||
}
|
||||
|
|
|
@ -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