9747d05ae1
* refactor: take clients out of internal * refactor: move stdio to pkg * Move internal/api to api * refactor: final changes for Kapacitor to access shared functionality * chore: regenerate mocks * fix: bad automated refactor * chore: extra formatting not caught by make fmt
88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package ping_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/influxdata/influx-cli/v2/api"
|
|
"github.com/influxdata/influx-cli/v2/clients"
|
|
"github.com/influxdata/influx-cli/v2/clients/ping"
|
|
"github.com/influxdata/influx-cli/v2/internal/mock"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_PingSuccess(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctrl := gomock.NewController(t)
|
|
client := mock.NewMockHealthApi(ctrl)
|
|
client.EXPECT().GetHealth(gomock.Any()).Return(api.ApiGetHealthRequest{ApiService: client})
|
|
client.EXPECT().GetHealthExecute(gomock.Any()).Return(api.HealthCheck{Status: api.HEALTHCHECKSTATUS_PASS}, nil)
|
|
|
|
stdio := mock.NewMockStdIO(ctrl)
|
|
bytesWritten := bytes.Buffer{}
|
|
stdio.EXPECT().Write(gomock.Any()).DoAndReturn(bytesWritten.Write).AnyTimes()
|
|
cli := ping.Client{
|
|
CLI: clients.CLI{StdIO: stdio},
|
|
HealthApi: client,
|
|
}
|
|
|
|
require.NoError(t, cli.Ping(context.Background()))
|
|
require.Equal(t, "OK\n", bytesWritten.String())
|
|
}
|
|
|
|
func Test_PingFailedRequest(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
e := "the internet is down"
|
|
ctrl := gomock.NewController(t)
|
|
client := mock.NewMockHealthApi(ctrl)
|
|
client.EXPECT().GetHealth(gomock.Any()).Return(api.ApiGetHealthRequest{ApiService: client})
|
|
client.EXPECT().GetHealthExecute(gomock.Any()).Return(api.HealthCheck{}, errors.New(e))
|
|
cli := ping.Client{
|
|
HealthApi: client,
|
|
}
|
|
|
|
err := cli.Ping(context.Background())
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), e)
|
|
}
|
|
|
|
func Test_PingFailedStatus(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
e := "I broke"
|
|
ctrl := gomock.NewController(t)
|
|
client := mock.NewMockHealthApi(ctrl)
|
|
client.EXPECT().GetHealth(gomock.Any()).Return(api.ApiGetHealthRequest{ApiService: client})
|
|
client.EXPECT().GetHealthExecute(gomock.Any()).
|
|
Return(api.HealthCheck{}, &api.HealthCheck{Status: api.HEALTHCHECKSTATUS_FAIL, Message: &e})
|
|
cli := ping.Client{
|
|
HealthApi: client,
|
|
}
|
|
err := cli.Ping(context.Background())
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), e)
|
|
}
|
|
|
|
func Test_PingFailedStatusNoMessage(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
name := "foo"
|
|
ctrl := gomock.NewController(t)
|
|
client := mock.NewMockHealthApi(ctrl)
|
|
client.EXPECT().GetHealth(gomock.Any()).Return(api.ApiGetHealthRequest{ApiService: client})
|
|
client.EXPECT().GetHealthExecute(gomock.Any()).
|
|
Return(api.HealthCheck{}, &api.HealthCheck{Status: api.HEALTHCHECKSTATUS_FAIL, Name: name})
|
|
|
|
cli := ping.Client{
|
|
HealthApi: client,
|
|
}
|
|
err := cli.Ping(context.Background())
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), name)
|
|
}
|