44 lines
1.5 KiB
Go
44 lines
1.5 KiB
Go
|
package e2eutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
capi "github.com/hashicorp/consul/api"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RequireConsulStatus asserts the aggregate health of the service converges to
|
||
|
|
// the expected status
|
||
|
|
func RequireConsulStatus(require *require.Assertions,
|
||
|
|
client *capi.Client, serviceName, expectedStatus string) {
|
||
|
|
require.Eventually(func() bool {
|
||
|
|
_, status := serviceStatus(require, client, serviceName)
|
||
|
|
return status == expectedStatus
|
||
|
|
}, 30*time.Second, time.Second, // needs a long time for killing tasks/clients
|
||
|
|
"timed out expecting %q to become %q",
|
||
|
|
serviceName, expectedStatus,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
// serviceStatus gets the aggregate health of the service and returns
|
||
|
|
// the []ServiceEntry for further checking
|
||
|
|
func serviceStatus(require *require.Assertions,
|
||
|
|
client *capi.Client, serviceName string) ([]*capi.ServiceEntry, string) {
|
||
|
|
services, _, err := client.Health().Service(serviceName, "", false, nil)
|
||
|
|
require.NoError(err, "expected no error for %q, got %v", serviceName, err)
|
||
|
|
if len(services) > 0 {
|
||
|
|
return services, services[0].Checks.AggregatedStatus()
|
||
|
|
}
|
||
|
|
return nil, "(unknown status)"
|
||
|
|
}
|
||
|
|
|
||
|
|
// RequireConsulDeregistered asserts that the service eventually is deregistered from Consul
|
||
|
|
func RequireConsulDeregistered(require *require.Assertions,
|
||
|
|
client *capi.Client, serviceName string) {
|
||
|
|
require.Eventually(func() bool {
|
||
|
|
services, _, err := client.Health().Service(serviceName, "", false, nil)
|
||
|
|
require.NoError(err, "expected no error for %q, got %v", serviceName, err)
|
||
|
|
return len(services) == 0
|
||
|
|
}, 5*time.Second, time.Second)
|
||
|
|
}
|