Exemplo n.º 1
0
func TestRegister(t *testing.T) {
	var (
		namespace = "namespace"
		machines  = []string{"foo", "bar"}
		ttl       = uint64(1234)
		instance  = discovery.Instance{
			ID:     "id",
			Name:   "name",
			Env:    "env",
			Region: "region",
			Host:   "host",
			Port:   987,
		}
		path   = "namespace/region/env/name/id"
		json   = instance.String()
		client = new(fake.EtcdClient)
	)

	oldEtcdClient := injectClient(client, machines, t)
	defer func() { etcdClient = oldEtcdClient }()

	registry := NewRegistry(machines, namespace)

	err := registry.Register(instance, ttl)
	if want, have := error(nil), err; want != have {
		t.Fatalf("want %v, have %v", want, have)
	}

	if want, have := 1, client.SetCallCount(); want != have {
		t.Fatalf("want %v, have %v", want, have)
	}

	actualPath, actualJSON, actualTTL := client.SetArgsForCall(0)

	if want, have := path, actualPath; want != have {
		t.Fatalf("want %v, have %v", want, have)
	}

	if want, have := json, actualJSON; want != have {
		t.Fatalf("want %v, have %v", want, have)
	}

	if want, have := ttl, actualTTL; want != have {
		t.Fatalf("want %v, have %v", want, have)
	}
}
Exemplo n.º 2
0
func TestLookupPath(t *testing.T) {
	var (
		namespace = "namespace"
		machines  = []string{"foo", "bar"}
		client    = new(fake.EtcdClient)

		pathTests = []struct {
			name       string
			env        string
			region     string
			lookupPath string
		}{
			{"name", "env", "region", "namespace/region/env/name"},
			{"name", "env", "", "namespace"},
			{"name", "", "region", "namespace/region"},
			{"", "env", "region", "namespace/region/env"},
		}
	)

	payload := `{
		"action": "get",
		"node": {
			"key": "/namespace/region",
			"dir": true,
			"nodes": [
				{
					"key": "/namespace/region/env",
					"dir": true,
					"nodes": [
						{
							"key": "/namespace/region/env/name",
							"dir": true,
							"nodes": [
								{
									"key": "/namespace/region/env/name/id",
									"value": "{\"name\":\"name\",\"id\":\"id\",\"region\":\"region\",\"env\":\"env\",\"host\":\"host\",\"port\":987}",
									"expiration": "2015-08-23T19:22:45.549917226Z",
									"ttl": 79,
									"modifiedIndex": 214,
									"createdIndex": 214
								}
							],
							"modifiedIndex": 157,
							"createdIndex": 157
						}
					],
					"modifiedIndex": 10,
					"createdIndex": 10
				}
			],
			"modifiedIndex": 10,
			"createdIndex": 10
		}
	}`

	var resp = new(etcd.Response)
	if want, have := error(nil), json.Unmarshal([]byte(payload), resp); want != have {
		t.Fatalf("want %v, have %v", want, have)
	}
	client.GetReturns(resp, nil)

	oldEtcdClient := injectClient(client, machines, t)
	defer func() { etcdClient = oldEtcdClient }()

	registry := NewRegistry(machines, namespace)

	for i, test := range pathTests {
		instances, err := registry.Lookup(test.name, test.env, test.region)
		if want, have := error(nil), err; want != have {
			t.Fatalf("want %v, have %v", want, have)
		}

		actualPath, actualSort, actualRecursive := client.GetArgsForCall(i)

		if want, have := test.lookupPath, actualPath; want != have {
			t.Fatalf("want %v, have %v", want, have)
		}

		if want, have := false, actualSort; want != have {
			t.Fatalf("want %v, have %v", want, have)
		}

		if want, have := true, actualRecursive; want != have {
			t.Fatalf("want %v, have %v", want, have)
		}

		if want, have := 1, len(instances); want != have {
			t.Fatalf("want %v, have %v", want, have)
		}

		instance := instances[0]

		if want, have := "region", instance.Region; want != have {
			t.Fatalf("want %v, have %v", want, have)
		}

		if want, have := "env", instance.Env; want != have {
			t.Fatalf("want %v, have %v", want, have)
		}

		if want, have := "name", instance.Name; want != have {
			t.Fatalf("want %v, have %v", want, have)
		}

		if want, have := "id", instance.ID; want != have {
			t.Fatalf("want %v, have %v", want, have)
		}
	}
}