packer-cn/packer/plugin/plugin_test.go

package plugin

import (
	"fmt"
	"github.com/mitchellh/packer/packer"
	"log"
	"os"
	"os/exec"
	"testing"
	"time"
)

func helperProcess(s ...string) *exec.Cmd {
	cs := []string{"-test.run=TestHelperProcess", "--"}
	cs = append(cs, s...)
	env := []string{
		"GO_WANT_HELPER_PROCESS=1",
		"PACKER_PLUGIN_MIN_PORT=10000",
		"PACKER_PLUGIN_MAX_PORT=25000",
	}

	cmd := exec.Command(os.Args[0], cs...)
	cmd.Env = append(env, os.Environ()...)
	return cmd
}

// This is not a real test. This is just a helper process kicked off by
// tests.
func TestHelperProcess(*testing.T) {
	if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" {
		return
	}

	defer os.Exit(0)

	args := os.Args
	for len(args) > 0 {
		if args[0] == "--" {
			args = args[1:]
			break
		}

		args = args[1:]
	}

	if len(args) == 0 {
		fmt.Fprintf(os.Stderr, "No command\n")
		os.Exit(2)
	}

	cmd, args := args[0], args[1:]
	switch cmd {
	case "bad-version":
		fmt.Printf("%s1|tcp|:1234\n", APIVersion)
		<-make(chan int)
	case "builder":
		server, err := Server()
		if err != nil {
			log.Printf("[ERR] %s", err)
			os.Exit(1)
		}
		server.RegisterBuilder(new(packer.MockBuilder))
		server.Serve()
	case "command":
		server, err := Server()
		if err != nil {
			log.Printf("[ERR] %s", err)
			os.Exit(1)
		}
		server.RegisterCommand(new(helperCommand))
		server.Serve()
	case "hook":
		server, err := Server()
		if err != nil {
			log.Printf("[ERR] %s", err)
			os.Exit(1)
		}
		server.RegisterHook(new(packer.MockHook))
		server.Serve()
	case "invalid-rpc-address":
		fmt.Println("lolinvalid")
	case "mock":
		fmt.Printf("%s|tcp|:1234\n", APIVersion)
		<-make(chan int)
	case "post-processor":
		server, err := Server()
		if err != nil {
			log.Printf("[ERR] %s", err)
			os.Exit(1)
		}
		server.RegisterPostProcessor(new(helperPostProcessor))
		server.Serve()
	case "provisioner":
		server, err := Server()
		if err != nil {
			log.Printf("[ERR] %s", err)
			os.Exit(1)
		}
		server.RegisterProvisioner(new(packer.MockProvisioner))
		server.Serve()
	case "start-timeout":
		time.Sleep(1 * time.Minute)
		os.Exit(1)
	case "stderr":
		fmt.Printf("%s|tcp|:1234\n", APIVersion)
		log.Println("HELLO")
		log.Println("WORLD")
	case "stdin":
		fmt.Printf("%s|tcp|:1234\n", APIVersion)
		data := make([]byte, 5)
		if _, err := os.Stdin.Read(data); err != nil {
			log.Printf("stdin read error: %s", err)
			os.Exit(100)
		}

		if string(data) == "hello" {
			os.Exit(0)
		}

		os.Exit(1)
	default:
		fmt.Fprintf(os.Stderr, "Unknown command: %q\n", cmd)
		os.Exit(2)
	}
}