Add traces assertion in browser registry test

grafana · Nov 14, 2023 · c3b56ca · c3b56ca
1 parent d5d78bf
commit c3b56ca
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/browser/registry_test.go b/browser/registry_test.go
@@ -218,6 +218,11 @@ func TestBrowserRegistry(t *testing.T) {
 		assert.Equal(t, 3, len(browserRegistry.m))
 		browserRegistry.mu.RUnlock()
 
+		// Verify iteration traces are started
+		browserRegistry.tr.mu.Lock()
+		assert.Equal(t, 3, len(browserRegistry.tr.m))
+		browserRegistry.tr.mu.Unlock()
+
 		// Send IterEnd events
 		vu.EndIteration(t, k6test.WithIteration(0))
 		vu.EndIteration(t, k6test.WithIteration(1))
@@ -227,6 +232,11 @@ func TestBrowserRegistry(t *testing.T) {
 		browserRegistry.mu.RLock()
 		assert.Equal(t, 0, len(browserRegistry.m))
 		browserRegistry.mu.RUnlock()
+
+		// Verify iteration traces have been ended
+		browserRegistry.mu.RLock()
+		assert.Equal(t, 0, len(browserRegistry.m))
+		browserRegistry.mu.RUnlock()
 	})
 
 	t.Run("close_browsers_on_exit_event", func(t *testing.T) {