From 73d75609f395dd155d6aa9301e622c79f4488c36 Mon Sep 17 00:00:00 2001
From: Fupan Li <lifupan@gmail.com>
Date: Fri, 21 Dec 2018 16:27:45 +0000
Subject: [PATCH] benchmark: add a testcase for creating a pod and a container

Add a testcase for measuring the time of creating a pod
and container.

Signed-off-by: Fupan Li <lifupan@gmail.com>
---
 pkg/benchmark/pod_container.go | 74 ++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 pkg/benchmark/pod_container.go

diff --git a/pkg/benchmark/pod_container.go b/pkg/benchmark/pod_container.go
new file mode 100644
index 0000000000..1837d0da9b
--- /dev/null
+++ b/pkg/benchmark/pod_container.go
@@ -0,0 +1,74 @@
+/*
+Copyright 2018 The Kubernetes 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 benchmark
+
+import (
+	"github.com/kubernetes-sigs/cri-tools/pkg/framework"
+	internalapi "k8s.io/kubernetes/pkg/kubelet/apis/cri"
+	runtimeapi "k8s.io/kubernetes/pkg/kubelet/apis/cri/runtime/v1alpha2"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = framework.KubeDescribe("PodSandbox", func() {
+	f := framework.NewDefaultCRIFramework()
+
+	var rc internalapi.RuntimeService
+	var ic internalapi.ImageManagerService
+	var podID string
+
+	BeforeEach(func() {
+		rc = f.CRIClient.CRIRuntimeClient
+		ic = f.CRIClient.CRIImageClient
+	})
+
+	AfterEach(func() {
+		By("stop PodSandbox")
+		rc.StopPodSandbox(podID)
+		By("delete PodSandbox")
+		rc.RemovePodSandbox(podID)
+	})
+
+	Context("benchmark about start a container from scratch", func() {
+		Measure("benchmark about start a container from scratch", func(b Benchmarker) {
+			var err error
+
+			podSandboxName := "PodSandbox-for-creating-pod-and-container-performance-test-" + framework.NewUUID()
+			uid := framework.DefaultUIDPrefix + framework.NewUUID()
+			namespace := framework.DefaultNamespacePrefix + framework.NewUUID()
+
+			config := &runtimeapi.PodSandboxConfig{
+				Metadata: framework.BuildPodSandboxMetadata(podSandboxName, uid, namespace, framework.DefaultAttempt),
+				Linux:    &runtimeapi.LinuxPodSandboxConfig{},
+			}
+
+			operation := b.Time("create PodSandbox and container", func() {
+				By("run PodSandbox")
+				podID, err = rc.RunPodSandbox(config, framework.TestContext.RuntimeHandler)
+				framework.ExpectNoError(err, "failed to create PodSandbox: %v", err)
+				By("create container in PodSandbox")
+				containerID := framework.CreateDefaultContainer(rc, ic, podID, config, "Pod-Container-for-creating-benchmark-")
+				By("start container in PodSandbox")
+				err = rc.StartContainer(containerID)
+			})
+
+			framework.ExpectNoError(err, "failed to start Container: %v", err)
+			Expect(operation.Seconds()).Should(BeNumerically("<", 5), "create PodSandbox shouldn't take too long.")
+		}, defaultOperationTimes)
+	})
+})