github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-25 04:33:26 +00:00
Code Issues Projects Releases Wiki Activity

memory manager: provide unittest for the state package

Browse Source 
The commit includes tests to verify the functionallity:
- to restore state from the file
- to store the state to the file
- to clean the state from old data

Signed-off-by: Artyom Lukianov <alukiano@redhat.com>
This commit is contained in:
Artyom Lukianov 2020-10-08 17:05:12 +03:00
parent 48ca6e53e6
commit 86df524948
1 changed files with 390 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

390
pkg/kubelet/cm/memorymanager/state/state_checkpoint_test.go Normal file
View File

@ -0,0 +1,390 @@
/*
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 state
import (
"io/ioutil"
"os"
"strings"
"testing"
"github.com/stretchr/testify/assert"
v1 "k8s.io/api/core/v1"
"k8s.io/kubernetes/pkg/kubelet/checkpointmanager"
"k8s.io/kubernetes/pkg/kubelet/cm/containermap"
testutil "k8s.io/kubernetes/pkg/kubelet/cm/cpumanager/state/testing"
)
const testingCheckpoint = "memorymanager_checkpoint_test"
// assertStateEqual marks provided test as failed if provided states differ
func assertStateEqual(t *testing.T, restoredState, expectedState State) {
expectedMachineState := expectedState.GetMachineState()
restoredMachineState := restoredState.GetMachineState()
assert.Equal(t, expectedMachineState, restoredMachineState, "expected MachineState does not equal to restored one")
expectedMemoryAssignments := expectedState.GetMemoryAssignments()
restoredMemoryAssignments := restoredState.GetMemoryAssignments()
assert.Equal(t, expectedMemoryAssignments, restoredMemoryAssignments, "state memory assignments mismatch")
}
func TestCheckpointStateRestore(t *testing.T) {
testCases := []struct {
description string
checkpointContent string
initialContainers containermap.ContainerMap
expectedError string
expectedState *stateMemory
}{
{
"Restore non-existing checkpoint",
"",
containermap.ContainerMap{},
"",
&stateMemory{},
},
{
"Restore valid checkpoint",
`{
"policyName":"static",
"machineState":{"0":{"numberOfAssignments":0,"memoryMap":{"memory":{"total":2048,"systemReserved":512,"allocatable":1536,"reserved":512,"free":1024}},"nodes":[]}},
"entries":{"pod":{"container1":[{"numaAffinity":[0],"type":"memory","size":512}]}},
"checksum": 163710462
}`,
containermap.ContainerMap{},
"",
&stateMemory{
assignments: ContainerMemoryAssignments{
"pod": map[string][]Block{
"container1": {
{
NUMAAffinity: []int{0},
Type: v1.ResourceMemory,
Size: 512,
},
},
},
},
machineState: NodeMap{
0: &NodeState{
MemoryMap: map[v1.ResourceName]*MemoryTable{
v1.ResourceMemory: {
Allocatable: 1536,
Free: 1024,
Reserved: 512,
SystemReserved: 512,
TotalMemSize: 2048,
},
},
},
},
},
},
{
"Restore checkpoint with invalid checksum",
`{
"policyName":"static",
"machineState":{"0":{"numberOfAssignments":0,"memoryMap":{"memory":{"total":2048,"systemReserved":512,"allocatable":1536,"reserved":512,"free":1024}},"nodes":[]}},
"entries":{"pod":{"container1":[{"affinity":[0],"type":"memory","size":512}]}},
"checksum": 101010
}`,
containermap.ContainerMap{},
"checkpoint is corrupted",
&stateMemory{},
},
{
"Restore checkpoint with invalid JSON",
`{`,
containermap.ContainerMap{},
"unexpected end of JSON input",
&stateMemory{},
},
}
// create temp dir
testingDir, err := ioutil.TempDir("", "memorymanager_state_test")
if err != nil {
t.Fatal(err)
}
defer os.RemoveAll(testingDir)
// create checkpoint manager for testing
cpm, err := checkpointmanager.NewCheckpointManager(testingDir)
assert.NoError(t, err, "could not create testing checkpoint manager")
for _, tc := range testCases {
t.Run(tc.description, func(t *testing.T) {
// ensure there is no previous checkpoint
assert.NoError(t, cpm.RemoveCheckpoint(testingCheckpoint), "could not remove testing checkpoint")
// prepare checkpoint for testing
if strings.TrimSpace(tc.checkpointContent) != "" {
checkpoint := &testutil.MockCheckpoint{Content: tc.checkpointContent}
assert.NoError(t, cpm.CreateCheckpoint(testingCheckpoint, checkpoint), "could not create testing checkpoint")
}
restoredState, err := NewCheckpointState(testingDir, testingCheckpoint, "static", tc.initialContainers)
if strings.TrimSpace(tc.expectedError) != "" {
assert.Error(t, err)
assert.Contains(t, err.Error(), "could not restore state from checkpoint: "+tc.expectedError)
} else {
assert.NoError(t, err, "unexpected error while creating checkpointState")
// compare state after restoration with the one expected
assertStateEqual(t, restoredState, tc.expectedState)
}
})
}
}
func TestCheckpointStateStore(t *testing.T) {
expectedState := &stateMemory{
assignments: ContainerMemoryAssignments{
"pod": map[string][]Block{
"container1": {
{
NUMAAffinity: []int{0},
Type: v1.ResourceMemory,
Size: 1024,
},
},
},
},
machineState: NodeMap{
0: &NodeState{
MemoryMap: map[v1.ResourceName]*MemoryTable{
v1.ResourceMemory: {
Allocatable: 1536,
Free: 512,
Reserved: 1024,
SystemReserved: 512,
TotalMemSize: 2048,
},
},
},
},
}
// create temp dir
testingDir, err := ioutil.TempDir("", "memorymanager_state_test")
if err != nil {
t.Fatal(err)
}
defer os.RemoveAll(testingDir)
cpm, err := checkpointmanager.NewCheckpointManager(testingDir)
assert.NoError(t, err, "could not create testing checkpoint manager")
assert.NoError(t, cpm.RemoveCheckpoint(testingCheckpoint), "could not remove testing checkpoint")
cs1, err := NewCheckpointState(testingDir, testingCheckpoint, "static", nil)
assert.NoError(t, err, "could not create testing checkpointState instance")
// set values of cs1 instance so they are stored in checkpoint and can be read by cs2
cs1.SetMachineState(expectedState.machineState)
cs1.SetMemoryAssignments(expectedState.assignments)
// restore checkpoint with previously stored values
cs2, err := NewCheckpointState(testingDir, testingCheckpoint, "static", nil)
assert.NoError(t, err, "could not create testing checkpointState instance")
assertStateEqual(t, cs2, expectedState)
}
func TestCheckpointStateHelpers(t *testing.T) {
testCases := []struct {
description string
machineState NodeMap
assignments ContainerMemoryAssignments
}{
{
description: "One container",
assignments: ContainerMemoryAssignments{
"pod": map[string][]Block{
"container1": {
{
NUMAAffinity: []int{0},
Type: v1.ResourceMemory,
Size: 1024,
},
},
},
},
machineState: NodeMap{
0: &NodeState{
MemoryMap: map[v1.ResourceName]*MemoryTable{
v1.ResourceMemory: {
Allocatable: 1536,
Free: 512,
Reserved: 1024,
SystemReserved: 512,
TotalMemSize: 2048,
},
},
Nodes: []int{},
},
},
},
{
description: "Two containers",
assignments: ContainerMemoryAssignments{
"pod": map[string][]Block{
"container1": {
{
NUMAAffinity: []int{0},
Type: v1.ResourceMemory,
Size: 512,
},
},
"container2": {
{
NUMAAffinity: []int{0},
Type: v1.ResourceMemory,
Size: 512,
},
},
},
},
machineState: NodeMap{
0: &NodeState{
MemoryMap: map[v1.ResourceName]*MemoryTable{
v1.ResourceMemory: {
Allocatable: 1536,
Free: 512,
Reserved: 1024,
SystemReserved: 512,
TotalMemSize: 2048,
},
},
Nodes: []int{},
},
},
},
{
description: "Container without assigned memory",
assignments: ContainerMemoryAssignments{
"pod": map[string][]Block{
"container1": {},
},
},
machineState: NodeMap{
0: &NodeState{
MemoryMap: map[v1.ResourceName]*MemoryTable{
v1.ResourceMemory: {
Allocatable: 1536,
Free: 1536,
Reserved: 0,
SystemReserved: 512,
TotalMemSize: 2048,
},
},
Nodes: []int{},
},
},
},
}
// create temp dir
testingDir, err := ioutil.TempDir("", "memorymanager_state_test")
if err != nil {
t.Fatal(err)
}
defer os.RemoveAll(testingDir)
cpm, err := checkpointmanager.NewCheckpointManager(testingDir)
assert.NoError(t, err, "could not create testing checkpoint manager")
for _, tc := range testCases {
t.Run(tc.description, func(t *testing.T) {
// ensure there is no previous checkpoint
assert.NoError(t, cpm.RemoveCheckpoint(testingCheckpoint), "could not remove testing checkpoint")
state, err := NewCheckpointState(testingDir, testingCheckpoint, "static", nil)
assert.NoError(t, err, "could not create testing checkpoint manager")
state.SetMachineState(tc.machineState)
assert.Equal(t, tc.machineState, state.GetMachineState(), "machine state inconsistent")
for pod := range tc.assignments {
for container, blocks := range tc.assignments[pod] {
state.SetMemoryBlocks(pod, container, blocks)
assert.Equal(t, blocks, state.GetMemoryBlocks(pod, container), "memory block inconsistent")
state.Delete(pod, container)
assert.Nil(t, state.GetMemoryBlocks(pod, container), "deleted container still existing in state")
}
}
})
}
}
func TestCheckpointStateClear(t *testing.T) {
testCases := []struct {
description string
machineState NodeMap
assignments ContainerMemoryAssignments
}{
{
description: "Valid state cleaning",
assignments: ContainerMemoryAssignments{
"pod": map[string][]Block{
"container1": {
{
NUMAAffinity: []int{0},
Type: v1.ResourceMemory,
Size: 1024,
},
},
},
},
machineState: NodeMap{
0: &NodeState{
MemoryMap: map[v1.ResourceName]*MemoryTable{
v1.ResourceMemory: {
Allocatable: 1536,
Free: 512,
Reserved: 1024,
SystemReserved: 512,
TotalMemSize: 2048,
},
},
},
},
},
}
// create temp dir
testingDir, err := ioutil.TempDir("", "memorymanager_state_test")
if err != nil {
t.Fatal(err)
}
defer os.RemoveAll(testingDir)
for _, tc := range testCases {
t.Run(tc.description, func(t *testing.T) {
state, err := NewCheckpointState(testingDir, testingCheckpoint, "static", nil)
assert.NoError(t, err, "could not create testing checkpoint manager")
state.SetMachineState(tc.machineState)
state.SetMemoryAssignments(tc.assignments)
state.ClearState()
assert.Equal(t, NodeMap{}, state.GetMachineState(), "cleared state with non-empty machine state")
assert.Equal(t, ContainerMemoryAssignments{}, state.GetMemoryAssignments(), "cleared state with non-empty memory assignments")
})
}
}