Function TestHandleExtractMetadata

pkg/mcp/tools_test.go:242–274

--------------------------------------------------------------------------- handleExtractMetadata ---------------------------------------------------------------------------

(t *testing.T)

// ---------------------------------------------------------------------------

func TestHandleExtractMetadata(t *testing.T) {
	ctx := context.Background()

	t.Run("valid SQL returns metadata keys", func(t *testing.T) {
		req := makeReq(map[string]any{"sql": "SELECT u.id, o.name FROM users u JOIN orders o ON u.id = o.user_id"})
		res, err := handleExtractMetadata(ctx, req)
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		data := unmarshalResult(t, res)
		for _, key := range []string{"tables", "columns", "functions"} {
			if _, ok := data[key]; !ok {
				t.Errorf("expected key %q in result", key)
			}
		}
	})

	t.Run("empty sql returns protocol error", func(t *testing.T) {
		req := makeReq(map[string]any{"sql": ""})
		_, err := handleExtractMetadata(ctx, req)
		if err == nil {
			t.Fatal("expected error for empty sql, got nil")
		}
	})

	t.Run("missing sql param returns protocol error", func(t *testing.T) {
		req := makeReq(map[string]any{})
		_, err := handleExtractMetadata(ctx, req)
		if err == nil {
			t.Fatal("expected error for missing sql param, got nil")
		}
	})
}

// ---------------------------------------------------------------------------
// handleSecurityScan