.

go/internal/xtracing/tracetest/example_test.go

@@ -100,35 +100,34 @@ func setupTraceEvents(t *tracetest.T) {
 
 // TestExample demonstrates how to use tracetest to verify concurrent system with 2 threads.
 func TestExample(t *testing.T) {
-	tracetest.Verify(t, tracetestExample)
-}
-func tracetestExample(t *tracetest.T) {
-	// setup tracing to deliver trace events to t and route them appropriately.
-	setupTraceEvents(t)
-
-	// run the workload
-	var wg sync.WaitGroup
-	defer wg.Wait()
-	wg.Add(2)
-
-	go func() { // thread1
-		defer wg.Done()
-		hi("T1·A")
-		hello("T1·B")
-	}()
-
-	go func() { // thread2
-		defer wg.Done()
-		hi("T2·C")
-	}()
-
-	// assert that events come as expected
-	t.Expect("t2", eventHi("T2·C"))
-	t.Expect("t1", eventHi("T1·A"))
-	t.Expect("t1", eventHello("T1·B"))
-
-	// XXX also t.Recv, t.ExpectNoAck ?
-
-	//t1 := t.OnStream("t1")
-	//t1.Expect(eventHi("X·A"))
+	tracetest.Verify(t, func(t *tracetest.T) {
+		// setup tracing to deliver trace events to t and route them appropriately.
+		setupTraceEvents(t)
+
+		// run the workload
+		var wg sync.WaitGroup
+		defer wg.Wait()
+		wg.Add(2)
+
+		go func() { // thread1
+			defer wg.Done()
+			hi("T1·A")
+			hello("T1·B")
+		}()
+
+		go func() { // thread2
+			defer wg.Done()
+			hi("T2·C")
+		}()
+
+		// assert that events come as expected
+		t.Expect("t2", eventHi("T2·C"))
+		t.Expect("t1", eventHi("T1·A"))
+		t.Expect("t1", eventHello("T1·B"))
+
+		// XXX also t.Recv, t.ExpectNoAck ?
+
+		//t1 := t.OnStream("t1")
+		//t1.Expect(eventHi("X·A"))
+	})
 }