Function TestEndpointPendingLimits

micro/test/service_test.go:1914–2092

(t *testing.T)

}

func TestEndpointPendingLimits(t *testing.T) {
	s := RunServerOnPort(-1)
	defer s.Shutdown()

	nc, err := nats.Connect(s.ClientURL())
	if err != nil {
		t.Fatalf("Error connecting to server: %v", err)
	}
	defer nc.Close()

	handler := micro.HandlerFunc(func(req micro.Request) {
		time.Sleep(10 * time.Millisecond)
		req.Respond([]byte("OK"))
	})

	t.Run("with pending limits", func(t *testing.T) {
		srv, err := micro.AddService(nc, micro.Config{
			Name:    "test-service",
			Version: "1.0.0",
		})
		if err != nil {
			t.Fatalf("Error adding service: %v", err)
		}
		defer srv.Stop()

		msgLimit := 10
		bytesLimit := 1024

		err = srv.AddEndpoint("test", handler,
			micro.WithEndpointSubject("test.pending"),
			micro.WithEndpointPendingLimits(msgLimit, bytesLimit))
		if err != nil {
			t.Fatalf("Error adding endpoint: %v", err)
		}

		// Verify endpoint was created
		info := srv.Info()
		if len(info.Endpoints) != 1 {
			t.Fatalf("Expected 1 endpoint, got %d", len(info.Endpoints))
		}

		// Send messages and verify they're handled
		for i := range 5 {
			resp, err := nc.Request("test.pending", []byte("test"), 100*time.Millisecond)
			if err != nil {
				t.Fatalf("Error on request %d: %v", i, err)
			}
			if string(resp.Data) != "OK" {
				t.Fatalf("Expected OK response, got %s", string(resp.Data))
			}
		}
	})

	t.Run("with only msg limit", func(t *testing.T) {
		srv, err := micro.AddService(nc, micro.Config{
			Name:    "test-service-msg-only",
			Version: "1.0.0",
		})