From 6797373344b3d2be5aecdeb32603acdca1c1d902 Mon Sep 17 00:00:00 2001
From: Luca Bruno <luca.bruno@coreos.com>
Date: Mon, 14 Nov 2016 14:08:11 +0000
Subject: [PATCH] tests: add smoke test for app sandbox

---
 tests/rkt_app_sandbox_test.go | 116 ++++++++++++++++++++++++++++++++++
 1 file changed, 116 insertions(+)
 create mode 100644 tests/rkt_app_sandbox_test.go

diff --git a/tests/rkt_app_sandbox_test.go b/tests/rkt_app_sandbox_test.go
new file mode 100644
index 0000000000..5f304525d9
--- /dev/null
+++ b/tests/rkt_app_sandbox_test.go
@@ -0,0 +1,116 @@
+// Copyright 2016 The rkt 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.
+
+// Disabled on kvm due to https://github.com/coreos/rkt/issues/3382
+// +build !fly,!kvm
+
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/coreos/rkt/tests/testutils"
+)
+
+// TestAppSandboxSmoke is a basic smoke test for `rkt app` sandbox
+// and related commands.
+func TestAppSandboxSmoke(t *testing.T) {
+	actionTimeout := 30 * time.Second
+	imageName := "coreos.com/rkt-inspect/hello"
+	appName := "hello-app"
+	msg := "HelloFromAppInSandbox"
+
+	aciHello := patchTestACI("rkt-inspect-hello.aci", fmt.Sprintf("--name=%s", imageName), fmt.Sprintf("--exec=/inspect --print-msg=%s", msg))
+	defer os.Remove(aciHello)
+
+	ctx := testutils.NewRktRunCtx()
+	defer ctx.Cleanup()
+
+	cmd := strings.Fields(fmt.Sprintf("%s fetch --insecure-options=image %s", ctx.Cmd(), aciHello))
+	fetchCmd := exec.Command(cmd[0], cmd[1:]...)
+	fetchCmd.Env = append(fetchCmd.Env, "RKT_EXPERIMENT_APP=true")
+	fetchOutput, err := fetchCmd.CombinedOutput()
+	if err != nil {
+		t.Fatalf("Unexpected error: %v\n%s", err, fetchOutput)
+	}
+
+	tmpDir := createTempDirOrPanic("rkt-test-cri-")
+	defer os.RemoveAll(tmpDir)
+
+	rktCmd := fmt.Sprintf("%s app sandbox --debug --uuid-file-save=%s/uuid", ctx.Cmd(), tmpDir)
+	err = os.Setenv("RKT_EXPERIMENT_APP", "true")
+	if err != nil {
+		panic(err)
+	}
+	child := spawnOrFail(t, rktCmd)
+	err = os.Unsetenv("RKT_EXPERIMENT_APP")
+	if err != nil {
+		panic(err)
+	}
+
+	expected := "Reached target rkt apps target."
+	if err := expectTimeoutWithOutput(child, expected, actionTimeout); err != nil {
+		t.Fatalf("Expected %q but not found: %v", expected, err)
+	}
+
+	podUUID, err := ioutil.ReadFile(filepath.Join(tmpDir, "uuid"))
+	if err != nil {
+		t.Fatalf("Can't read pod UUID: %v", err)
+	}
+
+	cmd = strings.Fields(fmt.Sprintf("%s app add --debug %s %s --name=%s", ctx.Cmd(), podUUID, imageName, appName))
+	addCmd := exec.Command(cmd[0], cmd[1:]...)
+	addCmd.Env = append(addCmd.Env, "RKT_EXPERIMENT_APP=true")
+	output, err := addCmd.CombinedOutput()
+	if err != nil {
+		t.Fatalf("Unexpected error: %v\n%s", err, output)
+	}
+
+	cmd = strings.Fields(fmt.Sprintf("%s app start --debug %s --app=%s", ctx.Cmd(), podUUID, appName))
+	startCmd := exec.Command(cmd[0], cmd[1:]...)
+	startCmd.Env = append(startCmd.Env, "RKT_EXPERIMENT_APP=true")
+	output, err = startCmd.CombinedOutput()
+	if err != nil {
+		t.Fatalf("Unexpected error: %v\n%s", err, output)
+	}
+
+	if err := expectTimeoutWithOutput(child, msg, actionTimeout); err != nil {
+		t.Fatalf("Expected %q but not found: %v", expected, err)
+	}
+
+	cmd = strings.Fields(fmt.Sprintf("%s app rm --debug %s --app=%s", ctx.Cmd(), podUUID, appName))
+	removeCmd := exec.Command(cmd[0], cmd[1:]...)
+	removeCmd.Env = append(removeCmd.Env, "RKT_EXPERIMENT_APP=true")
+	output, err = removeCmd.CombinedOutput()
+	if err != nil {
+		t.Fatalf("Unexpected error: %v\n%s", err, output)
+	}
+
+	cmd = strings.Fields(fmt.Sprintf("%s stop %s", ctx.Cmd(), podUUID))
+	output, err = exec.Command(cmd[0], cmd[1:]...).CombinedOutput()
+	if err != nil {
+		t.Fatalf("Unexpected error: %v\n%s", err, output)
+	}
+
+	if err := child.Wait(); err != nil {
+		t.Fatalf("rkt didn't terminate correctly: %v", err)
+	}
+}