opencode/yao
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #1327 from trheyi/main

Browse source 
Refactor test files for improved readability and consistency
This commit is contained in:
Max 2025-11-18 19:41:40 +08:00 • committed by GitHub
commit 85ed9b9a26
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 18 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
trace/trace_concurrent_test.go
View file

@ -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"}},

18
trace/trace_node_test.go
View file

@ -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"},