mirror of https://github.com/k3s-io/k3s
add unit tests for volume rounding helpers
Andrew Kim
parent
a62ee8b0a9
commit
6d90f52f15
|
|
@ -20,10 +20,14 @@ go_library(
|
|||
|
||||
go_test(
|
||||
name = "go_default_test",
|
||||
srcs = ["zones_test.go"],
|
||||
srcs = [
|
||||
"rounding_test.go",
|
||||
"zones_test.go",
|
||||
],
|
||||
embed = [":go_default_library"],
|
||||
deps = [
|
||||
"//staging/src/k8s.io/api/core/v1:go_default_library",
|
||||
"//staging/src/k8s.io/apimachinery/pkg/api/resource:go_default_library",
|
||||
"//staging/src/k8s.io/apimachinery/pkg/util/sets:go_default_library",
|
||||
],
|
||||
)
|
||||
|
|
|
|||
|
|
@ -0,0 +1,323 @@
|
|||
/*
|
||||
Copyright 2019 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 helpers
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"k8s.io/apimachinery/pkg/api/resource"
|
||||
)
|
||||
|
||||
func Test_RoundUpToGB(t *testing.T) {
|
||||
testcases := []struct {
|
||||
name string
|
||||
resource resource.Quantity
|
||||
roundedVal int64
|
||||
}{
|
||||
{
|
||||
name: "round Ki to GB",
|
||||
resource: resource.MustParse("1000Ki"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round k to GB",
|
||||
resource: resource.MustParse("1000k"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round Mi to GB",
|
||||
resource: resource.MustParse("1000Mi"),
|
||||
roundedVal: int64(2),
|
||||
},
|
||||
{
|
||||
name: "round M to GB",
|
||||
resource: resource.MustParse("1000M"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round G to GB",
|
||||
resource: resource.MustParse("1000G"),
|
||||
roundedVal: int64(1000),
|
||||
},
|
||||
{
|
||||
name: "round Gi to GB",
|
||||
resource: resource.MustParse("1000Gi"),
|
||||
roundedVal: int64(1074),
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range testcases {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
val := RoundUpToGB(test.resource)
|
||||
if val != test.roundedVal {
|
||||
t.Logf("actual rounded value: %d", val)
|
||||
t.Logf("expected rounded value: %d", test.roundedVal)
|
||||
t.Error("unexpected rounded value")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_RoundUpToGiB(t *testing.T) {
|
||||
testcases := []struct {
|
||||
name string
|
||||
resource resource.Quantity
|
||||
roundedVal int64
|
||||
}{
|
||||
{
|
||||
name: "round Ki to GiB",
|
||||
resource: resource.MustParse("1000Ki"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round k to GiB",
|
||||
resource: resource.MustParse("1000k"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round Mi to GiB",
|
||||
resource: resource.MustParse("1000Mi"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round M to GiB",
|
||||
resource: resource.MustParse("1000M"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round G to GiB",
|
||||
resource: resource.MustParse("1000G"),
|
||||
roundedVal: int64(932),
|
||||
},
|
||||
{
|
||||
name: "round Gi to GiB",
|
||||
resource: resource.MustParse("1000Gi"),
|
||||
roundedVal: int64(1000),
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range testcases {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
val := RoundUpToGiB(test.resource)
|
||||
if val != test.roundedVal {
|
||||
t.Logf("actual rounded value: %d", val)
|
||||
t.Logf("expected rounded value: %d", test.roundedVal)
|
||||
t.Error("unexpected rounded value")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_RoundUpToMB(t *testing.T) {
|
||||
testcases := []struct {
|
||||
name string
|
||||
resource resource.Quantity
|
||||
roundedVal int64
|
||||
}{
|
||||
{
|
||||
name: "round Ki to MB",
|
||||
resource: resource.MustParse("1000Ki"),
|
||||
roundedVal: int64(2),
|
||||
},
|
||||
{
|
||||
name: "round k to MB",
|
||||
resource: resource.MustParse("1000k"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round Mi to MB",
|
||||
resource: resource.MustParse("1000Mi"),
|
||||
roundedVal: int64(1049),
|
||||
},
|
||||
{
|
||||
name: "round M to MB",
|
||||
resource: resource.MustParse("1000M"),
|
||||
roundedVal: int64(1000),
|
||||
},
|
||||
{
|
||||
name: "round G to MB",
|
||||
resource: resource.MustParse("1000G"),
|
||||
roundedVal: int64(1000000),
|
||||
},
|
||||
{
|
||||
name: "round Gi to MB",
|
||||
resource: resource.MustParse("1000Gi"),
|
||||
roundedVal: int64(1073742),
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range testcases {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
val := RoundUpToMB(test.resource)
|
||||
if val != test.roundedVal {
|
||||
t.Logf("actual rounded value: %d", val)
|
||||
t.Logf("expected rounded value: %d", test.roundedVal)
|
||||
t.Error("unexpected rounded value")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_RoundUpToMiB(t *testing.T) {
|
||||
testcases := []struct {
|
||||
name string
|
||||
resource resource.Quantity
|
||||
roundedVal int64
|
||||
}{
|
||||
{
|
||||
name: "round Ki to MiB",
|
||||
resource: resource.MustParse("1000Ki"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round k to MiB",
|
||||
resource: resource.MustParse("1000k"),
|
||||
roundedVal: int64(1),
|
||||
},
|
||||
{
|
||||
name: "round Mi to MiB",
|
||||
resource: resource.MustParse("1000Mi"),
|
||||
roundedVal: int64(1000),
|
||||
},
|
||||
{
|
||||
name: "round M to MiB",
|
||||
resource: resource.MustParse("1000M"),
|
||||
roundedVal: int64(954),
|
||||
},
|
||||
{
|
||||
name: "round G to MiB",
|
||||
resource: resource.MustParse("1000G"),
|
||||
roundedVal: int64(953675),
|
||||
},
|
||||
{
|
||||
name: "round Gi to MiB",
|
||||
resource: resource.MustParse("1000Gi"),
|
||||
roundedVal: int64(1024000),
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range testcases {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
val := RoundUpToMiB(test.resource)
|
||||
if val != test.roundedVal {
|
||||
t.Logf("actual rounded value: %d", val)
|
||||
t.Logf("expected rounded value: %d", test.roundedVal)
|
||||
t.Error("unexpected rounded value")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_RoundUpToKB(t *testing.T) {
|
||||
testcases := []struct {
|
||||
name string
|
||||
resource resource.Quantity
|
||||
roundedVal int64
|
||||
}{
|
||||
{
|
||||
name: "round Ki to KB",
|
||||
resource: resource.MustParse("1000Ki"),
|
||||
roundedVal: int64(1024),
|
||||
},
|
||||
{
|
||||
name: "round k to KB",
|
||||
resource: resource.MustParse("1000k"),
|
||||
roundedVal: int64(1000),
|
||||
},
|
||||
{
|
||||
name: "round Mi to KB",
|
||||
resource: resource.MustParse("1000Mi"),
|
||||
roundedVal: int64(1048576),
|
||||
},
|
||||
{
|
||||
name: "round M to KB",
|
||||
resource: resource.MustParse("1000M"),
|
||||
roundedVal: int64(1000000),
|
||||
},
|
||||
{
|
||||
name: "round G to KB",
|
||||
resource: resource.MustParse("1000G"),
|
||||
roundedVal: int64(1000000000),
|
||||
},
|
||||
{
|
||||
name: "round Gi to KB",
|
||||
resource: resource.MustParse("1000Gi"),
|
||||
roundedVal: int64(1073741824),
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range testcases {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
val := RoundUpToKB(test.resource)
|
||||
if val != test.roundedVal {
|
||||
t.Logf("actual rounded value: %d", val)
|
||||
t.Logf("expected rounded value: %d", test.roundedVal)
|
||||
t.Error("unexpected rounded value")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_RoundUpToKiB(t *testing.T) {
|
||||
testcases := []struct {
|
||||
name string
|
||||
resource resource.Quantity
|
||||
roundedVal int64
|
||||
}{
|
||||
{
|
||||
name: "round Ki to KiB",
|
||||
resource: resource.MustParse("1000Ki"),
|
||||
roundedVal: int64(1000),
|
||||
},
|
||||
{
|
||||
name: "round k to KiB",
|
||||
resource: resource.MustParse("1000k"),
|
||||
roundedVal: int64(977),
|
||||
},
|
||||
{
|
||||
name: "round Mi to KiB",
|
||||
resource: resource.MustParse("1000Mi"),
|
||||
roundedVal: int64(1024000),
|
||||
},
|
||||
{
|
||||
name: "round M to KiB",
|
||||
resource: resource.MustParse("1000M"),
|
||||
roundedVal: int64(976563),
|
||||
},
|
||||
{
|
||||
name: "round G to KiB",
|
||||
resource: resource.MustParse("1000G"),
|
||||
roundedVal: int64(976562500),
|
||||
},
|
||||
{
|
||||
name: "round Gi to KiB",
|
||||
resource: resource.MustParse("1000Gi"),
|
||||
roundedVal: int64(1048576000),
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range testcases {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
val := RoundUpToKiB(test.resource)
|
||||
if val != test.roundedVal {
|
||||
t.Logf("actual rounded value: %d", val)
|
||||
t.Logf("expected rounded value: %d", test.roundedVal)
|
||||
t.Error("unexpected rounded value")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue