up test demo

liangchenye · liangchenye · commit b55cfcd789fa · 2016-12-22T00:26:39.000+08:00
Signed-off-by: liangchenye &lt;liangchenye@huawei.com&gt;
diff --git a/cmd/oci-lifecycle-demo/main.go b/cmd/oci-lifecycle-demo/main.go
@@ -3,36 +3,83 @@ package main
 import (
 	"fmt"
 
-	"github.com/opencontainers/runtime-tools/lifecycle"
+	l "github.com/opencontainers/runtime-tools/lifecycle"
 )
 
+func TestDemo() {
+	runtime := "runc"
+
+	type ActionCase struct {
+		action   l.LifecycleAction
+		args     []string
+		expected bool
+	}
+	cases := []struct {
+		bundle  string
+		id      string
+		actions []ActionCase
+	}{
+		{"noexist_bundle", "1", []ActionCase{
+			{action: l.LifecycleCreate, expected: false}},
+		},
+		{"exist_bundle", "1", []ActionCase{
+			{action: l.LifecycleCreate, expected: true}},
+		},
+		{"exist_bundle", "1", []ActionCase{
+			{action: l.LifecycleCreate, expected: true},
+			{action: l.LifecycleCreate, expected: false}},
+		},
+		{"exist_bundle_run_err", "1", []ActionCase{
+			{action: l.LifecycleCreate, expected: true},
+			{action: l.LifecycleStart, expected: false}},
+		},
+	}
+	for _, c := range cases {
+		//TODO: Prepare c.bundle
+		lc, _ := l.NewLifecycle(runtime, c.bundle, c.id)
+		for _, a := range c.actions {
+			_, err := lc.Operate(a.action)
+			if err != nil && a.expected == true {
+				fmt.Printf("Failed in test")
+				break
+			} else if err == nil && a.expected == false {
+				fmt.Printf("Failed in test")
+				break
+			}
+		}
+
+		//TODO: Remove it anyway
+		lc.Operate(l.LifecycleDelete)
+	}
+}
+
 func main() {
 	runtime := "runc"
 	bundle := "bundle"
 	id := "25"
-	lc, _ := lifecycle.NewLifecycle(runtime, bundle, id)
+	lc, _ := l.NewLifecycle(runtime, bundle, id)
 
 	fmt.Println("create")
-	out, err := lc.Operate(lifecycle.LifecycleCreate)
+	out, err := lc.Operate(l.LifecycleCreate)
 	fmt.Println(string(out), err)
 
 	fmt.Println("state")
-	out, err = lc.Operate(lifecycle.LifecycleState)
+	out, err = lc.Operate(l.LifecycleState)
 	fmt.Println(string(out), err)
 
 	fmt.Println("start")
-	out, err = lc.Operate(lifecycle.LifecycleStart)
+	out, err = lc.Operate(l.LifecycleStart)
 	fmt.Println(string(out), err)
 
 	fmt.Println("state")
-	out, _ = lc.Operate(lifecycle.LifecycleState)
+	out, _ = lc.Operate(l.LifecycleState)
 	fmt.Println(string(out))
 
 	fmt.Println("delete")
-	out, _ = lc.Operate(lifecycle.LifecycleDelete)
+	out, _ = lc.Operate(l.LifecycleDelete)
 	fmt.Println(string(out))
 
 	fmt.Println("state")
-	out, _ = lc.Operate(lifecycle.LifecycleState)
+	out, _ = lc.Operate(l.LifecycleState)
 	fmt.Println(string(out))
 }
diff --git a/lifecycle/lifecycle.go b/lifecycle/lifecycle.go
@@ -1,7 +1,6 @@
 package lifecycle
 
 import (
-	"bytes"
 	"errors"
 	"fmt"
 	"os/exec"
