f95668ada6
* build: add goimports to fmt target to remove unused imports * feat: update codegen template to support returning raw response body * feat: add support for gunzip-ing response bodies * refactor: remove unused piece from codegen return values
77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
package internal_test
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/influxdata/influx-cli/v2/internal"
|
|
"github.com/influxdata/influx-cli/v2/internal/api"
|
|
"github.com/influxdata/influx-cli/v2/internal/mock"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_PingSuccess(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
client := &mock.HealthApi{
|
|
GetHealthExecuteFn: func(req api.ApiGetHealthRequest) (api.HealthCheck, error) {
|
|
return api.HealthCheck{Status: api.HEALTHCHECKSTATUS_PASS}, nil
|
|
},
|
|
}
|
|
|
|
stdio := mock.NewMockStdio(nil, true)
|
|
cli := &internal.CLI{StdIO: stdio}
|
|
|
|
require.NoError(t, cli.Ping(context.Background(), client))
|
|
require.Equal(t, "OK\n", stdio.Stdout())
|
|
}
|
|
|
|
func Test_PingFailedRequest(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
e := "the internet is down"
|
|
client := &mock.HealthApi{
|
|
GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, error) {
|
|
return api.HealthCheck{}, errors.New(e)
|
|
},
|
|
}
|
|
|
|
cli := &internal.CLI{}
|
|
err := cli.Ping(context.Background(), client)
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), e)
|
|
}
|
|
|
|
func Test_PingFailedStatus(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
e := "I broke"
|
|
client := &mock.HealthApi{
|
|
GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, error) {
|
|
return api.HealthCheck{}, &api.HealthCheck{Status: api.HEALTHCHECKSTATUS_FAIL, Message: &e}
|
|
},
|
|
}
|
|
|
|
cli := &internal.CLI{}
|
|
err := cli.Ping(context.Background(), client)
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), e)
|
|
}
|
|
|
|
func Test_PingFailedStatusNoMessage(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
name := "foo"
|
|
client := &mock.HealthApi{
|
|
GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, error) {
|
|
return api.HealthCheck{}, &api.HealthCheck{Status: api.HEALTHCHECKSTATUS_FAIL, Name: name}
|
|
},
|
|
}
|
|
|
|
cli := &internal.CLI{}
|
|
err := cli.Ping(context.Background(), client)
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), name)
|
|
}
|