0.29.1原版

2026-07-02 19:14:14 +08:00
commit d94008f0fb
947 changed files with 174905 additions and 0 deletions
@@ -0,0 +1,39 @@
+package v1
+
+import (
+	"context"
+	"testing"
+
+	"connectrpc.com/connect"
+	"google.golang.org/grpc/metadata"
+	"google.golang.org/protobuf/types/known/emptypb"
+)
+
+func TestMetadataInterceptorForwardsSecurityHeaders(t *testing.T) {
+	interceptor := NewMetadataInterceptor()
+	req := connect.NewRequest(&emptypb.Empty{})
+	req.Header().Set("Origin", "https://memos.example")
+	req.Header().Set("X-Forwarded-Proto", "https")
+	req.Header().Set("Forwarded", "for=203.0.113.1;proto=https")
+
+	handler := interceptor.WrapUnary(func(ctx context.Context, _ connect.AnyRequest) (connect.AnyResponse, error) {
+		md, ok := metadata.FromIncomingContext(ctx)
+		if !ok {
+			t.Fatal("expected metadata in context")
+		}
+		if got := md.Get("origin"); len(got) != 1 || got[0] != "https://memos.example" {
+			t.Fatalf("unexpected origin metadata: %v", got)
+		}
+		if got := md.Get("x-forwarded-proto"); len(got) != 1 || got[0] != "https" {
+			t.Fatalf("unexpected x-forwarded-proto metadata: %v", got)
+		}
+		if got := md.Get("forwarded"); len(got) != 1 || got[0] != "for=203.0.113.1;proto=https" {
+			t.Fatalf("unexpected forwarded metadata: %v", got)
+		}
+		return connect.NewResponse(&emptypb.Empty{}), nil
+	})
+
+	if _, err := handler(context.Background(), req); err != nil {
+		t.Fatalf("metadata interceptor returned error: %v", err)
+	}
+}