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)
}