Test state RunOnEventLoop

grafana · Sep 11, 2023 · 390493d · 390493d
1 parent d32a7db
commit 390493d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/grpc/stream_test.go b/grpc/stream_test.go
@@ -82,9 +82,7 @@ func TestStream_RequestHeaders(t *testing.T) {
 
 	ts.ToVUContext()
 
-	val, err = ts.Run(vuString.code)
-
-	ts.EventLoop.WaitOnRegistered()
+	val, err = ts.RunOnEventLoop(vuString.code)
 
 	assertResponse(t, vuString, err, val, ts)
 
@@ -162,9 +160,7 @@ func TestStream_ErrorHandling(t *testing.T) {
 
 	ts.ToVUContext()
 
-	val, err = ts.Run(vuString.code)
-
-	ts.EventLoop.WaitOnRegistered()
+	val, err = ts.RunOnEventLoop(vuString.code)
 
 	assertResponse(t, vuString, err, val, ts)
 
@@ -255,9 +251,7 @@ func TestStream_ReceiveAllServerResponsesAfterEnd(t *testing.T) {
 
 	ts.ToVUContext()
 
-	val, err = ts.Run(vuString.code)
-
-	ts.EventLoop.WaitOnRegistered()
+	val, err = ts.RunOnEventLoop(vuString.code)
 
 	assertResponse(t, vuString, err, val, ts)
 

diff --git a/grpc/teststate_test.go b/grpc/teststate_test.go
@@ -86,6 +86,11 @@ func (ts *testState) Run(code string) (goja.Value, error) {
 	return ts.VU.Runtime().RunString(ts.httpBin.Replacer.Replace(code))
 }
 
+// RunOnEventLoop replaces the httpbin address and run the code on event loop
+func (ts *testState) RunOnEventLoop(code string) (goja.Value, error) {
+	return ts.Runtime.RunOnEventLoop(ts.httpBin.Replacer.Replace(code))
+}
+
 // newTestState creates a new test state.
 func newTestState(t *testing.T) testState {
 	t.Helper()