From 44eef72d755bcf58a2b25730f70d72b976292ae9 Mon Sep 17 00:00:00 2001
From: chchliang <chen.chuanliang@zte.com.cn>
Date: Tue, 24 Oct 2017 09:49:58 +0800
Subject: [PATCH] add testcase IsValidStateString

Signed-off-by: chchliang <chen.chuanliang@zte.com.cn>
Upstream-commit: 675ac37482432e13d1312647762d9db8b9bb175e
Component: engine
---
 components/engine/container/state_test.go | 24 +++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/components/engine/container/state_test.go b/components/engine/container/state_test.go
index 2a90e5541d..14f24600e0 100644
--- a/components/engine/container/state_test.go
+++ b/components/engine/container/state_test.go
@@ -166,3 +166,27 @@ func TestStateTimeoutWait(t *testing.T) {
 		}
 	}
 }
+
+func TestIsValidStateString(t *testing.T) {
+	states := []struct {
+		state    string
+		expected bool
+	}{
+		{"paused", true},
+		{"restarting", true},
+		{"running", true},
+		{"dead", true},
+		{"start", false},
+		{"created", true},
+		{"exited", true},
+		{"removing", true},
+		{"stop", false},
+	}
+
+	for _, s := range states {
+		v := IsValidStateString(s.state)
+		if v != s.expected {
+			t.Fatalf("Expected %t, but got %t", s.expected, v)
+		}
+	}
+}