@@ -20,7 +20,6 @@ import {
2020waitForLoaderToBeRemoved ,
2121} from "testHelpers/renderHelpers" ;
2222import { createMockWebSocket } from "testHelpers/websockets" ;
23- import type { MockWebSocketServer } from "testHelpers/websockets" ;
2423import CreateWorkspacePageExperimental from "./CreateWorkspacePageExperimental" ;
2524
2625describe ( "CreateWorkspacePageExperimental" , ( ) => {
@@ -210,21 +209,16 @@ describe("CreateWorkspacePageExperimental", () => {
210209} ) ;
211210
212211it ( "only parameters from latest response are displayed" , async ( ) => {
213- let publisher :MockWebSocketServer ;
214-
212+ const [ mockWebSocket , mockPublisher ] = createMockWebSocket ( "ws://test" ) ;
215213jest
216214. spyOn ( API , "templateVersionDynamicParameters" )
217215. mockImplementation ( ( _versionId , _ownerId , callbacks ) => {
218- const [ mockWebSocket , mockPublisher ] =
219- createMockWebSocket ( "ws://test" ) ;
220- publisher = mockPublisher ;
221-
222216mockWebSocket . addEventListener ( "message" , ( event ) => {
223217callbacks . onMessage ( JSON . parse ( event . data ) ) ;
224218} ) ;
225219
226- publisher . publishOpen ( new Event ( "open" ) ) ;
227- publisher . publishMessage (
220+ mockPublisher . publishOpen ( new Event ( "open" ) ) ;
221+ mockPublisher . publishMessage (
228222new MessageEvent ( "message" , {
229223data :JSON . stringify ( {
230224id :0 ,
@@ -252,11 +246,11 @@ describe("CreateWorkspacePageExperimental", () => {
252246} ;
253247
254248await waitFor ( ( ) => {
255- publisher . publishMessage (
249+ mockPublisher . publishMessage (
256250new MessageEvent ( "message" , { data :JSON . stringify ( response1 ) } ) ,
257251) ;
258252
259- publisher . publishMessage (
253+ mockPublisher . publishMessage (
260254new MessageEvent ( "message" , { data :JSON . stringify ( response2 ) } ) ,
261255) ;
262256} ) ;