|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119 |
- package client
-
- import (
- "testing"
- )
-
- var (
- pool = NewPool()
- )
-
- func TestPoolAdd(t *testing.T) {
- t.Log("Add servers")
- if err := pool.Add("tcp4", "127.0.0.1:4730", 1); err != nil {
- t.Error(err)
- }
- if err := pool.Add("tcp4", "127.0.1.1:4730", 1); err != nil {
- t.Error(err)
- }
- if len(pool.clients) != 2 {
- t.Errorf("2 servers expected, %d got.", len(pool.clients))
- }
- }
-
- func TestPoolEcho(t *testing.T) {
- echo, err := pool.Echo("", []byte(TestStr))
- if err != nil {
- t.Error(err)
- return
- }
- if string(echo) != TestStr {
- t.Errorf("Invalid echo data: %s", echo)
- return
- }
-
- _, err = pool.Echo("not exists", []byte(TestStr))
- if err != ErrNotFound {
- t.Errorf("ErrNotFound expected, got %s", err)
- }
- }
-
- func TestPoolDoBg(t *testing.T) {
- addr, handle, err := pool.DoBg("ToUpper",
- []byte("abcdef"), JOB_LOW)
- if err != nil {
- t.Error(err)
- return
- }
- if handle == "" {
- t.Error("Handle is empty.")
- } else {
- t.Log(addr, handle)
- }
- }
-
- func TestPoolDo(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
- }
- addr, handle, err := pool.Do("ToUpper",
- []byte("abcdef"), JOB_LOW, jobHandler)
- if err != nil {
- t.Error(err)
- }
- if handle == "" {
- t.Error("Handle is empty.")
- } else {
- t.Log(addr, handle)
- }
- }
-
- func TestPoolStatus(t *testing.T) {
- status, err := pool.Status("127.0.0.1:4730", "handle not exists")
- if err != nil {
- t.Error(err)
- return
- }
- if status.Known {
- t.Errorf("The job (%s) shouldn't be known.", status.Handle)
- }
- if status.Running {
- t.Errorf("The job (%s) shouldn't be running.", status.Handle)
- }
- addr, handle, err := pool.Do("Delay5sec",
- []byte("abcdef"), JOB_LOW, nil)
- if err != nil {
- t.Error(err)
- return
- }
- status, err = pool.Status(addr, handle)
- if err != nil {
- t.Error(err)
- return
- }
-
- if !status.Known {
- t.Errorf("The job (%s) should be known.", status.Handle)
- }
- if status.Running {
- t.Errorf("The job (%s) shouldn't be running.", status.Handle)
- }
- status, err = pool.Status("not exists", "not exists")
- if err != ErrNotFound {
- t.Error(err)
- return
- }
- }
-
- func TestPoolClose(t *testing.T) {
- return
- if err := pool.Close(); err != nil {
- t.Error(err)
- }
- }
|