github/kubernetes
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix test for CheckpointStateRestore

Browse Source
This commit is contained in:
cncal 2022-01-21 12:53:08 +08:00
parent 7380fc735a
commit a64b9cee21
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pkg/kubelet/cm/cpumanager/state/state_checkpoint_test.go
View File

@ -225,15 +225,19 @@ func TestCheckpointStateRestore(t *testing.T) {
}
restoredState, err := NewCheckpointState(testingDir, testingCheckpoint, tc.policyName, tc.initialContainers)
if err != nil {
if strings.TrimSpace(tc.expectedError) != "" {
if strings.Contains(err.Error(), "could not restore state from checkpoint") &&
strings.Contains(err.Error(), tc.expectedError) {
t.Logf("got expected error: %v", err)
return
}
if strings.TrimSpace(tc.expectedError) == "" {
if err != nil {
t.Fatalf("unexpected error while creating checkpointState: %v", err)
}
} else {
if err == nil {
t.Fatalf("expected error: %s, got nil", tc.expectedError)
}
if strings.Contains(err.Error(), "could not restore state from checkpoint") &&
strings.Contains(err.Error(), tc.expectedError) {
t.Logf("got expected error: %v", err)
return
}
t.Fatalf("unexpected error while creatng checkpointState: %v", err)
}
// compare state after restoration with the one expected