Codebase list golang-github-go-kit-kit / f9d8373b-9935-4125-b2b2-694ccabcf82b/upstream/0.12.0 sd / consul / registrar_test.go
f9d8373b-9935-4125-b2b2-694ccabcf82b/upstream/0.12.0

Tree @f9d8373b-9935-4125-b2b2-694ccabcf82b/upstream/0.12.0 (Download .tar.gz)

registrar_test.go @f9d8373b-9935-4125-b2b2-694ccabcf82b/upstream/0.12.0raw · history · blame

package consul

import (
	"testing"

	stdconsul "github.com/hashicorp/consul/api"

	"github.com/go-kit/log"
)

func TestRegistrar(t *testing.T) {
	client := newTestClient([]*stdconsul.ServiceEntry{})
	p := NewRegistrar(client, testRegistration, log.NewNopLogger())
	if want, have := 0, len(client.entries); want != have {
		t.Errorf("want %d, have %d", want, have)
	}

	p.Register()
	if want, have := 1, len(client.entries); want != have {
		t.Errorf("want %d, have %d", want, have)
	}

	p.Deregister()
	if want, have := 0, len(client.entries); want != have {
		t.Errorf("want %d, have %d", want, have)
	}
}