Using array to build sub-tests is to avoid random pick. The shuffle thing should be handled by go-test framework. And we should capture range var before runing sub-test. Signed-off-by: Wei Fu <fuweid89@gmail.com>
		
			
				
	
	
		
			664 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			664 lines
		
	
	
		
			18 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 server
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"os"
 | 
						|
	goruntime "runtime"
 | 
						|
	"strings"
 | 
						|
	"testing"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"github.com/containerd/containerd/containers"
 | 
						|
	"github.com/containerd/containerd/errdefs"
 | 
						|
	"github.com/containerd/containerd/oci"
 | 
						|
	criconfig "github.com/containerd/containerd/pkg/cri/config"
 | 
						|
	containerstore "github.com/containerd/containerd/pkg/cri/store/container"
 | 
						|
	imagestore "github.com/containerd/containerd/pkg/cri/store/image"
 | 
						|
	"github.com/containerd/containerd/plugin"
 | 
						|
	"github.com/containerd/containerd/protobuf/types"
 | 
						|
	"github.com/containerd/containerd/reference/docker"
 | 
						|
	runcoptions "github.com/containerd/containerd/runtime/v2/runc/options"
 | 
						|
	"github.com/containerd/typeurl/v2"
 | 
						|
 | 
						|
	imagedigest "github.com/opencontainers/go-digest"
 | 
						|
	runtimespec "github.com/opencontainers/runtime-spec/specs-go"
 | 
						|
	"github.com/pelletier/go-toml"
 | 
						|
	"github.com/stretchr/testify/assert"
 | 
						|
	"github.com/stretchr/testify/require"
 | 
						|
 | 
						|
	runtime "k8s.io/cri-api/pkg/apis/runtime/v1"
 | 
						|
)
 | 
						|
 | 
						|
// TestGetUserFromImage tests the logic of getting image uid or user name of image user.
 | 
						|
func TestGetUserFromImage(t *testing.T) {
 | 
						|
	newI64 := func(i int64) *int64 { return &i }
 | 
						|
	for _, test := range []struct {
 | 
						|
		desc string
 | 
						|
		user string
 | 
						|
		uid  *int64
 | 
						|
		name string
 | 
						|
	}{
 | 
						|
		{
 | 
						|
			desc: "no gid",
 | 
						|
			user: "0",
 | 
						|
			uid:  newI64(0),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "uid/gid",
 | 
						|
			user: "0:1",
 | 
						|
			uid:  newI64(0),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "empty user",
 | 
						|
			user: "",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "multiple separators",
 | 
						|
			user: "1:2:3",
 | 
						|
			uid:  newI64(1),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "root username",
 | 
						|
			user: "root:root",
 | 
						|
			name: "root",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "username",
 | 
						|
			user: "test:test",
 | 
						|
			name: "test",
 | 
						|
		},
 | 
						|
	} {
 | 
						|
		test := test
 | 
						|
		t.Run(test.desc, func(t *testing.T) {
 | 
						|
			actualUID, actualName := getUserFromImage(test.user)
 | 
						|
			assert.Equal(t, test.uid, actualUID)
 | 
						|
			assert.Equal(t, test.name, actualName)
 | 
						|
		})
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetRepoDigestAndTag(t *testing.T) {
 | 
						|
	digest := imagedigest.Digest("sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582")
 | 
						|
	for _, test := range []struct {
 | 
						|
		desc               string
 | 
						|
		ref                string
 | 
						|
		schema1            bool
 | 
						|
		expectedRepoDigest string
 | 
						|
		expectedRepoTag    string
 | 
						|
	}{
 | 
						|
		{
 | 
						|
			desc:               "repo tag should be empty if original ref has no tag",
 | 
						|
			ref:                "gcr.io/library/busybox@" + digest.String(),
 | 
						|
			expectedRepoDigest: "gcr.io/library/busybox@" + digest.String(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc:               "repo tag should not be empty if original ref has tag",
 | 
						|
			ref:                "gcr.io/library/busybox:latest",
 | 
						|
			expectedRepoDigest: "gcr.io/library/busybox@" + digest.String(),
 | 
						|
			expectedRepoTag:    "gcr.io/library/busybox:latest",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc:               "repo digest should be empty if original ref is schema1 and has no digest",
 | 
						|
			ref:                "gcr.io/library/busybox:latest",
 | 
						|
			schema1:            true,
 | 
						|
			expectedRepoDigest: "",
 | 
						|
			expectedRepoTag:    "gcr.io/library/busybox:latest",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc:               "repo digest should not be empty if original ref is schema1 but has digest",
 | 
						|
			ref:                "gcr.io/library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59594",
 | 
						|
			schema1:            true,
 | 
						|
			expectedRepoDigest: "gcr.io/library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59594",
 | 
						|
			expectedRepoTag:    "",
 | 
						|
		},
 | 
						|
	} {
 | 
						|
		test := test
 | 
						|
		t.Run(test.desc, func(t *testing.T) {
 | 
						|
			named, err := docker.ParseDockerRef(test.ref)
 | 
						|
			assert.NoError(t, err)
 | 
						|
			repoDigest, repoTag := getRepoDigestAndTag(named, digest, test.schema1)
 | 
						|
			assert.Equal(t, test.expectedRepoDigest, repoDigest)
 | 
						|
			assert.Equal(t, test.expectedRepoTag, repoTag)
 | 
						|
		})
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestBuildLabels(t *testing.T) {
 | 
						|
	imageConfigLabels := map[string]string{
 | 
						|
		"a":          "z",
 | 
						|
		"d":          "y",
 | 
						|
		"long-label": strings.Repeat("example", 10000),
 | 
						|
	}
 | 
						|
	configLabels := map[string]string{
 | 
						|
		"a": "b",
 | 
						|
		"c": "d",
 | 
						|
	}
 | 
						|
	newLabels := buildLabels(configLabels, imageConfigLabels, containerKindSandbox)
 | 
						|
	assert.Len(t, newLabels, 4)
 | 
						|
	assert.Equal(t, "b", newLabels["a"])
 | 
						|
	assert.Equal(t, "d", newLabels["c"])
 | 
						|
	assert.Equal(t, "y", newLabels["d"])
 | 
						|
	assert.Equal(t, containerKindSandbox, newLabels[containerKindLabel])
 | 
						|
	assert.NotContains(t, newLabels, "long-label")
 | 
						|
 | 
						|
	newLabels["a"] = "e"
 | 
						|
	assert.Empty(t, configLabels[containerKindLabel], "should not add new labels into original label")
 | 
						|
	assert.Equal(t, "b", configLabels["a"], "change in new labels should not affect original label")
 | 
						|
}
 | 
						|
 | 
						|
func TestParseImageReferences(t *testing.T) {
 | 
						|
	refs := []string{
 | 
						|
		"gcr.io/library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
 | 
						|
		"gcr.io/library/busybox:1.2",
 | 
						|
		"sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
 | 
						|
		"arbitrary-ref",
 | 
						|
	}
 | 
						|
	expectedTags := []string{
 | 
						|
		"gcr.io/library/busybox:1.2",
 | 
						|
	}
 | 
						|
	expectedDigests := []string{"gcr.io/library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582"}
 | 
						|
	tags, digests := parseImageReferences(refs)
 | 
						|
	assert.Equal(t, expectedTags, tags)
 | 
						|
	assert.Equal(t, expectedDigests, digests)
 | 
						|
}
 | 
						|
 | 
						|
func TestLocalResolve(t *testing.T) {
 | 
						|
	image := imagestore.Image{
 | 
						|
		ID:      "sha256:c75bebcdd211f41b3a460c7bf82970ed6c75acaab9cd4c9a4e125b03ca113799",
 | 
						|
		ChainID: "test-chain-id-1",
 | 
						|
		References: []string{
 | 
						|
			"docker.io/library/busybox:latest",
 | 
						|
			"docker.io/library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
 | 
						|
		},
 | 
						|
		Size: 10,
 | 
						|
	}
 | 
						|
	c := newTestCRIService()
 | 
						|
	var err error
 | 
						|
	c.imageStore, err = imagestore.NewFakeStore([]imagestore.Image{image})
 | 
						|
	assert.NoError(t, err)
 | 
						|
 | 
						|
	for _, ref := range []string{
 | 
						|
		"sha256:c75bebcdd211f41b3a460c7bf82970ed6c75acaab9cd4c9a4e125b03ca113799",
 | 
						|
		"busybox",
 | 
						|
		"busybox:latest",
 | 
						|
		"busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
 | 
						|
		"library/busybox",
 | 
						|
		"library/busybox:latest",
 | 
						|
		"library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
 | 
						|
		"docker.io/busybox",
 | 
						|
		"docker.io/busybox:latest",
 | 
						|
		"docker.io/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
 | 
						|
		"docker.io/library/busybox",
 | 
						|
		"docker.io/library/busybox:latest",
 | 
						|
		"docker.io/library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
 | 
						|
	} {
 | 
						|
		img, err := c.localResolve(ref)
 | 
						|
		assert.NoError(t, err)
 | 
						|
		assert.Equal(t, image, img)
 | 
						|
	}
 | 
						|
	img, err := c.localResolve("randomid")
 | 
						|
	assert.Equal(t, errdefs.IsNotFound(err), true)
 | 
						|
	assert.Equal(t, imagestore.Image{}, img)
 | 
						|
}
 | 
						|
 | 
						|
func TestGenerateRuntimeOptions(t *testing.T) {
 | 
						|
	nilOpts := `
 | 
						|
[containerd]
 | 
						|
  no_pivot = true
 | 
						|
  default_runtime_name = "default"
 | 
						|
[containerd.runtimes.runcv2]
 | 
						|
  runtime_type = "` + plugin.RuntimeRuncV2 + `"
 | 
						|
`
 | 
						|
	nonNilOpts := `
 | 
						|
[containerd]
 | 
						|
  no_pivot = true
 | 
						|
  default_runtime_name = "default"
 | 
						|
[containerd.runtimes.legacy.options]
 | 
						|
  Runtime = "legacy"
 | 
						|
  RuntimeRoot = "/legacy"
 | 
						|
[containerd.runtimes.runc.options]
 | 
						|
  BinaryName = "runc"
 | 
						|
  Root = "/runc"
 | 
						|
  NoNewKeyring = true
 | 
						|
[containerd.runtimes.runcv2]
 | 
						|
  runtime_type = "` + plugin.RuntimeRuncV2 + `"
 | 
						|
[containerd.runtimes.runcv2.options]
 | 
						|
  BinaryName = "runc"
 | 
						|
  Root = "/runcv2"
 | 
						|
  NoNewKeyring = true
 | 
						|
`
 | 
						|
	var nilOptsConfig, nonNilOptsConfig criconfig.Config
 | 
						|
	tree, err := toml.Load(nilOpts)
 | 
						|
	require.NoError(t, err)
 | 
						|
	err = tree.Unmarshal(&nilOptsConfig)
 | 
						|
	require.NoError(t, err)
 | 
						|
	require.Len(t, nilOptsConfig.Runtimes, 1)
 | 
						|
 | 
						|
	tree, err = toml.Load(nonNilOpts)
 | 
						|
	require.NoError(t, err)
 | 
						|
	err = tree.Unmarshal(&nonNilOptsConfig)
 | 
						|
	require.NoError(t, err)
 | 
						|
	require.Len(t, nonNilOptsConfig.Runtimes, 3)
 | 
						|
 | 
						|
	for _, test := range []struct {
 | 
						|
		desc            string
 | 
						|
		r               criconfig.Runtime
 | 
						|
		c               criconfig.Config
 | 
						|
		expectedOptions interface{}
 | 
						|
	}{
 | 
						|
		{
 | 
						|
			desc:            "when options is nil, should return nil option for io.containerd.runc.v2",
 | 
						|
			r:               nilOptsConfig.Runtimes["runcv2"],
 | 
						|
			c:               nilOptsConfig,
 | 
						|
			expectedOptions: nil,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "when options is not nil, should be able to decode for io.containerd.runc.v2",
 | 
						|
			r:    nonNilOptsConfig.Runtimes["runcv2"],
 | 
						|
			c:    nonNilOptsConfig,
 | 
						|
			expectedOptions: &runcoptions.Options{
 | 
						|
				BinaryName:   "runc",
 | 
						|
				Root:         "/runcv2",
 | 
						|
				NoNewKeyring: true,
 | 
						|
			},
 | 
						|
		},
 | 
						|
	} {
 | 
						|
		test := test
 | 
						|
		t.Run(test.desc, func(t *testing.T) {
 | 
						|
			opts, err := generateRuntimeOptions(test.r, test.c)
 | 
						|
			assert.NoError(t, err)
 | 
						|
			assert.Equal(t, test.expectedOptions, opts)
 | 
						|
		})
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestEnvDeduplication(t *testing.T) {
 | 
						|
	for _, test := range []struct {
 | 
						|
		desc     string
 | 
						|
		existing []string
 | 
						|
		kv       [][2]string
 | 
						|
		expected []string
 | 
						|
	}{
 | 
						|
		{
 | 
						|
			desc: "single env",
 | 
						|
			kv: [][2]string{
 | 
						|
				{"a", "b"},
 | 
						|
			},
 | 
						|
			expected: []string{"a=b"},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "multiple envs",
 | 
						|
			kv: [][2]string{
 | 
						|
				{"a", "b"},
 | 
						|
				{"c", "d"},
 | 
						|
				{"e", "f"},
 | 
						|
			},
 | 
						|
			expected: []string{
 | 
						|
				"a=b",
 | 
						|
				"c=d",
 | 
						|
				"e=f",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "env override",
 | 
						|
			kv: [][2]string{
 | 
						|
				{"k1", "v1"},
 | 
						|
				{"k2", "v2"},
 | 
						|
				{"k3", "v3"},
 | 
						|
				{"k3", "v4"},
 | 
						|
				{"k1", "v5"},
 | 
						|
				{"k4", "v6"},
 | 
						|
			},
 | 
						|
			expected: []string{
 | 
						|
				"k1=v5",
 | 
						|
				"k2=v2",
 | 
						|
				"k3=v4",
 | 
						|
				"k4=v6",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "existing env",
 | 
						|
			existing: []string{
 | 
						|
				"k1=v1",
 | 
						|
				"k2=v2",
 | 
						|
				"k3=v3",
 | 
						|
			},
 | 
						|
			kv: [][2]string{
 | 
						|
				{"k3", "v4"},
 | 
						|
				{"k2", "v5"},
 | 
						|
				{"k4", "v6"},
 | 
						|
			},
 | 
						|
			expected: []string{
 | 
						|
				"k1=v1",
 | 
						|
				"k2=v5",
 | 
						|
				"k3=v4",
 | 
						|
				"k4=v6",
 | 
						|
			},
 | 
						|
		},
 | 
						|
	} {
 | 
						|
		test := test
 | 
						|
		t.Run(test.desc, func(t *testing.T) {
 | 
						|
			var spec runtimespec.Spec
 | 
						|
			if len(test.existing) > 0 {
 | 
						|
				spec.Process = &runtimespec.Process{
 | 
						|
					Env: test.existing,
 | 
						|
				}
 | 
						|
			}
 | 
						|
			for _, kv := range test.kv {
 | 
						|
				oci.WithEnv([]string{kv[0] + "=" + kv[1]})(context.Background(), nil, nil, &spec)
 | 
						|
			}
 | 
						|
			assert.Equal(t, test.expected, spec.Process.Env)
 | 
						|
		})
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestPassThroughAnnotationsFilter(t *testing.T) {
 | 
						|
	for _, test := range []struct {
 | 
						|
		desc                   string
 | 
						|
		podAnnotations         map[string]string
 | 
						|
		runtimePodAnnotations  []string
 | 
						|
		passthroughAnnotations map[string]string
 | 
						|
	}{
 | 
						|
		{
 | 
						|
			desc:                   "should support direct match",
 | 
						|
			podAnnotations:         map[string]string{"c": "d", "d": "e"},
 | 
						|
			runtimePodAnnotations:  []string{"c"},
 | 
						|
			passthroughAnnotations: map[string]string{"c": "d"},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "should support wildcard match",
 | 
						|
			podAnnotations: map[string]string{
 | 
						|
				"t.f":  "j",
 | 
						|
				"z.g":  "o",
 | 
						|
				"z":    "o",
 | 
						|
				"y.ca": "b",
 | 
						|
				"y":    "b",
 | 
						|
			},
 | 
						|
			runtimePodAnnotations: []string{"*.f", "z*g", "y.c*"},
 | 
						|
			passthroughAnnotations: map[string]string{
 | 
						|
				"t.f":  "j",
 | 
						|
				"z.g":  "o",
 | 
						|
				"y.ca": "b",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "should support wildcard match all",
 | 
						|
			podAnnotations: map[string]string{
 | 
						|
				"t.f":  "j",
 | 
						|
				"z.g":  "o",
 | 
						|
				"z":    "o",
 | 
						|
				"y.ca": "b",
 | 
						|
				"y":    "b",
 | 
						|
			},
 | 
						|
			runtimePodAnnotations: []string{"*"},
 | 
						|
			passthroughAnnotations: map[string]string{
 | 
						|
				"t.f":  "j",
 | 
						|
				"z.g":  "o",
 | 
						|
				"z":    "o",
 | 
						|
				"y.ca": "b",
 | 
						|
				"y":    "b",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc: "should support match including path separator",
 | 
						|
			podAnnotations: map[string]string{
 | 
						|
				"matchend.com/end":    "1",
 | 
						|
				"matchend.com/end1":   "2",
 | 
						|
				"matchend.com/1end":   "3",
 | 
						|
				"matchmid.com/mid":    "4",
 | 
						|
				"matchmid.com/mi1d":   "5",
 | 
						|
				"matchmid.com/mid1":   "6",
 | 
						|
				"matchhead.com/head":  "7",
 | 
						|
				"matchhead.com/1head": "8",
 | 
						|
				"matchhead.com/head1": "9",
 | 
						|
				"matchall.com/abc":    "10",
 | 
						|
				"matchall.com/def":    "11",
 | 
						|
				"end/matchend":        "12",
 | 
						|
				"end1/matchend":       "13",
 | 
						|
				"1end/matchend":       "14",
 | 
						|
				"mid/matchmid":        "15",
 | 
						|
				"mi1d/matchmid":       "16",
 | 
						|
				"mid1/matchmid":       "17",
 | 
						|
				"head/matchhead":      "18",
 | 
						|
				"1head/matchhead":     "19",
 | 
						|
				"head1/matchhead":     "20",
 | 
						|
				"abc/matchall":        "21",
 | 
						|
				"def/matchall":        "22",
 | 
						|
				"match1/match2":       "23",
 | 
						|
				"nomatch/nomatch":     "24",
 | 
						|
			},
 | 
						|
			runtimePodAnnotations: []string{
 | 
						|
				"matchend.com/end*",
 | 
						|
				"matchmid.com/mi*d",
 | 
						|
				"matchhead.com/*head",
 | 
						|
				"matchall.com/*",
 | 
						|
				"end*/matchend",
 | 
						|
				"mi*d/matchmid",
 | 
						|
				"*head/matchhead",
 | 
						|
				"*/matchall",
 | 
						|
				"match*/match*",
 | 
						|
			},
 | 
						|
			passthroughAnnotations: map[string]string{
 | 
						|
				"matchend.com/end":    "1",
 | 
						|
				"matchend.com/end1":   "2",
 | 
						|
				"matchmid.com/mid":    "4",
 | 
						|
				"matchmid.com/mi1d":   "5",
 | 
						|
				"matchhead.com/head":  "7",
 | 
						|
				"matchhead.com/1head": "8",
 | 
						|
				"matchall.com/abc":    "10",
 | 
						|
				"matchall.com/def":    "11",
 | 
						|
				"end/matchend":        "12",
 | 
						|
				"end1/matchend":       "13",
 | 
						|
				"mid/matchmid":        "15",
 | 
						|
				"mi1d/matchmid":       "16",
 | 
						|
				"head/matchhead":      "18",
 | 
						|
				"1head/matchhead":     "19",
 | 
						|
				"abc/matchall":        "21",
 | 
						|
				"def/matchall":        "22",
 | 
						|
				"match1/match2":       "23",
 | 
						|
			},
 | 
						|
		},
 | 
						|
	} {
 | 
						|
		test := test
 | 
						|
		t.Run(test.desc, func(t *testing.T) {
 | 
						|
			passthroughAnnotations := getPassthroughAnnotations(test.podAnnotations, test.runtimePodAnnotations)
 | 
						|
			assert.Equal(t, test.passthroughAnnotations, passthroughAnnotations)
 | 
						|
		})
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestEnsureRemoveAllNotExist(t *testing.T) {
 | 
						|
	// should never return an error for a non-existent path
 | 
						|
	if err := ensureRemoveAll(context.Background(), "/non/existent/path"); err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestEnsureRemoveAllWithDir(t *testing.T) {
 | 
						|
	dir := t.TempDir()
 | 
						|
	if err := ensureRemoveAll(context.Background(), dir); err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestEnsureRemoveAllWithFile(t *testing.T) {
 | 
						|
	tmp, err := os.CreateTemp("", "test-ensure-removeall-with-dir")
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	tmp.Close()
 | 
						|
	if err := ensureRemoveAll(context.Background(), tmp.Name()); err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
// Helper function for setting up an environment to test PID namespace targeting.
 | 
						|
func addContainer(c *criService, containerID, sandboxID string, PID uint32, createdAt, startedAt, finishedAt int64) error {
 | 
						|
	meta := containerstore.Metadata{
 | 
						|
		ID:        containerID,
 | 
						|
		SandboxID: sandboxID,
 | 
						|
	}
 | 
						|
	status := containerstore.Status{
 | 
						|
		Pid:        PID,
 | 
						|
		CreatedAt:  createdAt,
 | 
						|
		StartedAt:  startedAt,
 | 
						|
		FinishedAt: finishedAt,
 | 
						|
	}
 | 
						|
	container, err := containerstore.NewContainer(meta,
 | 
						|
		containerstore.WithFakeStatus(status),
 | 
						|
	)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	return c.containerStore.Add(container)
 | 
						|
}
 | 
						|
 | 
						|
func TestValidateTargetContainer(t *testing.T) {
 | 
						|
	testSandboxID := "test-sandbox-uid"
 | 
						|
 | 
						|
	// The existing container that will be targeted.
 | 
						|
	testTargetContainerID := "test-target-container"
 | 
						|
	testTargetContainerPID := uint32(4567)
 | 
						|
 | 
						|
	// A container that has finished running and cannot be targeted.
 | 
						|
	testStoppedContainerID := "stopped-target-container"
 | 
						|
	testStoppedContainerPID := uint32(6789)
 | 
						|
 | 
						|
	// A container from another pod.
 | 
						|
	testOtherContainerSandboxID := "other-sandbox-uid"
 | 
						|
	testOtherContainerID := "other-target-container"
 | 
						|
	testOtherContainerPID := uint32(7890)
 | 
						|
 | 
						|
	// Container create/start/stop times.
 | 
						|
	createdAt := time.Now().Add(-15 * time.Second).UnixNano()
 | 
						|
	startedAt := time.Now().Add(-10 * time.Second).UnixNano()
 | 
						|
	finishedAt := time.Now().Add(-5 * time.Second).UnixNano()
 | 
						|
 | 
						|
	c := newTestCRIService()
 | 
						|
 | 
						|
	// Create a target container.
 | 
						|
	err := addContainer(c, testTargetContainerID, testSandboxID, testTargetContainerPID, createdAt, startedAt, 0)
 | 
						|
	require.NoError(t, err, "error creating test target container")
 | 
						|
 | 
						|
	// Create a stopped container.
 | 
						|
	err = addContainer(c, testStoppedContainerID, testSandboxID, testStoppedContainerPID, createdAt, startedAt, finishedAt)
 | 
						|
	require.NoError(t, err, "error creating test stopped container")
 | 
						|
 | 
						|
	// Create a container in another pod.
 | 
						|
	err = addContainer(c, testOtherContainerID, testOtherContainerSandboxID, testOtherContainerPID, createdAt, startedAt, 0)
 | 
						|
	require.NoError(t, err, "error creating test container in other pod")
 | 
						|
 | 
						|
	for _, test := range []struct {
 | 
						|
		desc              string
 | 
						|
		targetContainerID string
 | 
						|
		expectError       bool
 | 
						|
	}{
 | 
						|
		{
 | 
						|
			desc:              "target container in pod",
 | 
						|
			targetContainerID: testTargetContainerID,
 | 
						|
			expectError:       false,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc:              "target stopped container in pod",
 | 
						|
			targetContainerID: testStoppedContainerID,
 | 
						|
			expectError:       true,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc:              "target container does not exist",
 | 
						|
			targetContainerID: "no-container-with-this-id",
 | 
						|
			expectError:       true,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			desc:              "target container in other pod",
 | 
						|
			targetContainerID: testOtherContainerID,
 | 
						|
			expectError:       true,
 | 
						|
		},
 | 
						|
	} {
 | 
						|
		test := test
 | 
						|
		t.Run(test.desc, func(t *testing.T) {
 | 
						|
			targetContainer, err := c.validateTargetContainer(testSandboxID, test.targetContainerID)
 | 
						|
			if test.expectError {
 | 
						|
				require.Error(t, err, "target should have been invalid but no error")
 | 
						|
				return
 | 
						|
			}
 | 
						|
			require.NoErrorf(t, err, "target should have been valid but got error")
 | 
						|
 | 
						|
			assert.Equal(t, test.targetContainerID, targetContainer.ID, "returned target container does not have expected ID")
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
func TestGetRuntimeOptions(t *testing.T) {
 | 
						|
	_, err := getRuntimeOptions(containers.Container{})
 | 
						|
	require.NoError(t, err)
 | 
						|
 | 
						|
	var pbany *types.Any               // This is nil.
 | 
						|
	var typeurlAny typeurl.Any = pbany // This is typed nil.
 | 
						|
	_, err = getRuntimeOptions(containers.Container{Runtime: containers.RuntimeInfo{Options: typeurlAny}})
 | 
						|
	require.NoError(t, err)
 | 
						|
}
 | 
						|
 | 
						|
func TestHostNetwork(t *testing.T) {
 | 
						|
	tests := []struct {
 | 
						|
		name     string
 | 
						|
		c        *runtime.PodSandboxConfig
 | 
						|
		expected bool
 | 
						|
	}{
 | 
						|
		{
 | 
						|
			name: "when pod namespace return false",
 | 
						|
			c: &runtime.PodSandboxConfig{
 | 
						|
				Linux: &runtime.LinuxPodSandboxConfig{
 | 
						|
					SecurityContext: &runtime.LinuxSandboxSecurityContext{
 | 
						|
						NamespaceOptions: &runtime.NamespaceOption{
 | 
						|
							Network: runtime.NamespaceMode_POD,
 | 
						|
						},
 | 
						|
					},
 | 
						|
				},
 | 
						|
			},
 | 
						|
			expected: false,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "when node namespace return true",
 | 
						|
			c: &runtime.PodSandboxConfig{
 | 
						|
				Linux: &runtime.LinuxPodSandboxConfig{
 | 
						|
					SecurityContext: &runtime.LinuxSandboxSecurityContext{
 | 
						|
						NamespaceOptions: &runtime.NamespaceOption{
 | 
						|
							Network: runtime.NamespaceMode_NODE,
 | 
						|
						},
 | 
						|
					},
 | 
						|
				},
 | 
						|
			},
 | 
						|
			expected: true,
 | 
						|
		},
 | 
						|
	}
 | 
						|
 | 
						|
	for _, tt := range tests {
 | 
						|
		if goruntime.GOOS != "linux" {
 | 
						|
			t.Skip()
 | 
						|
		}
 | 
						|
		tt := tt
 | 
						|
		t.Run(tt.name, func(t *testing.T) {
 | 
						|
			if hostNetwork(tt.c) != tt.expected {
 | 
						|
				t.Errorf("failed hostNetwork got %t expected %t", hostNetwork(tt.c), tt.expected)
 | 
						|
			}
 | 
						|
		})
 | 
						|
	}
 | 
						|
}
 |