func TestRetry(t *testing.T) {
th := test.NewServiceHost(t)
state := &host.ExampleServiceDocument{}
stateURI := th.StartMock(state)
client.DefaultTransport.DisableKeepAlives = true
get := func(retry int) error {
ctx := context.Background()
op := operation.NewGet(ctx, stateURI)
op.SetRetryCount(retry)
return client.Send(op).Wait()
}
if err := get(0); err != nil {
t.Fatal(err)
}
client.DefaultTransport.CloseIdleConnections()
if err := th.Stop(); err != nil {
t.Fatal(err)
}
if err := get(1); err == nil {
t.Fatal("expected error")
}
}
func GetAuthCredentialsServiceState(ctx context.Context, a *AuthCredentialsServiceState, u uri.URI) error {
op := operation.NewGet(ctx, u)
err := client.Send(op).Wait()
if err != nil {
return err
}
if err := op.DecodeBody(a); err != nil {
return err
}
return nil
}
func WaitForNonEmptyBody(ctx context.Context, u uri.URI) error {
return poll(ctx, func() (bool, error) {
op := operation.NewGet(ctx, u)
if err := client.Send(op).Wait(); err != nil {
return false, fmt.Errorf("Error issuing GET: %s", err)
}
var m map[string]interface{}
if err := op.DecodeBody(&m); err != nil {
return false, fmt.Errorf("Error decoding body: %s", err)
}
return len(m) > 0, nil
})
}
// GetServiceHostManagementState returns the local node's NodeState
func GetServiceHostManagementState(ctx context.Context) (*common.ServiceHostState, error) {
u := uri.Extend(uri.Local(), common.Management)
p := operation.NewGet(ctx, u)
client.Send(p)
if err := p.Wait(); err != nil {
return nil, err
}
var state common.ServiceHostState
err := p.DecodeBody(&state)
if err != nil {
return nil, err
}
return &state, nil
}