gogrlx/nats-server
1
0
Fork 0
mirror of https://github.com/gogrlx/nats-server.git synced 2026-04-02 03:38:42 -07:00
Code Issues Packages Projects Releases Wiki Activity

Added more tests for unsub and pubarg/msgs splits

Browse Source
This commit is contained in:
Derek Collison
2012-11-20 08:11:43 -08:00
parent f2db136e58
commit b466290e0a
1 changed files with 86 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
server/split_test.go
View File

@@ -46,6 +46,40 @@ func TestSplitBufferSubOp(t *testing.T) {
}
}
func TestSplitBufferUnsubOp(t *testing.T) {
s := &Server{ sl: sublist.New() }
c := &client{srv:s, subs: hashmap.New()}
subop := []byte("SUB foo 1024\r\n")
if err := c.parse(subop) ; err != nil {
t.Fatalf("Unexpected parse error: %v\n", err)
}
if c.state != OP_START {
t.Fatalf("Expected OP_START state vs %d\n", c.state)
}
unsubop := []byte("UNSUB 1024\r\n")
unsubop1 := unsubop[:8]
unsubop2 := unsubop[8:]
if err := c.parse(unsubop1) ; err != nil {
t.Fatalf("Unexpected parse error: %v\n", err)
}
if c.state != UNSUB_ARG {
t.Fatalf("Expected UNSUB_ARG state vs %d\n", c.state)
}
if err := c.parse(unsubop2) ; err != nil {
t.Fatalf("Unexpected parse error: %v\n", err)
}
if c.state != OP_START {
t.Fatalf("Expected OP_START state vs %d\n", c.state)
}
r := s.sl.Match([]byte("foo"))
if r != nil && len(r) != 0 {
t.Fatalf("Should be no subscriptions in results: %+v\n", r)
}
}
func TestSplitBufferPubOp(t *testing.T) {
c := &client{subs: hashmap.New()}
pub := []byte("PUB foo.bar INBOX.22 11\r\nhello world\r")
@@ -131,3 +165,55 @@ func TestSplitBufferPubOp2(t *testing.T) {
t.Fatalf("Expected OP_START state vs %d\n", c.state)
}
}
func TestSplitBufferPubOp3(t *testing.T) {
c := &client{subs: hashmap.New()}
pubAll := []byte("PUB foo bar 11\r\nhello world\r\n")
pub := pubAll[:16]
if err := c.parse(pub); err != nil {
t.Fatalf("Unexpected parse error: %v\n", err)
}
if !bytes.Equal(c.pa.subject, []byte("foo")) {
t.Fatalf("Unexpected subject: '%s' vs '%s'\n", c.pa.subject, "foo")
}
// Simulate next read of network, make sure pub state is saved
// until msg payload has cleared.
copy(pubAll, "XXXXXXXXXXXXXXXX")
if !bytes.Equal(c.pa.subject, []byte("foo")) {
t.Fatalf("Unexpected subject: '%s' vs '%s'\n", c.pa.subject, "foo")
}
if !bytes.Equal(c.pa.reply, []byte("bar")) {
t.Fatalf("Unexpected reply: '%s' vs '%s'\n", c.pa.reply, "bar")
}
if !bytes.Equal(c.pa.szb, []byte("11")) {
t.Fatalf("Unexpected size bytes: '%s' vs '%s'\n", c.pa.szb, "11")
}
}
func TestSplitBufferPubOp4(t *testing.T) {
c := &client{subs: hashmap.New()}
pubAll := []byte("PUB foo 11\r\nhello world\r\n")
pub := pubAll[:12]
if err := c.parse(pub); err != nil {
t.Fatalf("Unexpected parse error: %v\n", err)
}
if !bytes.Equal(c.pa.subject, []byte("foo")) {
t.Fatalf("Unexpected subject: '%s' vs '%s'\n", c.pa.subject, "foo")
}
// Simulate next read of network, make sure pub state is saved
// until msg payload has cleared.
copy(pubAll, "XXXXXXXXXXXX")
if !bytes.Equal(c.pa.subject, []byte("foo")) {
t.Fatalf("Unexpected subject: '%s' vs '%s'\n", c.pa.subject, "foo")
}
if !bytes.Equal(c.pa.reply, []byte("")) {
t.Fatalf("Unexpected reply: '%s' vs '%s'\n", c.pa.reply, "")
}
if !bytes.Equal(c.pa.szb, []byte("11")) {
t.Fatalf("Unexpected size bytes: '%s' vs '%s'\n", c.pa.szb, "11")
}
}