2012-05-22 20:05:39 +08:00
|
|
|
package client
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
|
|
|
var client *Client
|
|
|
|
|
|
|
|
func TestClientAddServer(t *testing.T) {
|
|
|
|
t.Log("Add local server 127.0.0.1:4730")
|
|
|
|
var err error
|
2012-05-23 15:22:29 +08:00
|
|
|
if client, err = New("127.0.0.1:4730"); err != nil {
|
2012-05-22 20:05:39 +08:00
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
}
|
2013-01-14 17:59:48 +08:00
|
|
|
|
2012-05-22 20:05:39 +08:00
|
|
|
func TestClientEcho(t *testing.T) {
|
2013-01-15 17:55:44 +08:00
|
|
|
if echo := string(client.Echo([]byte("Hello world"))); echo == "Hello world" {
|
|
|
|
t.Log(echo)
|
|
|
|
} else {
|
|
|
|
t.Errorf("Invalid echo data: %s", echo)
|
2013-01-14 17:59:48 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestClientDoBg(t *testing.T) {
|
2013-01-15 17:55:44 +08:00
|
|
|
if handle := client.DoBg("ToUpper", []byte("abcdef"),
|
|
|
|
JOB_LOW); handle == "" {
|
|
|
|
t.Error("Handle is empty.")
|
2013-01-14 17:59:48 +08:00
|
|
|
} else {
|
|
|
|
t.Log(handle)
|
2012-05-22 20:05:39 +08:00
|
|
|
}
|
|
|
|
}
|
2013-01-14 17:59:48 +08:00
|
|
|
|
2012-05-22 20:05:39 +08:00
|
|
|
func TestClientDo(t *testing.T) {
|
2013-01-14 17:59:48 +08:00
|
|
|
jobHandler := func(job *Job) {
|
|
|
|
str := string(job.Data)
|
|
|
|
if str == "ABCDEF" {
|
|
|
|
t.Log(str)
|
|
|
|
} else {
|
|
|
|
t.Errorf("Invalid data: %s", job.Data)
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
2013-01-15 17:55:44 +08:00
|
|
|
if handle := client.Do("ToUpper", []byte("abcdef"),
|
|
|
|
JOB_LOW, jobHandler); handle == "" {
|
|
|
|
t.Error("Handle is empty.")
|
2012-05-22 20:05:39 +08:00
|
|
|
} else {
|
|
|
|
t.Log(handle)
|
|
|
|
}
|
|
|
|
}
|
2013-01-14 17:59:48 +08:00
|
|
|
|
|
|
|
func TestClientStatus(t *testing.T) {
|
|
|
|
|
2013-01-15 17:55:44 +08:00
|
|
|
s1 := client.Status("handle not exists")
|
|
|
|
if s1.Known {
|
|
|
|
t.Errorf("The job (%s) shouldn't be known.", s1.Handle)
|
|
|
|
}
|
|
|
|
if s1.Running {
|
|
|
|
t.Errorf("The job (%s) shouldn't be running.", s1.Handle)
|
|
|
|
}
|
2013-01-14 17:59:48 +08:00
|
|
|
|
2013-01-15 17:55:44 +08:00
|
|
|
handle := client.Do("Delay5sec", []byte("abcdef"), JOB_LOW, nil);
|
|
|
|
s2 := client.Status(handle)
|
|
|
|
if !s2.Known {
|
|
|
|
t.Errorf("The job (%s) should be known.", s2.Handle)
|
|
|
|
}
|
|
|
|
if s2.Running {
|
|
|
|
t.Errorf("The job (%s) shouldn't be running.", s2.Handle)
|
2013-01-14 17:59:48 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-05-22 20:05:39 +08:00
|
|
|
func TestClientClose(t *testing.T) {
|
|
|
|
if err := client.Close(); err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
}
|