gearman-go/client/client_test.go

144 lines
2.8 KiB
Go
Raw Normal View History

package client
import (
"flag"
"os"
2013-08-29 16:51:23 +08:00
"testing"
)
2013-08-30 11:20:51 +08:00
const (
TestStr = "Hello world"
)
var (
client *Client
runIntegrationTests bool
)
func TestMain(m *testing.M) {
integrationsTestFlag := flag.Bool("integration", false, "Run the integration tests (in addition to the unit tests)")
if integrationsTestFlag != nil {
runIntegrationTests = *integrationsTestFlag
}
code := m.Run()
os.Exit(code)
}
2013-08-29 18:08:05 +08:00
func TestClientAddServer(t *testing.T) {
if !runIntegrationTests {
t.Skip("To run this test, use: go test -integration")
}
2013-08-29 16:51:23 +08:00
t.Log("Add local server 127.0.0.1:4730")
var err error
if client, err = New(Network, "127.0.0.1:4730"); err != nil {
2013-09-22 22:02:05 +08:00
t.Fatal(err)
2013-08-29 16:51:23 +08:00
}
2013-08-29 18:08:05 +08:00
client.ErrorHandler = func(e error) {
2013-08-29 16:51:23 +08:00
t.Log(e)
}
}
2013-01-14 17:59:48 +08:00
func TestClientEcho(t *testing.T) {
if !runIntegrationTests {
t.Skip("To run this test, use: go test -integration")
}
2013-08-30 11:20:51 +08:00
echo, err := client.Echo([]byte(TestStr))
2013-08-29 16:51:23 +08:00
if err != nil {
t.Error(err)
return
}
2013-08-30 11:20:51 +08:00
if string(echo) != TestStr {
t.Errorf("Echo error, %s expected, %s got", TestStr, echo)
return
}
2013-01-14 17:59:48 +08:00
}
func TestClientDoBg(t *testing.T) {
if !runIntegrationTests {
t.Skip("To run this test, use: go test -integration")
}
handle, err := client.DoBg("ToUpper", []byte("abcdef"), JobLow)
2013-08-30 11:41:18 +08:00
if err != nil {
t.Error(err)
return
}
if handle == "" {
2013-08-29 16:51:23 +08:00
t.Error("Handle is empty.")
2013-08-30 11:20:51 +08:00
} else {
t.Log(handle)
2013-08-29 16:51:23 +08:00
}
}
2013-01-14 17:59:48 +08:00
func TestClientDo(t *testing.T) {
if !runIntegrationTests {
t.Skip("To run this test, use: go test -integration")
}
2013-08-29 18:08:05 +08:00
jobHandler := func(job *Response) {
2013-08-29 16:51:23 +08:00
str := string(job.Data)
if str == "ABCDEF" {
t.Log(str)
} else {
t.Errorf("Invalid data: %s", job.Data)
}
return
}
2013-08-30 11:41:18 +08:00
handle, err := client.Do("ToUpper", []byte("abcdef"),
JobLow, jobHandler)
2013-08-30 11:41:18 +08:00
if err != nil {
t.Error(err)
return
}
if handle == "" {
2013-08-29 16:51:23 +08:00
t.Error("Handle is empty.")
} else {
t.Log(handle)
}
}
2013-01-14 17:59:48 +08:00
func TestClientStatus(t *testing.T) {
if !runIntegrationTests {
t.Skip("To run this test, use: go test -integration")
}
2013-08-30 11:20:51 +08:00
status, err := client.Status("handle not exists")
2013-08-29 16:51:23 +08:00
if err != nil {
t.Error(err)
return
}
2013-08-30 11:20:51 +08:00
if status.Known {
t.Errorf("The job (%s) shouldn't be known.", status.Handle)
return
}
if status.Running {
t.Errorf("The job (%s) shouldn't be running.", status.Handle)
return
}
2013-01-14 17:59:48 +08:00
handle, err := client.Do("Delay5sec", []byte("abcdef"), JobLow, nil)
2013-08-30 11:41:18 +08:00
if err != nil {
t.Error(err)
return
}
2013-08-30 11:20:51 +08:00
status, err = client.Status(handle)
if err != nil {
t.Error(err)
return
}
if !status.Known {
t.Errorf("The job (%s) should be known.", status.Handle)
return
}
if status.Running {
t.Errorf("The job (%s) shouldn't be running.", status.Handle)
return
}
2013-01-14 17:59:48 +08:00
}
func TestClientClose(t *testing.T) {
if !runIntegrationTests {
t.Skip("To run this test, use: go test -integration")
}
2013-08-29 16:51:23 +08:00
if err := client.Close(); err != nil {
t.Error(err)
}
}