Function TestConsulSource

cert/source_test.go:223–301

(t *testing.T)

}

func TestConsulSource(t *testing.T) {
	const certURL = "http://127.0.0.1:8500/v1/kv/fabio/test/consul-server"

	// run a consul server if it isn't already running
	_, err := http.Get("http://127.0.0.1:8500/v1/status/leader")
	if err != nil {
		consul := os.Getenv("CONSUL_EXE")
		if consul == "" {
			consul = "consul"
		}

		version, err := exec.Command(consul, "--version").Output()
		if err != nil {
			t.Fatalf("Failed to run %s --version", consul)
		}
		cr := bytes.IndexRune(version, '\n')
		t.Logf("Starting %s: %s", consul, string(version[:cr]))

		start := time.Now()
		cmd := exec.Command(consul, "agent", "-bind", "127.0.0.1", "-server", "-dev")
		if err := cmd.Start(); err != nil {
			t.Fatalf("Failed to start consul server. %s", err)
		}
		defer cmd.Process.Kill()

		isUp := func() bool {
			resp, err := http.Get("http://127.0.0.1:8500/v1/status/leader")
			// /v1/status/leader returns '\n""' while consul is in leader election mode
			// and '"127.0.0.1:8300"' when not. So we punt by checking the
			// body length instead of the actual body content :)
			if err != nil {
				return false
			}
			defer resp.Body.Close()

			if resp.StatusCode != 200 {
				return false
			}

			n, err := io.Copy(io.Discard, resp.Body)
			return err == nil && n > 10
		}

		// We need give consul ~8-10 seconds to become ready until I've
		// figured out whether we can speed this up. Make sure that this is
		// less than the global test timeout in Makefile.
		if !waitFor(12*time.Second, isUp) {
			t.Fatalf("Timeout waiting for consul server after %2.1f seconds", time.Since(start).Seconds())
		}
		t.Logf("Consul is ready after %2.1f seconds", time.Since(start).Seconds())
	} else {
		t.Log("Using existing consul server")
	}

	config, key, err := parseConsulURL(certURL)
	if err != nil {
		t.Fatalf("Failed to parse consul url: %s", err)
	}