[test]Add feature for CRI plugins (#1565)

helsaawy · web-flow · commit f83cc58000f0 · 2022-11-17T17:29:27.000-05:00
Signed-off-by: Hamza El-Saawy &lt;hamzaelsaawy@microsoft.com&gt;

Signed-off-by: Hamza El-Saawy &lt;hamzaelsaawy@microsoft.com&gt;
diff --git a/test/cri-containerd/containerdrestart_test.go b/test/cri-containerd/containerdrestart_test.go
@@ -74,6 +74,8 @@ func Test_ContainerdRestart_LCOW(t *testing.T) {
 
 // test restarting containers and pods
 func Test_Container_CRI_Restart(t *testing.T) {
+	requireFeatures(t, featureCRIPlugin)
+
 	client := newTestRuntimeClient(t)
 	pluginClient := newTestPluginClient(t)
 	ctx, cancel := context.WithCancel(context.Background())
@@ -220,6 +222,8 @@ func Test_Container_CRI_Restart_State(t *testing.T) {
 	testFile := "t.txt"
 	wcowTestFile := `C:\Users\ContainerUser\t.txt`
 
+	requireFeatures(t, featureCRIPlugin)
+
 	client := newTestRuntimeClient(t)
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
diff --git a/test/cri-containerd/main_test.go b/test/cri-containerd/main_test.go
@@ -106,6 +106,7 @@ const (
 	featureTerminateOnRestart = "TerminateOnRestart"
 	featureLCOWIntegrity      = "LCOWIntegrity"
 	featureLCOWCrypt          = "LCOWCrypt"
+	featureCRIPlugin          = "CRIPlugin"
 )
 
 var allFeatures = []string{
@@ -117,6 +118,9 @@ var allFeatures = []string{
 	featureGPU,
 	featureCRIUpdateContainer,
 	featureTerminateOnRestart,
+	featureLCOWIntegrity,
+	featureLCOWCrypt,
+	featureCRIPlugin,
 }
 
 func init() {
