Function TestClientChanAutoUnsub

test/sub_test.go:111–147

(t *testing.T)

}

func TestClientChanAutoUnsub(t *testing.T) {
	s := RunDefaultServer()
	defer s.Shutdown()

	nc := NewDefaultConnection(t)
	defer nc.Close()
	received := 0
	max := 10
	ch := make(chan *nats.Msg, max)
	sub, _ := nc.ChanSubscribe("foo", ch)
	sub.AutoUnsubscribe(max)
	total := 100
	for range total {
		nc.Publish("foo", []byte("Hello"))
	}
	nc.Flush()

	// Drain the channel
	for {
		select {
		case <-ch:
			received++
		case <-time.After(10 * time.Millisecond):
			if received != max {
				t.Fatalf("Received %d msgs, wanted only %d", received, max)
			}
			if sub.IsValid() {
				t.Fatal("Expected subscription to be invalid after hitting max")
			}
			if err := sub.AutoUnsubscribe(10); err == nil {
				t.Fatal("Calling AutoUnsubscribe() ob closed subscription should fail")
			}
			return
		}
	}

}

func TestClientASyncAutoUnsub(t *testing.T) {
	s := RunDefaultServer()