hcq/influx-cli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: replace bash hacks with custom templates (#34)

Browse Source
This commit is contained in:
Daniel Moran
2021-04-19 15:51:53 -04:00
committed by GitHub
parent 4f62e469e9
commit 079f707c21
9 changed files with 1323 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
internal/ping_test.go
View File

@ -30,7 +30,8 @@ func Test_PingSuccess(t *testing.T) {
t.Parallel()
client := &testClient{
GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) {
GetHealthExecuteFn: func(req api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) {
require.Nil(t, req.GetZapTraceSpan())
return api.HealthCheck{Status: api.HEALTHCHECKSTATUS_PASS}, nil, nil
},
}
@ -42,6 +43,25 @@ func Test_PingSuccess(t *testing.T) {
require.Equal(t, "OK\n", out.String())
}
func Test_PingSuccessWithTracing(t *testing.T) {
t.Parallel()
traceId := "trace-id"
client := &testClient{
GetHealthExecuteFn: func(req api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) {
require.NotNil(t, req.GetZapTraceSpan())
require.Equal(t, traceId, *req.GetZapTraceSpan())
return api.HealthCheck{Status: api.HEALTHCHECKSTATUS_PASS}, nil, nil
},
}
out := &bytes.Buffer{}
cli := &internal.CLI{Stdout: out, TraceId: traceId}
require.NoError(t, cli.Ping(context.Background(), client))
require.Equal(t, "OK\n", out.String())
}
func Test_PingFailedRequest(t *testing.T) {
t.Parallel()