79 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			79 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| /*
 | |
|    Copyright The containerd 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 integration
 | |
| 
 | |
| import (
 | |
| 	"syscall"
 | |
| 	"testing"
 | |
| 	"time"
 | |
| 
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| 	runtime "k8s.io/cri-api/pkg/apis/runtime/v1"
 | |
| )
 | |
| 
 | |
| func TestContainerdRestartSandboxRecover(t *testing.T) {
 | |
| 	sbStatuses := map[string]runtime.PodSandboxState{
 | |
| 		// Sandbox with unknown status will be NotReady when returned from ListPodSandbox
 | |
| 		"sandbox_unknown":   runtime.PodSandboxState_SANDBOX_NOTREADY,
 | |
| 		"sandbox_not_ready": runtime.PodSandboxState_SANDBOX_NOTREADY,
 | |
| 		"sandbox_ready":     runtime.PodSandboxState_SANDBOX_READY,
 | |
| 	}
 | |
| 
 | |
| 	sbReadyConfig := PodSandboxConfig("sandbox_ready", "sandbox_ready")
 | |
| 	_, err := runtimeService.RunPodSandbox(sbReadyConfig, *runtimeHandler)
 | |
| 	assert.NoError(t, err)
 | |
| 
 | |
| 	sbNotReadyConfig := PodSandboxConfig("sandbox_not_ready", "sandbox_not_ready")
 | |
| 	notReadyID, err := runtimeService.RunPodSandbox(sbNotReadyConfig, *runtimeHandler)
 | |
| 	assert.NoError(t, err)
 | |
| 	err = runtimeService.StopPodSandbox(notReadyID)
 | |
| 	assert.NoError(t, err)
 | |
| 
 | |
| 	t.Logf("Create a pod config with shim create delay")
 | |
| 	sbUnknownConfig := PodSandboxConfig("sandbox_unknown", "sandbox_unknown_status")
 | |
| 	injectShimFailpoint(t, sbUnknownConfig, map[string]string{
 | |
| 		"Create": "1*delay(2000)",
 | |
| 	})
 | |
| 	waitCh := make(chan struct{})
 | |
| 	go func() {
 | |
| 		time.Sleep(time.Second)
 | |
| 		t.Logf("Create a sandbox with shim create delay")
 | |
| 		RestartContainerd(t, syscall.SIGTERM)
 | |
| 		waitCh <- struct{}{}
 | |
| 	}()
 | |
| 	t.Logf("Create a sandbox with shim create delay")
 | |
| 	_, err = runtimeService.RunPodSandbox(sbUnknownConfig, failpointRuntimeHandler)
 | |
| 	assert.Error(t, err)
 | |
| 	<-waitCh
 | |
| 	sbs, err := runtimeService.ListPodSandbox(nil)
 | |
| 	assert.NoError(t, err)
 | |
| 	foundUnkownSb := false
 | |
| 	for _, sb := range sbs {
 | |
| 		if sb.Metadata.Name == "sandbox_unknown" {
 | |
| 			foundUnkownSb = true
 | |
| 		}
 | |
| 		if status, ok := sbStatuses[sb.Metadata.Name]; ok {
 | |
| 			assert.Equal(t, status, sb.State)
 | |
| 			err = runtimeService.StopPodSandbox(sb.Id)
 | |
| 			assert.NoError(t, err)
 | |
| 			err = runtimeService.RemovePodSandbox(sb.Id)
 | |
| 			assert.NoError(t, err)
 | |
| 		}
 | |
| 	}
 | |
| 	assert.True(t, foundUnkownSb)
 | |
| }
 | 
