package client import ( "testing" ) const ( TestStr = "Hello world" ) var client *Client func TestClientAddServer(t *testing.T) { t.Log("Add local server 127.0.0.1:4730") var err error if client, err = New("tcp4", "127.0.0.1:4730"); err != nil { t.Error(err) return } client.ErrorHandler = func(e error) { t.Log(e) } } func TestClientEcho(t *testing.T) { echo, err := client.Echo([]byte(TestStr)) if err != nil { t.Error(err) return } if string(echo) != TestStr { t.Errorf("Echo error, %s expected, %s got", TestStr, echo) return } } func TestClientDoBg(t *testing.T) { handle, err := client.DoBg("ToUpper", []byte("abcdef"), JOB_LOW) if err != nil { t.Error(err) return } if handle == "" { t.Error("Handle is empty.") } else { t.Log(handle) } } func TestClientDo(t *testing.T) { jobHandler := func(job *Response) { str := string(job.Data) if str == "ABCDEF" { t.Log(str) } else { t.Errorf("Invalid data: %s", job.Data) } return } handle, err := client.Do("ToUpper", []byte("abcdef"), JOB_LOW, jobHandler) if err != nil { t.Error(err) return } if handle == "" { t.Error("Handle is empty.") } else { t.Log(handle) } } func TestClientStatus(t *testing.T) { status, err := client.Status("handle not exists") if err != nil { t.Error(err) return } 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 } handle, err := client.Do("Delay5sec", []byte("abcdef"), JOB_LOW, nil) if err != nil { t.Error(err) return } 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 } } func TestClientClose(t *testing.T) { if err := client.Close(); err != nil { t.Error(err) } }