From dfeb39febcfef9d0386c1cbee94ca118a477b37d Mon Sep 17 00:00:00 2001
From: Yifan Gu <yifan.gu@coreos.com>
Date: Fri, 22 May 2015 12:15:31 -0700
Subject: [PATCH] tests: Add rkt_cat_manifest_test to exercise 'rkt image
 cat-manifest'.

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

diff --git a/tests/rkt_cat_manifest_test.go b/tests/rkt_cat_manifest_test.go
new file mode 100644
index 0000000..3675cee
--- /dev/null
+++ b/tests/rkt_cat_manifest_test.go
@@ -0,0 +1,121 @@
+// Copyright 2015 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.
+
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"os/exec"
+	"strings"
+	"testing"
+
+	"github.com/coreos/rkt/Godeps/_workspace/src/github.com/ThomasRooney/gexpect"
+)
+
+const (
+
+	// The expected image manifest of the 'rkt-inspect-image-cat-manifest.aci'.
+	manifestTemplate = `{"acKind":"ImageManifest","acVersion":"0.5.1","name":"IMG_NAME","labels":[{"name":"version","value":"1.0.0"},{"name":"arch","value":"amd64"},{"name":"os","value":"linux"}],"app":{"exec":["/inspect"],"user":"0","group":"0","workingDirectory":"/","environment":[{"name":"VAR_FROM_MANIFEST","value":"manifest"}]}}`
+)
+
+// TestCatManifest tests 'rkt image cat-manifest', it will:
+// Read some existing image manifest via the image name, and verify the result.
+// Read some existing image manifest via the image hash, and verify the result.
+// Read some non-existing image manifest via the image name, and verify nothing is found.
+// Read some non-existing image manifest via the image hash, and verify nothing is found.
+func TestCatManifest(t *testing.T) {
+	var testImageHash string
+	testImage := "rkt-inspect-image-cat-manifest.aci"
+	testImageName := "coreos.com/rkt-cat-manifest-test"
+	expectManifest := strings.Replace(manifestTemplate, "IMG_NAME", testImageName, -1)
+
+	tmpManifest, err := ioutil.TempFile("", "rkt-TestCatManifest-")
+	if err != nil {
+		t.Fatalf("Cannot create temp manifest: %v", err)
+	}
+	if err := ioutil.WriteFile(tmpManifest.Name(), []byte(expectManifest), 0600); err != nil {
+		t.Fatalf("Cannot write to temp manifest: %v", err)
+	}
+	defer os.Remove(tmpManifest.Name())
+
+	patchTestACI(testImage, "--manifest", tmpManifest.Name())
+	defer os.Remove(testImage)
+	ctx := newRktRunCtx()
+	defer ctx.cleanup()
+
+	// Import the test image into store manually.
+	cmds := strings.Fields(ctx.cmd())
+	fetchCmd := exec.Command(cmds[0], cmds[1:]...)
+	fetchCmd.Args = append(fetchCmd.Args, "--insecure-skip-verify", "fetch", testImage)
+	output, err := fetchCmd.Output()
+	if err != nil {
+		t.Fatalf("Cannot read the output: %v", err)
+	}
+
+	// Read out the image hash.
+	ix := strings.Index(string(output), "sha512-")
+	if ix < 0 {
+		t.Fatalf("Unexpected result: %v, expecting a sha512 hash", testImageHash)
+	}
+	testImageHash = strings.TrimSpace(string(output)[ix:])
+
+	tests := []struct {
+		image      string
+		shouldFind bool
+		expect     string
+	}{
+		{
+			testImageName,
+			true,
+			expectManifest,
+		},
+		{
+			testImageHash,
+			true,
+			expectManifest,
+		},
+		{
+			"sha512-not-existed",
+			false,
+			"",
+		},
+		{
+			"some~random~aci~name",
+			false,
+			"",
+		},
+	}
+
+	for i, tt := range tests {
+		runCmd := fmt.Sprintf("%s image cat-manifest %s", ctx.cmd(), tt.image)
+		t.Logf("Running 'run' test #%v: %v", i, runCmd)
+		child, err := gexpect.Spawn(runCmd)
+		if err != nil {
+			t.Fatalf("Cannot exec rkt #%v: %v", i, err)
+		}
+
+		if tt.expect != "" {
+			if err := child.Expect(tt.expect); err != nil {
+				t.Fatalf("Expected %q but not found: %v", tt.expect, err)
+			}
+		}
+		if err := child.Wait(); err != nil {
+			if tt.shouldFind || err.Error() != "exit status 1" {
+				t.Fatalf("rkt didn't terminate correctly: %v", err)
+			}
+		}
+	}
+}