From 33ae9cb2bb9134942609d178fe5421507d817c23 Mon Sep 17 00:00:00 2001
From: Zbigniew Kostrzewa <zkostrzewa@gmail.com>
Date: Mon, 10 Jul 2017 08:19:29 +0200
Subject: [PATCH] Add test for playbook_files using docker builder.

---
 provisioner/ansible-local/provisioner_test.go | 175 ++++++++++++++----
 .../ansible-local/test-fixtures/hello.yml     |   5 +
 .../ansible-local/test-fixtures/world.yml     |   5 +
 3 files changed, 149 insertions(+), 36 deletions(-)
 create mode 100644 provisioner/ansible-local/test-fixtures/hello.yml
 create mode 100644 provisioner/ansible-local/test-fixtures/world.yml

diff --git a/provisioner/ansible-local/provisioner_test.go b/provisioner/ansible-local/provisioner_test.go
index 5fdcdfe8e..bd6568afc 100644
--- a/provisioner/ansible-local/provisioner_test.go
+++ b/provisioner/ansible-local/provisioner_test.go
@@ -8,45 +8,13 @@ import (
 	"testing"
 
 	"fmt"
+	"github.com/hashicorp/packer/builder/docker"
 	"github.com/hashicorp/packer/packer"
+	"github.com/hashicorp/packer/provisioner/file"
+	"github.com/hashicorp/packer/template"
+	"os/exec"
 )
 
-func testConfig() map[string]interface{} {
-	m := make(map[string]interface{})
-	return m
-}
-
-func createTempFile() string {
-	file, err := ioutil.TempFile("", "")
-	if err != nil {
-		panic(fmt.Sprintf("err: %s", err))
-	}
-	return file.Name()
-}
-
-func createTempFiles(numFiles int) []string {
-	files := make([]string, 0, numFiles)
-	defer func() {
-		// Cleanup the files if not all were created.
-		if len(files) < numFiles {
-			for _, file := range files {
-				os.Remove(file)
-			}
-		}
-	}()
-
-	for i := 0; i < numFiles; i++ {
-		files = append(files, createTempFile())
-	}
-	return files
-}
-
-func removeFiles(files ...string) {
-	for _, file := range files {
-		os.Remove(file)
-	}
-}
-
 func TestProvisioner_Impl(t *testing.T) {
 	var raw interface{}
 	raw = &Provisioner{}
@@ -318,3 +286,138 @@ func TestProvisionerPrepare_Dirs(t *testing.T) {
 		t.Fatalf("err: %s", err)
 	}
 }
+
+func TestProvisionerProvisionDocker_PlaybookFiles(t *testing.T) {
+	if os.Getenv("PACKER_ACC") == "" {
+		t.Skip("This test is only run with PACKER_ACC=1")
+	}
+
+	ui := packer.TestUi(t)
+	cache := &packer.FileCache{CacheDir: os.TempDir()}
+
+	tpl, err := template.Parse(strings.NewReader(playbookFilesDockerConfig))
+	if err != nil {
+		t.Fatalf("Unable to parse config: %s", err)
+	}
+
+	// Check if docker executable can be found.
+	_, err = exec.LookPath("docker")
+	if err != nil {
+		t.Error("docker command not found; please make sure docker is installed")
+	}
+
+	// Setup the builder
+	builder := &docker.Builder{}
+	warnings, err := builder.Prepare(tpl.Builders["docker"].Config)
+	if err != nil {
+		t.Fatalf("Error preparing configuration %s", err)
+	}
+	if len(warnings) > 0 {
+		t.Fatal("Encountered configuration warnings; aborting")
+	}
+
+	ansible := &Provisioner{}
+	err = ansible.Prepare(tpl.Provisioners[0].Config)
+	if err != nil {
+		t.Fatalf("Error preparing ansible-local provisioner: %s", err)
+	}
+
+	download := &file.Provisioner{}
+	err = download.Prepare(tpl.Provisioners[1].Config)
+	if err != nil {
+		t.Fatalf("Error preparing download: %s", err)
+	}
+	defer os.Remove("hello_world")
+
+	// Add hooks so the provisioners run during the build
+	hooks := map[string][]packer.Hook{}
+	hooks[packer.HookProvision] = []packer.Hook{
+		&packer.ProvisionHook{
+			Provisioners: []packer.Provisioner{
+				ansible,
+				download,
+			},
+			ProvisionerTypes: []string{tpl.Provisioners[0].Type, tpl.Provisioners[1].Type},
+		},
+	}
+	hook := &packer.DispatchHook{Mapping: hooks}
+
+	artifact, err := builder.Run(ui, hook, cache)
+	if err != nil {
+		t.Fatalf("Error running build %s", err)
+	}
+	defer artifact.Destroy()
+
+	actualContent, err := ioutil.ReadFile("hello_world")
+	if err != nil {
+		t.Fatalf("Expected file not found: %s", err)
+	}
+
+	expectedContent := "Hello world!"
+	if string(actualContent) != expectedContent {
+		t.Fatalf(`Unexpected file content: expected="%s", actual="%s"`, expectedContent, actualContent)
+	}
+}
+
+func testConfig() map[string]interface{} {
+	m := make(map[string]interface{})
+	return m
+}
+
+func createTempFile() string {
+	file, err := ioutil.TempFile("", "")
+	if err != nil {
+		panic(fmt.Sprintf("err: %s", err))
+	}
+	return file.Name()
+}
+
+func createTempFiles(numFiles int) []string {
+	files := make([]string, 0, numFiles)
+	defer func() {
+		// Cleanup the files if not all were created.
+		if len(files) < numFiles {
+			for _, file := range files {
+				os.Remove(file)
+			}
+		}
+	}()
+
+	for i := 0; i < numFiles; i++ {
+		files = append(files, createTempFile())
+	}
+	return files
+}
+
+func removeFiles(files ...string) {
+	for _, file := range files {
+		os.Remove(file)
+	}
+}
+
+const playbookFilesDockerConfig = `
+{
+	"builders": [
+		{
+			"type": "docker",
+			"image": "williamyeh/ansible:centos7",
+			"discard": true
+		}
+	],
+	"provisioners": [
+		{
+			"type": "ansible-local",
+			"playbook_files": [
+				"test-fixtures/hello.yml",
+				"test-fixtures/world.yml"
+			]
+		},
+		{
+			"type": "file",
+			"source": "/tmp/hello_world",
+			"destination": "hello_world",
+			"direction": "download"
+		}
+	]
+}
+`
diff --git a/provisioner/ansible-local/test-fixtures/hello.yml b/provisioner/ansible-local/test-fixtures/hello.yml
new file mode 100644
index 000000000..6bb8797d8
--- /dev/null
+++ b/provisioner/ansible-local/test-fixtures/hello.yml
@@ -0,0 +1,5 @@
+---
+- hosts: all
+  tasks:
+    - name: write Hello
+      shell: echo -n "Hello" >> /tmp/hello_world
\ No newline at end of file
diff --git a/provisioner/ansible-local/test-fixtures/world.yml b/provisioner/ansible-local/test-fixtures/world.yml
new file mode 100644
index 000000000..98a205c7b
--- /dev/null
+++ b/provisioner/ansible-local/test-fixtures/world.yml
@@ -0,0 +1,5 @@
+---
+- hosts: all
+  tasks:
+    - name: write world!
+      shell: echo -n " world!" >> /tmp/hello_world
\ No newline at end of file