Function TestOpenSubscriptionFromURL

pubsub/rabbitpubsub/rabbit_test.go:444–497

(t *testing.T)

}

func TestOpenSubscriptionFromURL(t *testing.T) {
	t.Setenv("RABBIT_SERVER_URL", rabbitURL)

	tests := []struct {
		label       string
		URLTemplate string
		WantErr     bool
	}{
		{"url with no QoS prefetch count", "rabbit://%s", false},
		{"invalid parameters", "rabbit://%s?param=value", true},
		{"valid url with QoS prefetch count", "rabbit://%s?prefetch_count=1024", false},
		{"invalid url with QoS prefetch count", "rabbit://%s?prefetch_count=value", true},
	}

	for _, test := range tests {
		t.Run(test.label, func(t *testing.T) {
			conn := mustDialRabbit(t)
			_, isFake := conn.(*fakeConnection)
			if isFake {
				t.Skip("test requires real rabbitmq")
			}

			h := &harness{conn: conn}

			ctx := context.Background()

			dt, cleanupTopic, err := h.CreateTopic(ctx, t.Name())
			if err != nil {
				t.Fatalf("unable to create topic: %v", err)
			}

			t.Cleanup(cleanupTopic)

			ds, cleanupSubscription, err := h.CreateSubscription(ctx, dt, t.Name())
			if err != nil {
				t.Fatalf("unable to create subscription: %v", err)
			}

			t.Cleanup(cleanupSubscription)

			queue := ds.(*subscription).queue
			url := fmt.Sprintf(test.URLTemplate, queue)

			sub, err := pubsub.OpenSubscription(ctx, url)
			if (err != nil) != test.WantErr {
				t.Errorf("%s: got error %v, want error %v", test.URLTemplate, err, test.WantErr)
			}

			if sub != nil {
				sub.Shutdown(ctx)
			}
		})
	}
}