You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
1.9 KiB
83 lines
1.9 KiB
package v1 |
|
|
|
import ( |
|
"context" |
|
"testing" |
|
"time" |
|
|
|
"go-common/app/service/main/broadcast/model" |
|
"go-common/library/log" |
|
"go-common/library/naming/discovery" |
|
"go-common/library/net/netutil/breaker" |
|
"go-common/library/net/rpc/warden" |
|
"go-common/library/net/rpc/warden/resolver" |
|
xtime "go-common/library/time" |
|
) |
|
|
|
func testInit() ZergClient { |
|
log.Init(nil) |
|
conf := &warden.ClientConfig{ |
|
Dial: xtime.Duration(time.Second * 10), |
|
Timeout: xtime.Duration(time.Second * 10), |
|
Breaker: &breaker.Config{ |
|
Window: xtime.Duration(3 * time.Second), |
|
Sleep: xtime.Duration(3 * time.Second), |
|
Bucket: 10, |
|
Ratio: 0.3, |
|
Request: 20, |
|
}, |
|
} |
|
wc := warden.NewClient(conf) |
|
resolver.Register(discovery.New(nil)) |
|
conn, err := wc.Dial(context.TODO(), "discovery://default/push.interface.broadcast") |
|
if err != nil { |
|
panic(err) |
|
} |
|
return NewZergClient(conn) |
|
} |
|
|
|
func TestPushMsg(t *testing.T) { |
|
client := testInit() |
|
time.Sleep(10 * time.Second) |
|
client.PushMsg(context.Background(), &PushMsgReq{ |
|
Keys: []string{"test"}, |
|
ProtoOp: model.OpSendMsg, |
|
Proto: &model.Proto{ |
|
Ver: 0, |
|
SeqId: 0, |
|
Operation: model.OpSendMsgReply, |
|
Body: []byte("{\"test1111111\"}"), |
|
}, |
|
}) |
|
} |
|
|
|
/* |
|
func TestBroadcastMsg(t *testing.T) { |
|
client := testInit() |
|
client.Broadcast(context.Background(), 102, &model.Proto{ |
|
Ver: 0, |
|
SeqId: 0, |
|
Operation: define.OP_SEND_SMS_REPLY, |
|
Body: []byte("{\"test broadcast 104\"}"), |
|
}) |
|
} |
|
|
|
func TestBroadcastRoom(t *testing.T) { |
|
client := testInit() |
|
client.BroadcastRoom(context.Background(), "test_room", &model.Proto{ |
|
Ver: 0, |
|
SeqId: 0, |
|
Operation: define.OP_SEND_SMS_REPLY, |
|
Body: []byte("{\"test broadcast\"}"), |
|
}) |
|
} |
|
|
|
func TestRooms(t *testing.T) { |
|
client := testInit() |
|
rooms, err := client.Rooms(context.Background()) |
|
if err != nil { |
|
t.Fatal(err) |
|
} |
|
t.Logf("TestRooms.rooms:%v", rooms) |
|
} |
|
*/
|
|
|