packer-cn/packer/plugin/plugin_test.go

126 lines
2.5 KiB
Go
Raw Normal View History

2013-05-05 00:26:30 -04:00
package plugin
import (
"fmt"
2013-08-30 20:03:55 -04:00
"github.com/mitchellh/packer/packer"
"log"
2013-05-05 00:26:30 -04:00
"os"
"os/exec"
"testing"
"time"
2013-05-05 00:26:30 -04:00
)
2013-05-10 20:01:24 -04:00
func helperProcess(s ...string) *exec.Cmd {
2013-05-05 00:26:30 -04:00
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",
}
2013-05-05 00:26:30 -04:00
cmd := exec.Command(os.Args[0], cs...)
cmd.Env = append(env, os.Environ()...)
2013-05-05 00:26:30 -04:00
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 {
2013-09-22 12:51:14 -04:00
case "bad-version":
fmt.Printf("%s1|tcp|:1234\n", APIVersion)
2013-09-22 12:51:14 -04:00
<-make(chan int)
case "builder":
2013-12-10 16:47:18 -05:00
server, err := Server()
if err != nil {
log.Printf("[ERR] %s", err)
os.Exit(1)
}
server.RegisterBuilder(new(packer.MockBuilder))
server.Serve()
case "command":
2013-12-10 16:47:18 -05:00
server, err := Server()
if err != nil {
log.Printf("[ERR] %s", err)
os.Exit(1)
}
server.RegisterCommand(new(helperCommand))
server.Serve()
2013-05-11 13:46:17 -04:00
case "hook":
2013-12-10 16:47:18 -05:00
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)
2013-06-18 16:49:07 -04:00
case "post-processor":
2013-12-10 16:47:18 -05:00
server, err := Server()
if err != nil {
log.Printf("[ERR] %s", err)
os.Exit(1)
}
server.RegisterPostProcessor(new(helperPostProcessor))
server.Serve()
2013-05-24 00:37:16 -04:00
case "provisioner":
2013-12-10 16:47:18 -05:00
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)
}
2013-05-05 00:26:30 -04:00
}