126 lines
2.4 KiB
Go
126 lines
2.4 KiB
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|: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|: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|:1234\n", APIVersion)
|
|
log.Println("HELLO")
|
|
log.Println("WORLD")
|
|
case "stdin":
|
|
fmt.Printf("%s|: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)
|
|
}
|
|
}
|