nri/pkg/api/owners_test.go at main · containerd/nri · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
/*
   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 api_test

import (
	"testing"

	"github.com/containerd/nri/pkg/api"

	"github.com/stretchr/testify/require"
)

func TestSimpleClaims(t *testing.T) {
	o := api.NewOwningPlugins()

	// claim hooks
	err := o.ClaimHooks("ctr0", "test0")
	require.NoError(t, err, "hooks")

	// claim memory limit
	err = o.ClaimMemLimit("ctr0", "test0")
	require.NoError(t, err, "memory limit")

	// claim memory limit of another container
	err = o.ClaimMemLimit("ctr1", "test0")
	require.NoError(t, err, "another memory limit")

	// claim memory limit of same container by another plugin
	err = o.ClaimMemLimit("ctr0", "test1")
	require.Error(t, err, "same memory limit by another plugin, should conflict")

	// claim CPU shares
	err = o.ClaimCPUShares("ctr0", "test0")
	require.NoError(t, err, "CPU shares")

	// claim CPU shares of another container
	err = o.ClaimCPUShares("ctr1", "test0")
	require.NoError(t, err, "other CPU shares")

	// claim CPU shares of same container by another plugin
	err = o.ClaimCPUShares("ctr0", "test1")
	require.Error(t, err, "same CPU shares by another plugin, should conflict")

	// claim args
	err = o.ClaimArgs("ctr0", "test0")
	require.NoError(t, err, "args")

	// claim same args by another plugin
	err = o.ClaimArgs("ctr0", "test1")
	require.Error(t, err, "same args by another plugin, should conflict")

	// clear args
	o.ClearArgs("ctr0", "test1")

	// claim args after clearing
	err = o.ClaimArgs("ctr0", "test1")
	require.NoError(t, err, "try again same args, should not conflict")

	// clear args
	o.ClearArgs("ctr0", "test1")

	// claim args by another plugin
	err = o.ClaimArgs("ctr0", "test0")
	require.Error(t, err, "try again same args by non-clearing plugin, should conflict")
}

func TestCompoundClaims(t *testing.T) {
	o := api.NewOwningPlugins()

	// claim environment variable
	err := o.ClaimEnv("ctr0", "VAR0", "test0")
	require.NoError(t, err, "env VAR0")

	// claim another environment variable
	err = o.ClaimEnv("ctr0", "VAR1", "test0")
	require.NoError(t, err, "env VAR1")

	// claim environment variable of another container
	err = o.ClaimEnv("ctr1", "VAR0", "test1")
	require.NoError(t, err, "env VAR0 of another container")

	// claim already claimed environment variable by another plugin
	err = o.ClaimEnv("ctr0", "VAR1", "test1")
	require.Error(t, err, "env VAR1 of same container by another plugin, should conflict")

	// clear environment variable
	o.ClearEnv("ctr0", "VAR1", "test1")

	// claim same environment variable
	err = o.ClaimEnv("ctr0", "VAR1", "test1")
	require.NoError(t, err, "try again env VAR1, should not conflict")

	// clear again environment variable
	o.ClearEnv("ctr0", "VAR1", "test0")

	// claim same environment variable by another plugin
	err = o.ClaimEnv("ctr0", "VAR1", "test1")
	require.Error(t, err, "try again env VAR1 by non-clearing plugin, should conflict")

	require.Equal(t, api.Field_Annotations.String(), "Annotations", "annotation field name")
}

func TestAccumulatingOwnership(t *testing.T) {
	o := api.NewOwningPlugins()

	// claim OCI hooks of ctr0
	err := o.ClaimHooks("ctr0", "plugin0")
	require.NoError(t, err, "ctr0 OCI hooks by plugin0")

	// claim OCI hooks of ctr0
	err = o.ClaimHooks("ctr0", "plugin1")
	require.NoError(t, err, "ctr0 OCI hooks by plugin1")

	// claim OCI hooks of ctr0
	err = o.ClaimHooks("ctr0", "plugin2")
	require.NoError(t, err, "ctr0 OCI hooks by plugin2")

	owners, ok := o.HooksOwner("ctr0")
	require.True(t, ok, "ctr0 has hooks owners")
	require.Equal(t, "plugin0,plugin1,plugin2", owners, "ctr0 hooks owners")

}