fix(tests): flush spans before checking

Signed-off-by: Valery Piashchynski <piashchynski.valery@gmail.com>
roadrunner-server · Jul 8, 2024 · 28f0d02 · 28f0d02
1 parent b0a5ffd
commit 28f0d02
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/grpc_plugin_test.go b/tests/grpc_plugin_test.go
@@ -184,6 +184,9 @@ func TestGrpcOtel(t *testing.T) {
 	require.NoError(t, err)
 	require.Equal(t, "TOST", resp.Msg)
 
+	stopCh <- struct{}{}
+	wg.Wait()
+
 	time.Sleep(time.Second * 3)
 
 	req2, err := http.NewRequestWithContext(context.Background(), http.MethodGet, "http://127.0.0.1:9411/api/v2/spans?serviceName=rr_test_grpc", nil)
@@ -199,9 +202,6 @@ func TestGrpcOtel(t *testing.T) {
 	// contains spans
 	assert.Contains(t, string(bd), "service.echo/ping")
 	_ = resp2.Body.Close()
-
-	stopCh <- struct{}{}
-	wg.Wait()
 }
 
 func TestGrpcCheckStatus(t *testing.T) {