feat(rpc): 完成 issue #1 Go 调用边界

2026-09-18 08:14:55 +08:00 · 2026-07-24 17:58:34 +08:00
parent ecda08ed97
commit 102b49b666
20 changed files with 1011 additions and 34 deletions
@@ -0,0 +1,207 @@
+package backendrpc
+
+import (
+	"bufio"
+	"context"
+	"encoding/json"
+	"errors"
+	"net"
+	"testing"
+	"time"
+)
+
+type testRequest struct {
+	JSONRPC string          `json:"jsonrpc"`
+	ID      json.RawMessage `json:"id"`
+	Method  string          `json:"method"`
+	Params  json.RawMessage `json:"params,omitempty"`
+}
+
+type testEnvelope struct {
+	OK        bool   `json:"ok"`
+	Status    string `json:"status"`
+	Data      any    `json:"data,omitempty"`
+	Error     any    `json:"error,omitempty"`
+	RequestID string `json:"request_id"`
+}
+
+type testResponse struct {
+	JSONRPC string          `json:"jsonrpc"`
+	ID      json.RawMessage `json:"id"`
+	Result  any             `json:"result,omitempty"`
+	Error   *JSONRPCError   `json:"error,omitempty"`
+}
+
+func newTestClient(t *testing.T, handler func(t *testing.T, req testRequest) testResponse) *Client {
+	t.Helper()
+	client := New("bat.sock")
+	client.DialContext = func(ctx context.Context, network string, address string) (net.Conn, error) {
+		clientConn, serverConn := net.Pipe()
+		go func(conn net.Conn) {
+			defer conn.Close()
+			line, err := bufio.NewReader(conn).ReadBytes('\n')
+			if err != nil {
+				return
+			}
+			var req testRequest
+			if err := json.Unmarshal(line, &req); err != nil {
+				return
+			}
+			resp := handler(t, req)
+			if len(resp.ID) == 0 {
+				resp.ID = req.ID
+			}
+			if resp.JSONRPC == "" {
+				resp.JSONRPC = jsonRPCVersion
+			}
+			_ = json.NewEncoder(conn).Encode(resp)
+		}(serverConn)
+		return clientConn, nil
+	}
+	return client
+}
+
+func TestResourceRepairQueuesTask(t *testing.T) {
+	client := newTestClient(t, func(t *testing.T, req testRequest) testResponse {
+		if req.Method != "resource.repair" {
+			t.Fatalf("method = %s", req.Method)
+		}
+		return testResponse{
+			Result: testEnvelope{
+				OK:        true,
+				Status:    "accepted",
+				RequestID: "req-test-1",
+				Data: map[string]any{
+					"task_id": "task-1",
+					"kind":    "resource.repair",
+				},
+			},
+		}
+	})
+
+	task, err := client.ResourceRepair(context.Background())
+	if err != nil {
+		t.Fatalf("ResourceRepair error: %v", err)
+	}
+	if task.TaskID != "task-1" || task.Kind != "resource.repair" {
+		t.Fatalf("unexpected task: %#v", task)
+	}
+}
+
+func TestResourceListSendsPagination(t *testing.T) {
+	client := newTestClient(t, func(t *testing.T, req testRequest) testResponse {
+		if req.Method != "resource.list" {
+			t.Fatalf("method = %s", req.Method)
+		}
+		var params pageParam
+		if err := json.Unmarshal(req.Params, &params); err != nil {
+			t.Fatalf("decode params: %v", err)
+		}
+		if params.Offset != 2 || params.Limit != 1 {
+			t.Fatalf("params = %#v", params)
+		}
+		bytes := uint64(42)
+		return testResponse{
+			Result: testEnvelope{
+				OK:        true,
+				Status:    "ok",
+				RequestID: "req-test-2",
+				Data: map[string]any{
+					"available":        true,
+					"resource_root":    "/tmp/resources/current",
+					"manifest_version": 1,
+					"total_entries":    3,
+					"offset":           2,
+					"limit":            1,
+					"entries": []map[string]any{
+						{
+							"url":         "https://example.invalid/a.zip",
+							"destination": "a.zip",
+							"bytes":       bytes,
+							"blake3":      "abc",
+						},
+					},
+				},
+			},
+		}
+	})
+
+	page, err := client.ResourceList(context.Background(), 2, 1)
+	if err != nil {
+		t.Fatalf("ResourceList error: %v", err)
+	}
+	if !page.Available || page.TotalEntries != 3 || len(page.Entries) != 1 {
+		t.Fatalf("unexpected page: %#v", page)
+	}
+	if page.Entries[0].Bytes == nil || *page.Entries[0].Bytes != 42 {
+		t.Fatalf("unexpected entry bytes: %#v", page.Entries[0].Bytes)
+	}
+}
+
+func TestApplicationErrorReturnsAPIError(t *testing.T) {
+	client := newTestClient(t, func(t *testing.T, req testRequest) testResponse {
+		if req.Method != "task.status" {
+			t.Fatalf("method = %s", req.Method)
+		}
+		return testResponse{
+			Result: testEnvelope{
+				OK:     false,
+				Status: "error",
+				Error: APIError{
+					Code:      "BAT-ERR-700004",
+					Kind:      "task_not_found",
+					Domain:    "rpc",
+					Location:  "task.status",
+					Message:   "任务不存在：missing",
+					Retryable: false,
+				},
+				RequestID: "req-test-3",
+			},
+		}
+	})
+
+	_, err := client.TaskStatus(context.Background(), "missing")
+	var apiErr *APIError
+	if !errors.As(err, &apiErr) {
+		t.Fatalf("expected APIError, got %T %v", err, err)
+	}
+	if apiErr.Code != "BAT-ERR-700004" {
+		t.Fatalf("code = %s", apiErr.Code)
+	}
+}
+
+func TestTransportErrorReturnsJSONRPCError(t *testing.T) {
+	client := newTestClient(t, func(t *testing.T, req testRequest) testResponse {
+		return testResponse{
+			Error: &JSONRPCError{
+				Code:    -32700,
+				Message: "parse error",
+			},
+		}
+	})
+
+	_, err := client.DaemonStatus(context.Background())
+	var rpcErr *JSONRPCError
+	if !errors.As(err, &rpcErr) {
+		t.Fatalf("expected JSONRPCError, got %T %v", err, err)
+	}
+	if rpcErr.Code != -32700 {
+		t.Fatalf("code = %d", rpcErr.Code)
+	}
+}
+
+func TestContextDeadlineIsApplied(t *testing.T) {
+	client := New("bat.sock")
+	client.Timeout = time.Millisecond
+	client.DialContext = func(ctx context.Context, network string, address string) (net.Conn, error) {
+		<-ctx.Done()
+		return nil, ctx.Err()
+	}
+	ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)
+	defer cancel()
+
+	_, err := client.DaemonStatus(ctx)
+	if err == nil {
+		t.Fatal("expected dial error")
+	}
+}