From 70a8ef790bfe08c72338e9427389d064019e4d4a Mon Sep 17 00:00:00 2001 From: Roman Date: Thu, 10 Feb 2022 23:42:07 +0300 Subject: [PATCH] fix go routines leakage (#726) * fix go routines leakage * Update manager.go fmt Co-authored-by: Tim Voronov --- pkg/drivers/cdp/network/manager.go | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/pkg/drivers/cdp/network/manager.go b/pkg/drivers/cdp/network/manager.go index 2caaf222..14d07130 100644 --- 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