diff --git a/pkg/model/defaults/BUILD.bazel b/pkg/model/defaults/BUILD.bazel index a750351763dce..6dac5e9b0ad37 100644 --- a/pkg/model/defaults/BUILD.bazel +++ b/pkg/model/defaults/BUILD.bazel @@ -1,4 +1,4 @@ -load("@io_bazel_rules_go//go:def.bzl", "go_library") +load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test") go_library( name = "go_default_library", @@ -7,3 +7,10 @@ go_library( visibility = ["//visibility:public"], deps = ["//pkg/apis/kops:go_default_library"], ) + +go_test( + name = "go_default_test", + srcs = ["volumes_test.go"], + embed = [":go_default_library"], + deps = ["//pkg/apis/kops:go_default_library"], +) diff --git a/pkg/model/defaults/volumes_test.go b/pkg/model/defaults/volumes_test.go new file mode 100644 index 0000000000000..1bffc9526c60c --- /dev/null +++ b/pkg/model/defaults/volumes_test.go @@ -0,0 +1,38 @@ +/* +Copyright 2020 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 defaults + +import ( + "testing" + + "k8s.io/kops/pkg/apis/kops" +) + +func TestDefaultInstanceGroupVolumeSize(t *testing.T) { + tests := []struct { + role kops.InstanceGroupRole + expected int32 + }{ + { + role: "Node2", + expected: -1, + }, + } + for _, test := range tests { + result, _ := DefaultInstanceGroupVolumeSize(test.role) + if test.expected != result { + t.Errorf("Expected %d, got %d", test.expected, result) + } + } +}