fix go routines leakage (#726)

* fix go routines leakage * Update manager.go fmt Co-authored-by: Tim Voronov <ziflex@users.noreply.github.com>
MontFerret · Feb 10, 2022 · 70a8ef7 · 70a8ef7
1 parent 0c6d4e6
commit 70a8ef7
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pkg/drivers/cdp/network/manager.go b/pkg/drivers/cdp/network/manager.go
@@ -494,17 +494,16 @@ func (m *Manager) NavigateBack(ctx context.Context, skip values.Int) (values.Boo
 }
 
 func (m *Manager) WaitForNavigation(ctx context.Context, opts WaitEventOptions) error {
-	ctx, cancel := context.WithCancel(ctx)
-	defer cancel()
-
 	stream, err := m.OnNavigation(ctx)
-
 	if err != nil {
 		return err
 	}
 
 	defer stream.Close(ctx)
 
+	ctx, cancel := context.WithCancel(ctx)
+	defer cancel()
+
 	for evt := range stream.Read(ctx) {
 		if err := ctx.Err(); err != nil {
 			return err