github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-16 15:20:17 +00:00
Code Issues Projects Releases Wiki Activity

use require to simplify testcases

Browse Source
This commit is contained in:
cncal 2022-04-09 10:45:13 +08:00
parent a64b9cee21
commit fa1d1edbef
1 changed files with 10 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -17,6 +17,7 @@ limitations under the License.
package state
import (
"github.com/stretchr/testify/require"
"io/ioutil"
"os"
"reflect"
@ -201,15 +202,11 @@ func TestCheckpointStateRestore(t *testing.T) {
// create temp dir
testingDir, err := ioutil.TempDir("", "cpumanager_state_test")
if err != nil {
t.Fatal(err)
}
require.NoError(t, err)
defer os.RemoveAll(testingDir)
// create checkpoint manager for testing
cpm, err := checkpointmanager.NewCheckpointManager(testingDir)
if err != nil {
t.Fatalf("could not create testing checkpoint manager: %v", err)
}
require.NoErrorf(t, err, "could not create testing checkpoint manager: %v", err)
for _, tc := range testCases {
t.Run(tc.description, func(t *testing.T) {
@ -219,26 +216,19 @@ func TestCheckpointStateRestore(t *testing.T) {
// prepare checkpoint for testing
if strings.TrimSpace(tc.checkpointContent) != "" {
checkpoint := &testutil.MockCheckpoint{Content: tc.checkpointContent}
if err := cpm.CreateCheckpoint(testingCheckpoint, checkpoint); err != nil {
t.Fatalf("could not create testing checkpoint: %v", err)
}
err = cpm.CreateCheckpoint(testingCheckpoint, checkpoint)
require.NoErrorf(t, err, "could not create testing checkpoint: %v", err)
}
restoredState, err := NewCheckpointState(testingDir, testingCheckpoint, tc.policyName, tc.initialContainers)
if strings.TrimSpace(tc.expectedError) == "" {
if err != nil {
t.Fatalf("unexpected error while creating checkpointState: %v", err)
}
require.NoError(t, 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)
require.Error(t, err)
require.Contains(t, err.Error(), "could not restore state from checkpoint")
require.Contains(t, err.Error(), tc.expectedError)
return
}
}
// compare state after restoration with the one expected
AssertStateEqual(t, restoredState, tc.expectedState)