From 494783951e5a4c7d37607533f66f9ef2344c90ca Mon Sep 17 00:00:00 2001
From: Zhou Hao <zhouhao@cn.fujitsu.com>
Date: Thu, 8 Feb 2018 17:50:56 +0800
Subject: [PATCH] validation: implement PosixHooksCalledInOrder test

Signed-off-by: Zhou Hao <zhouhao@cn.fujitsu.com>
---
 validation/hooks.go | 114 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 114 insertions(+)
 create mode 100644 validation/hooks.go

diff --git a/validation/hooks.go b/validation/hooks.go
new file mode 100644
index 00000000..774c7af9
--- /dev/null
+++ b/validation/hooks.go
@@ -0,0 +1,114 @@
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os/exec"
+	"path/filepath"
+	"time"
+
+	tap "github.com/mndrix/tap-go"
+	rspec "github.com/opencontainers/runtime-spec/specs-go"
+	"github.com/opencontainers/runtime-tools/specerror"
+	"github.com/opencontainers/runtime-tools/validation/util"
+	uuid "github.com/satori/go.uuid"
+)
+
+func main() {
+	t := tap.New()
+	t.Header(0)
+
+	var output string
+	config := util.LifecycleConfig{
+		Actions: util.LifecycleActionCreate | util.LifecycleActionStart | util.LifecycleActionDelete,
+		PreCreate: func(r *util.Runtime) error {
+			r.SetID(uuid.NewV4().String())
+			g := util.GetDefaultGenerator()
+			output = filepath.Join(r.BundleDir, g.Spec().Root.Path, "output")
+			shPath := filepath.Join(r.BundleDir, g.Spec().Root.Path, "/bin/sh")
+			err := g.AddPreStartHook(rspec.Hook{
+				Path: shPath,
+				Args: []string{
+					"sh", "-c", fmt.Sprintf("echo 'pre-start1 called' >> %s", output),
+				},
+			})
+			if err != nil {
+				return err
+			}
+			err = g.AddPreStartHook(rspec.Hook{
+				Path: shPath,
+				Args: []string{
+					"sh", "-c", fmt.Sprintf("echo 'pre-start2 called' >> %s", output),
+				},
+			})
+			if err != nil {
+				return err
+			}
+			err = g.AddPostStartHook(rspec.Hook{
+				Path: shPath,
+				Args: []string{
+					"sh", "-c", fmt.Sprintf("echo 'post-start1 called' >> %s", output),
+				},
+			})
+			if err != nil {
+				return err
+			}
+			err = g.AddPostStartHook(rspec.Hook{
+				Path: shPath,
+				Args: []string{
+					"sh", "-c", fmt.Sprintf("echo 'post-start2 called' >> %s", output),
+				},
+			})
+			if err != nil {
+				return err
+			}
+			err = g.AddPostStopHook(rspec.Hook{
+				Path: shPath,
+				Args: []string{
+					"sh", "-c", fmt.Sprintf("echo 'post-stop1 called' >> %s", output),
+				},
+			})
+			if err != nil {
+				return err
+			}
+			err = g.AddPostStopHook(rspec.Hook{
+				Path: shPath,
+				Args: []string{
+					"sh", "-c", fmt.Sprintf("echo 'post-stop2 called' >> %s", output),
+				},
+			})
+			if err != nil {
+				return err
+			}
+			g.SetProcessArgs([]string{"true"})
+			r.SetConfig(g)
+			return nil
+		},
+		PreDelete: func(r *util.Runtime) error {
+			util.WaitingForStatus(*r, util.LifecycleStatusStopped, time.Second*10, time.Second)
+			return nil
+		},
+	}
+
+	err := util.RuntimeLifecycleValidate(config)
+	outputData, _ := ioutil.ReadFile(output)
+	if err == nil && string(outputData) != "pre-start1 called\npre-start2 called\npost-start1\npost-start2\npost-stop1\npost-stop2\n" {
+		err := specerror.NewError(specerror.PosixHooksCalledInOrder, fmt.Errorf("Hooks MUST be called in the listed order"), rspec.Version)
+		diagnostic := map[string]string{
+			"error": err.Error(),
+		}
+		t.YAML(diagnostic)
+	} else {
+		diagnostic := map[string]string{
+			"error": err.Error(),
+		}
+		if e, ok := err.(*exec.ExitError); ok {
+			if len(e.Stderr) > 0 {
+				diagnostic["stderr"] = string(e.Stderr)
+			}
+		}
+		t.YAML(diagnostic)
+	}
+
+	t.AutoPlan()
+}