1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
|
// +build integration
package consul
import (
"io"
"os"
"testing"
"time"
"github.com/go-kit/kit/endpoint"
"github.com/go-kit/kit/log"
"github.com/go-kit/kit/sd"
stdconsul "github.com/hashicorp/consul/api"
)
func TestIntegration(t *testing.T) {
consulAddr := os.Getenv("CONSUL_ADDR")
if consulAddr == "" {
t.Fatal("CONSUL_ADDR is not set")
}
stdClient, err := stdconsul.NewClient(&stdconsul.Config{
Address: consulAddr,
})
if err != nil {
t.Fatal(err)
}
client := NewClient(stdClient)
logger := log.NewLogfmtLogger(os.Stderr)
// Produce a fake service registration.
r := &stdconsul.AgentServiceRegistration{
ID: "my-service-ID",
Name: "my-service-name",
Tags: []string{"alpha", "beta"},
Port: 12345,
Address: "my-address",
EnableTagOverride: false,
// skipping check(s)
}
// Build an Instancer on r.Name + r.Tags.
factory := func(instance string) (endpoint.Endpoint, io.Closer, error) {
t.Logf("factory invoked for %q", instance)
return endpoint.Nop, nil, nil
}
instancer := NewInstancer(
client,
log.With(logger, "component", "instancer"),
r.Name,
r.Tags,
true,
)
endpointer := sd.NewEndpointer(
instancer,
factory,
log.With(logger, "component", "endpointer"),
)
time.Sleep(time.Second)
// Before we publish, we should have no endpoints.
endpoints, err := endpointer.Endpoints()
if err != nil {
t.Error(err)
}
if want, have := 0, len(endpoints); want != have {
t.Errorf("want %d, have %d", want, have)
}
// Build a registrar for r.
registrar := NewRegistrar(client, r, log.With(logger, "component", "registrar"))
registrar.Register()
defer registrar.Deregister()
time.Sleep(time.Second)
// Now we should have one active endpoints.
endpoints, err = endpointer.Endpoints()
if err != nil {
t.Error(err)
}
if want, have := 1, len(endpoints); want != have {
t.Errorf("want %d, have %d", want, have)
}
}
|