scottk/vault
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/vault.git synced 2025-10-30 02:02:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removing timing-dependent aspects of test. (#8261)

Browse Source
This commit is contained in:
ncabatoff
2020-01-30 11:02:48 -05:00
committed by GitHub
parent f3d1cc21f2
commit 20c514cc60
1 changed files with 9 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
serviceregistration/consul/consul_service_registration_test.go
View File

@@ -64,9 +64,9 @@ func TestConsul_ServiceRegistration(t *testing.T) {
t.Fatal(err)
}
// transitionFrom waits patiently for the services in the Consul catalog to
// transition from a known value, and then returns the new value.
transitionFrom := func(t *testing.T, known map[string][]string) map[string][]string {
// waitForServices waits for the services in the Consul catalog to
// reach an expected value, returning the delta if that doesn't happen in time.
waitForServices := func(t *testing.T, expected map[string][]string) map[string][]string {
t.Helper()
// Wait for up to 10 seconds
for i := 0; i < 10; i++ {
@@ -74,12 +74,12 @@ func TestConsul_ServiceRegistration(t *testing.T) {
if err != nil {
t.Fatal(err)
}
if diff := deep.Equal(services, known); diff != nil {
if diff := deep.Equal(services, expected); diff == nil {
return services
}
time.Sleep(time.Second)
}
t.Fatalf("Catalog Services never transitioned from %v", known)
t.Fatalf("Catalog Services never reached expected state %v", expected)
return nil
}
@@ -122,27 +122,10 @@ func TestConsul_ServiceRegistration(t *testing.T) {
t.Fatal(err)
}
// Vault should not yet be registered with Consul
services, _, err := client.Catalog().Services(nil)
if err != nil {
t.Fatal(err)
}
if diff := deep.Equal(services, map[string][]string{
"consul": []string{},
}); diff != nil {
t.Fatal(diff)
}
// Vault should soon be registered with Consul in standby mode
services = transitionFrom(t, map[string][]string{
"consul": []string{},
})
if diff := deep.Equal(services, map[string][]string{
waitForServices(t, map[string][]string{
"consul": []string{},
"vault": []string{"standby"},
}); diff != nil {
t.Fatal(diff)
}
})
// Initialize and unseal the core
keys, _ := vault.TestCoreInit(t, core)
@@ -158,17 +141,10 @@ func TestConsul_ServiceRegistration(t *testing.T) {
// Wait for the core to become active
vault.TestWaitActive(t, core)
// Vault should soon be registered with Consul in active mode
services = transitionFrom(t, map[string][]string{
"consul": []string{},
"vault": []string{"standby"},
})
if diff := deep.Equal(services, map[string][]string{
waitForServices(t, map[string][]string{
"consul": []string{},
"vault": []string{"active"},
}); diff != nil {
t.Fatal(diff)
}
})
}
func TestConsul_ServiceTags(t *testing.T) {