mirror of https://github.com/status-im/consul.git
66 lines
2.1 KiB
Go
66 lines
2.1 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package assert
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/consul/api"
|
|
"github.com/hashicorp/consul/sdk/testutil/retry"
|
|
)
|
|
|
|
// PeeringStatus verifies the peering connection is the specified state with a default retry.
|
|
func PeeringStatus(t *testing.T, client *api.Client, peerName string, status api.PeeringState) {
|
|
PeeringStatusOpts(t, client, peerName, status, nil)
|
|
}
|
|
|
|
// PeeringStatusOpts verifies the peering connection is the specified
|
|
// state with a default retry with options.
|
|
func PeeringStatusOpts(t *testing.T, client *api.Client, peerName string, status api.PeeringState, opts *api.QueryOptions) {
|
|
failer := func() *retry.Timer {
|
|
return &retry.Timer{Timeout: 180 * time.Second, Wait: defaultWait}
|
|
}
|
|
|
|
retry.RunWith(failer(), t, func(r *retry.R) {
|
|
peering, _, err := client.Peerings().Read(context.Background(), peerName, opts)
|
|
if err != nil {
|
|
r.Fatal("error reading peering data")
|
|
}
|
|
if peering == nil {
|
|
r.Fatal("peering not found")
|
|
}
|
|
if status != peering.State {
|
|
r.Fatal("peering state did not match: got ", peering.State, " want ", status)
|
|
}
|
|
})
|
|
}
|
|
|
|
// PeeringExports verifies the correct number of exported services with a default retry.
|
|
func PeeringExports(t *testing.T, client *api.Client, peerName string, exports int) {
|
|
PeeringExportsOpts(t, client, peerName, exports, nil)
|
|
}
|
|
|
|
// PeeringExportsOpts verifies the correct number of exported services
|
|
// with a default retry with options.
|
|
func PeeringExportsOpts(t *testing.T, client *api.Client, peerName string, exports int, opts *api.QueryOptions) {
|
|
failer := func() *retry.Timer {
|
|
return &retry.Timer{Timeout: defaultTimeout, Wait: defaultWait}
|
|
}
|
|
|
|
retry.RunWith(failer(), t, func(r *retry.R) {
|
|
peering, _, err := client.Peerings().Read(context.Background(), peerName, opts)
|
|
if err != nil {
|
|
r.Fatal("error reading peering data")
|
|
}
|
|
if peering == nil {
|
|
r.Fatal("peering not found")
|
|
}
|
|
if exports != len(peering.StreamStatus.ExportedServices) {
|
|
r.Fatal("peering exported services did not match: got ", len(peering.StreamStatus.ExportedServices), " want ", exports)
|
|
}
|
|
})
|
|
}
|