api: add Resource.Canonicalize test and fix tests to handle ReservedCores field

This commit is contained in:
Nick Ethier
2021-03-19 22:08:27 -04:00
parent 94c7aec159
commit ee5b13a77b
3 changed files with 58 additions and 1 deletions

View File

@@ -652,6 +652,7 @@ func TestJobs_Canonicalize(t *testing.T) {
},
Resources: &Resources{
CPU: intToPtr(500),
Cores: intToPtr(0),
MemoryMB: intToPtr(256),
Networks: []*NetworkResource{
{

55
api/resources_test.go Normal file
View File

@@ -0,0 +1,55 @@
package api
import (
"reflect"
"testing"
"github.com/kr/pretty"
)
func TestResources_Canonicalize(t *testing.T) {
testCases := []struct {
name string
input *Resources
expected *Resources
}{
{
name: "empty",
input: &Resources{},
expected: DefaultResources(),
},
{
name: "cores",
input: &Resources{
Cores: intToPtr(2),
MemoryMB: intToPtr(1024),
},
expected: &Resources{
CPU: intToPtr(0),
Cores: intToPtr(2),
MemoryMB: intToPtr(1024),
},
},
{
name: "cpu",
input: &Resources{
CPU: intToPtr(500),
MemoryMB: intToPtr(1024),
},
expected: &Resources{
CPU: intToPtr(500),
Cores: intToPtr(0),
MemoryMB: intToPtr(1024),
},
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
tc.input.Canonicalize()
if !reflect.DeepEqual(tc.input, tc.expected) {
t.Fatalf("Name: %v, Diffs:\n%v", tc.name, pretty.Diff(tc.expected, tc.input))
}
})
}
}