116 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			116 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| //go:build !windows && !linux
 | |
| 
 | |
| /*
 | |
|    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 server
 | |
| 
 | |
| import (
 | |
| 	"testing"
 | |
| 
 | |
| 	imagespec "github.com/opencontainers/image-spec/specs-go/v1"
 | |
| 	runtimespec "github.com/opencontainers/runtime-spec/specs-go"
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| 	runtime "k8s.io/cri-api/pkg/apis/runtime/v1"
 | |
| 
 | |
| 	"github.com/containerd/containerd/v2/internal/cri/annotations"
 | |
| )
 | |
| 
 | |
| // checkMount is defined by all tests but not used here
 | |
| var _ = checkMount
 | |
| 
 | |
| func getCreateContainerTestData() (*runtime.ContainerConfig, *runtime.PodSandboxConfig,
 | |
| 	*imagespec.ImageConfig, func(*testing.T, string, string, uint32, *runtimespec.Spec)) {
 | |
| 	config := &runtime.ContainerConfig{
 | |
| 		Metadata: &runtime.ContainerMetadata{
 | |
| 			Name:    "test-name",
 | |
| 			Attempt: 1,
 | |
| 		},
 | |
| 		Image: &runtime.ImageSpec{
 | |
| 			Image: "sha256:c75bebcdd211f41b3a460c7bf82970ed6c75acaab9cd4c9a4e125b03ca113799",
 | |
| 		},
 | |
| 		Command:    []string{"test", "command"},
 | |
| 		Args:       []string{"test", "args"},
 | |
| 		WorkingDir: "test-cwd",
 | |
| 		Envs: []*runtime.KeyValue{
 | |
| 			{Key: "k1", Value: "v1"},
 | |
| 			{Key: "k2", Value: "v2"},
 | |
| 			{Key: "k3", Value: "v3=v3bis"},
 | |
| 			{Key: "k4", Value: "v4=v4bis=foop"},
 | |
| 		},
 | |
| 		Labels:      map[string]string{"a": "b"},
 | |
| 		Annotations: map[string]string{"ca-c": "ca-d"},
 | |
| 		Mounts: []*runtime.Mount{
 | |
| 			// everything default
 | |
| 			{
 | |
| 				ContainerPath: "container-path-1",
 | |
| 				HostPath:      "host-path-1",
 | |
| 			},
 | |
| 			// readOnly
 | |
| 			{
 | |
| 				ContainerPath: "container-path-2",
 | |
| 				HostPath:      "host-path-2",
 | |
| 				Readonly:      true,
 | |
| 			},
 | |
| 		},
 | |
| 	}
 | |
| 	sandboxConfig := &runtime.PodSandboxConfig{
 | |
| 		Metadata: &runtime.PodSandboxMetadata{
 | |
| 			Name:      "test-sandbox-name",
 | |
| 			Uid:       "test-sandbox-uid",
 | |
| 			Namespace: "test-sandbox-ns",
 | |
| 			Attempt:   2,
 | |
| 		},
 | |
| 		Annotations: map[string]string{"c": "d"},
 | |
| 	}
 | |
| 	imageConfig := &imagespec.ImageConfig{
 | |
| 		Env:        []string{"ik1=iv1", "ik2=iv2", "ik3=iv3=iv3bis", "ik4=iv4=iv4bis=boop"},
 | |
| 		Entrypoint: []string{"/entrypoint"},
 | |
| 		Cmd:        []string{"cmd"},
 | |
| 		WorkingDir: "/workspace",
 | |
| 	}
 | |
| 	specCheck := func(t *testing.T, id string, sandboxID string, sandboxPid uint32, spec *runtimespec.Spec) {
 | |
| 		assert.Equal(t, []string{"test", "command", "test", "args"}, spec.Process.Args)
 | |
| 		assert.Equal(t, "test-cwd", spec.Process.Cwd)
 | |
| 		assert.Contains(t, spec.Process.Env, "k1=v1", "k2=v2", "k3=v3=v3bis", "ik4=iv4=iv4bis=boop")
 | |
| 		assert.Contains(t, spec.Process.Env, "ik1=iv1", "ik2=iv2", "ik3=iv3=iv3bis", "k4=v4=v4bis=foop")
 | |
| 
 | |
| 		t.Logf("Check bind mount")
 | |
| 		checkMount(t, spec.Mounts, "host-path-1", "container-path-1", "bind", []string{"rw"}, nil)
 | |
| 		checkMount(t, spec.Mounts, "host-path-2", "container-path-2", "bind", []string{"ro"}, nil)
 | |
| 
 | |
| 		t.Logf("Check PodSandbox annotations")
 | |
| 		assert.Contains(t, spec.Annotations, annotations.SandboxID)
 | |
| 		assert.EqualValues(t, spec.Annotations[annotations.SandboxID], sandboxID)
 | |
| 
 | |
| 		assert.Contains(t, spec.Annotations, annotations.ContainerType)
 | |
| 		assert.EqualValues(t, spec.Annotations[annotations.ContainerType], annotations.ContainerTypeContainer)
 | |
| 
 | |
| 		assert.Contains(t, spec.Annotations, annotations.SandboxNamespace)
 | |
| 		assert.EqualValues(t, spec.Annotations[annotations.SandboxNamespace], "test-sandbox-ns")
 | |
| 
 | |
| 		assert.Contains(t, spec.Annotations, annotations.SandboxUID)
 | |
| 		assert.EqualValues(t, spec.Annotations[annotations.SandboxUID], "test-sandbox-uid")
 | |
| 
 | |
| 		assert.Contains(t, spec.Annotations, annotations.SandboxName)
 | |
| 		assert.EqualValues(t, spec.Annotations[annotations.SandboxName], "test-sandbox-name")
 | |
| 
 | |
| 		assert.Contains(t, spec.Annotations, annotations.ImageName)
 | |
| 		assert.EqualValues(t, spec.Annotations[annotations.ImageName], testImageName)
 | |
| 	}
 | |
| 	return config, sandboxConfig, imageConfig, specCheck
 | |
| }
 | 
