activation: optionally split test compilation and run phases

Author: lucab

activation/common_test.go

@@ -0,0 +1,31 @@
+// Copyright 2018 CoreOS, Inc.
+//
+// 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 activation
+
+import (
+	"fmt"
+	"os"
+)
+
+// exampleCmd returns the command line for the specified example binary.
+func exampleCmd(binaryName string) (string, []string) {
+	sourcePath := fmt.Sprintf("../examples/activation/%s.go", binaryName)
+	sourceCmdLine := []string{"go", "run", sourcePath}
+	binaryPath := fmt.Sprintf("../test_bins/%s.example", binaryName)
+	if _, err := os.Stat(binaryPath); err != nil && os.IsNotExist(err) {
+		return sourceCmdLine[0], sourceCmdLine
+	}
+	return binaryPath, []string{binaryPath}
+}

activation/files_test.go

@@ -38,7 +38,8 @@ func correctStringWritten(t *testing.T, r *os.File, expected string) bool {
 // TestActivation forks out a copy of activation.go example and reads back two
 // strings from the pipes that are passed in.
 func TestActivation(t *testing.T) {
-	cmd := exec.Command("go", "run", "../examples/activation/activation.go")
+	arg0, cmdline := exampleCmd("activation")
+	cmd := exec.Command(arg0, cmdline...)
 
 	r1, w1, _ := os.Pipe()
 	r2, w2, _ := os.Pipe()
@@ -60,7 +61,8 @@ func TestActivation(t *testing.T) {
 }
 
 func TestActivationNoFix(t *testing.T) {
-	cmd := exec.Command("go", "run", "../examples/activation/activation.go")
+	arg0, cmdline := exampleCmd("activation")
+	cmd := exec.Command(arg0, cmdline...)
 	cmd.Env = os.Environ()
 	cmd.Env = append(cmd.Env, "LISTEN_FDS=2")
 
@@ -71,7 +73,8 @@ func TestActivationNoFix(t *testing.T) {
 }
 
 func TestActivationNoFiles(t *testing.T) {
-	cmd := exec.Command("go", "run", "../examples/activation/activation.go")
+	arg0, cmdline := exampleCmd("activation")
+	cmd := exec.Command(arg0, cmdline...)
 	cmd.Env = os.Environ()
 	cmd.Env = append(cmd.Env, "LISTEN_FDS=0", "FIX_LISTEN_PID=1")
 

activation/listeners_test.go

@@ -38,7 +38,8 @@ func correctStringWrittenNet(t *testing.T, r net.Conn, expected string) bool {
 // TestActivation forks out a copy of activation.go example and reads back two
 // strings from the pipes that are passed in.
 func TestListeners(t *testing.T) {
-	cmd := exec.Command("go", "run", "../examples/activation/listen.go")
+	arg0, cmdline := exampleCmd("listen")
+	cmd := exec.Command(arg0, cmdline...)
 
 	l1, err := net.Listen("tcp", ":9999")
 	if err != nil {

activation/packetconns_test.go

@@ -24,7 +24,8 @@ import (
 // TestActivation forks out a copy of activation.go example and reads back two
 // strings from the pipes that are passed in.
 func TestPacketConns(t *testing.T) {
-	cmd := exec.Command("go", "run", "../examples/activation/udpconn.go")
+	arg0, cmdline := exampleCmd("udpconn")
+	cmd := exec.Command(arg0, cmdline...)
 
 	u1, err := net.ListenUDP("udp", &net.UDPAddr{Port: 9999})
 	if err != nil {