From 251b0591a221a9806c4ba49e66ecc565ba255b06 Mon Sep 17 00:00:00 2001 From: Max Date: Tue, 18 Nov 2025 19:17:27 +0800 Subject: [PATCH] Refactor test files for improved readability and consistency - Adjusted indentation and formatting in `trace_concurrent_test.go` and `trace_node_test.go` to enhance code clarity. - Ensured consistent structure in test cases by aligning the setup and assertions, improving overall maintainability. --- trace/trace_concurrent_test.go | 18 +++++++++--------- trace/trace_node_test.go | 18 +++++++++--------- 2 files changed, 18 insertions(+), 18 deletions(-) diff --git a/trace/trace_concurrent_test.go b/trace/trace_concurrent_test.go index 0cdb198f..97e3c994 100644 --- a/trace/trace_concurrent_test.go +++ b/trace/trace_concurrent_test.go @@ -18,17 +18,17 @@ func TestConcurrentNodeOperations(t *testing.T) { t.Run(d.Name, func(t *testing.T) { ctx := context.Background() - traceID, manager, err := trace.New(ctx, d.DriverType, nil, d.DriverOptions...) - assert.NoError(t, err) - defer trace.Release(traceID) - defer trace.Remove(ctx, d.DriverType, traceID, d.DriverOptions...) + traceID, manager, err := trace.New(ctx, d.DriverType, nil, d.DriverOptions...) + assert.NoError(t, err) + defer trace.Release(traceID) + defer trace.Remove(ctx, d.DriverType, traceID, d.DriverOptions...) - // Add first node as root - _, err = manager.Add("root", types.TraceNodeOption{Label: "Root"}) - assert.NoError(t, err) + // Add first node as root + _, err = manager.Add("root", types.TraceNodeOption{Label: "Root"}) + assert.NoError(t, err) - // Create parallel nodes - nodes, err := manager.Parallel([]types.TraceParallelInput{ + // Create parallel nodes + nodes, err := manager.Parallel([]types.TraceParallelInput{ {Input: "task 1", Option: types.TraceNodeOption{Label: "Worker 1"}}, {Input: "task 2", Option: types.TraceNodeOption{Label: "Worker 2"}}, {Input: "task 3", Option: types.TraceNodeOption{Label: "Worker 3"}}, diff --git a/trace/trace_node_test.go b/trace/trace_node_test.go index 9d668117..0f8ae569 100644 --- a/trace/trace_node_test.go +++ b/trace/trace_node_test.go @@ -73,17 +73,17 @@ func TestParallelOperations(t *testing.T) { t.Run(d.Name, func(t *testing.T) { ctx := context.Background() - traceID, manager, err := trace.New(ctx, d.DriverType, nil, d.DriverOptions...) - assert.NoError(t, err) - defer trace.Release(traceID) - defer trace.Remove(ctx, d.DriverType, traceID, d.DriverOptions...) + traceID, manager, err := trace.New(ctx, d.DriverType, nil, d.DriverOptions...) + assert.NoError(t, err) + defer trace.Release(traceID) + defer trace.Remove(ctx, d.DriverType, traceID, d.DriverOptions...) - // Add first node as root - _, err = manager.Add("root", types.TraceNodeOption{Label: "Root"}) - assert.NoError(t, err) + // Add first node as root + _, err = manager.Add("root", types.TraceNodeOption{Label: "Root"}) + assert.NoError(t, err) - // Create parallel nodes - nodes, err := manager.Parallel([]types.TraceParallelInput{ + // Create parallel nodes + nodes, err := manager.Parallel([]types.TraceParallelInput{ { Input: "task A", Option: types.TraceNodeOption{Label: "Worker A", Icon: "cpu"},