pkg/task/client_test.go

55 lines
959 B
Go

package task
import (
"os"
"testing"
"github.com/stretchr/testify/assert"
)
func TestListTask(t *testing.T) {
host := os.Getenv("TEST_TASK_API_HOST")
if host == "" {
t.Skip()
}
c := NewClient(host)
resp, err := c.ListTask(t.Context(), &ListReq{})
assert.NoError(t, err)
t.Log(resp)
}
func TestGetTaskStatus(t *testing.T) {
host := os.Getenv("TEST_TASK_API_HOST")
if host == "" {
t.Skip()
}
c := NewClient(host)
resp, err := c.GetTaskStatus(t.Context(), &IDReq{TaskID: ""})
assert.NoError(t, err)
t.Log(resp)
}
func TestRetryTask(t *testing.T) {
host := os.Getenv("TEST_TASK_API_HOST")
if host == "" {
t.Skip()
}
c := NewClient(host)
err := c.RetryTask(t.Context(), &IDReq{TaskID: ""})
assert.NoError(t, err)
}
func TestRevokeTask(t *testing.T) {
host := os.Getenv("TEST_TASK_API_HOST")
if host == "" {
t.Skip()
}
c := NewClient(host)
err := c.RevokeTask(t.Context(), &IDReq{TaskID: ""})
assert.NoError(t, err)
}