Separate common used test functions and structs to test/utils

Put common used test functions and structs to a separated package.
So we can use them for more testsuites.

Signed-off-by: Yiqiao Pu <ypu@redhat.com>

1 files changed, 74 insertions, 0 deletions

diff --git a/test/utils/podmantest_test.go b/test/utils/podmantest_test.go
new file mode 100644
index 000000000..87f756920
--- /dev/null
+++ b/test/utils/podmantest_test.go
@@ -0,0 +1,74 @@
+package utils_test
+
+import (
+	"os"
+
+	. "github.com/containers/libpod/test/utils"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("PodmanTest test", func() {
+	var podmanTest *FakePodmanTest
+
+	BeforeEach(func() {
+		podmanTest = FakePodmanTestCreate()
+	})
+
+	AfterEach(func() {
+		FakeOutputs = make(map[string][]string)
+	})
+
+	It("Test PodmanAsUser", func() {
+		FakeOutputs["check"] = []string{"check"}
+		os.Setenv("HOOK_OPTION", "hook_option")
+		env := os.Environ()
+		session := podmanTest.PodmanAsUser([]string{"check"}, 1000, 1000, env)
+		os.Unsetenv("HOOK_OPTION")
+		session.WaitWithDefaultTimeout()
+		Expect(session.Command.Process).ShouldNot(BeNil())
+	})
+
+	It("Test NumberOfContainersRunning", func() {
+		FakeOutputs["ps -q"] = []string{"one", "two"}
+		Expect(podmanTest.NumberOfContainersRunning()).To(Equal(2))
+	})
+
+	It("Test NumberOfContainers", func() {
+		FakeOutputs["ps -aq"] = []string{"one", "two"}
+		Expect(podmanTest.NumberOfContainers()).To(Equal(2))
+	})
+
+	It("Test NumberOfPods", func() {
+		FakeOutputs["pod ps -q"] = []string{"one", "two"}
+		Expect(podmanTest.NumberOfPods()).To(Equal(2))
+	})
+
+	It("Test WaitForContainer", func() {
+		FakeOutputs["ps -q"] = []string{"one", "two"}
+		Expect(WaitForContainer(podmanTest)).To(BeTrue())
+
+		FakeOutputs["ps -q"] = []string{"one"}
+		Expect(WaitForContainer(podmanTest)).To(BeTrue())
+
+		FakeOutputs["ps -q"] = []string{""}
+		Expect(WaitForContainer(podmanTest)).To(Not(BeTrue()))
+	})
+
+	It("Test GetContainerStatus", func() {
+		FakeOutputs["ps --all --format={{.Status}}"] = []string{"Need func update"}
+		Expect(podmanTest.GetContainerStatus()).To(Equal("Need func update"))
+	})
+
+	It("Test WaitContainerReady", func() {
+		FakeOutputs["logs testimage"] = []string{""}
+		Expect(WaitContainerReady(podmanTest, "testimage", "ready", 2, 1)).To(Not(BeTrue()))
+
+		FakeOutputs["logs testimage"] = []string{"I am ready"}
+		Expect(WaitContainerReady(podmanTest, "testimage", "am ready", 2, 1)).To(BeTrue())
+
+		FakeOutputs["logs testimage"] = []string{"I am ready"}
+		Expect(WaitContainerReady(podmanTest, "testimage", "", 2, 1)).To(BeTrue())
+	})
+
+})