Function TestUnsubscribeAndNextMsgWithContext

test/context_test.go:749–790

(t *testing.T)

}

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

	nc := NewDefaultConnection(t)
	defer nc.Close()

	ctx, cancelCB := context.WithCancel(context.Background())
	defer cancelCB() // should always be called, not discarded, to prevent context leak

	sub, err := nc.SubscribeSync("foo")
	if err != nil {
		t.Fatalf("Expected to be able to subscribe: %s", err)
	}
	sub.Unsubscribe()
	if _, err = sub.NextMsgWithContext(ctx); err != nats.ErrBadSubscription {
		t.Fatalf("Expected '%v', but got: '%v'", nats.ErrBadSubscription, err)
	}

	ctx, cancelCB = context.WithCancel(context.Background())
	defer cancelCB() // should always be called, not discarded, to prevent context leak

	sub, err = nc.SubscribeSync("foo")
	if err != nil {
		t.Fatalf("Expected to be able to subscribe: %s", err)
	}

	// Now make sure we get same error when unsubscribing from separate routine
	// while in the call.
	wg := sync.WaitGroup{}
	wg.Add(1)
	go func() {
		time.Sleep(100 * time.Millisecond)
		sub.Unsubscribe()
		wg.Done()
	}()

	if _, err = sub.NextMsgWithContext(ctx); err != nats.ErrBadSubscription {
		t.Fatalf("Expected '%v', but got: '%v'", nats.ErrBadSubscription, err)
	}
	wg.Wait()
}

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