athens/e2etests/run_athens.go

// +build e2etests

package e2etests

import (
	"context"
	"fmt"
	"net/http"
	"os/exec"
	"path"
	"path/filepath"
	"time"
)

func buildAthens(goBin, destPath string, env []string) (string, error) {
	target := path.Join(destPath, "athens-proxy")
	binFolder, err := filepath.Abs("../cmd/proxy")
	if err != nil {
		return "", fmt.Errorf("Failed to get athens source path %v", err)
	}

	cmd := exec.Command(goBin, "build", "-o", target, binFolder)
	cmd.Env = env
	output, err := cmd.CombinedOutput()
	if err != nil {
		return "", fmt.Errorf("Failed to build athens: %v - %s", err, string(output))
	}
	return target, nil
}

func stopAthens() error {
	cmd := exec.Command("pkill", "athens-proxy")
	output, err := cmd.CombinedOutput()
	if err != nil {
		return fmt.Errorf("Failed to stop athens: %v - %s", err, string(output))
	}
	return err
}

func runAthensAndWait(ctx context.Context, athensBin string, env []string) error {
	cmd := exec.CommandContext(ctx, athensBin)
	cmd.Env = env

	cmd.Start()

	ticker := time.NewTicker(time.Second)
	timeout := time.After(20 * time.Second)
	defer ticker.Stop()

	for {
		select {
		case <-ticker.C:
			resp, _ := http.Get("http://localhost:3000/readyz")
			if resp.StatusCode == 200 {
				return nil
			}
		case <-timeout:
			return fmt.Errorf("Failed to run athens")
		}
	}
}