test(e2e): replace waitForVolumeLoad with viewport render waits (#6000)

tests/ArrowAnnotate.spec.ts

@@ -1,4 +1,10 @@
-import { checkForScreenshot, screenShotPaths, test, visitStudy } from './utils';
+import {
+  checkForScreenshot,
+  screenShotPaths,
+  test,
+  visitStudy,
+  waitForViewportRenderCycle,
+} from './utils';
 
 test.beforeEach(async ({ page }) => {
   const studyInstanceUID = '1.3.6.1.4.1.25403.345050719074.3824.20170125095438.5';
@@ -27,9 +33,11 @@ test('should display the arrow tool and allow free-form text to be entered', asy
     'Ringo Starr was the drummer for The Beatles'
   );
 
+  const viewportRenderCycle = waitForViewportRenderCycle(page);
+
   await DOMOverlayPageObject.viewport.measurementTracking.confirm.click();
 
-  await mainToolbarPageObject.waitForVolumeLoad();
+  await viewportRenderCycle;
 
   await checkForScreenshot({
     page,
@@ -43,8 +51,6 @@ test('should display the arrow tool and allow free-form text to be entered', asy
 
   await DOMOverlayPageObject.dialog.input.fillAndSave('Neil Peart was the drummer for Rush');
 
-  await mainToolbarPageObject.waitForVolumeLoad();
-
   await checkForScreenshot({
     page,
     maxDiffPixelRatio: 0.0075,
@@ -57,8 +63,6 @@ test('should display the arrow tool and allow free-form text to be entered', asy
     .nthMeasurement(0)
     .actions.rename('Drummer annotation arrow');
 
-  await mainToolbarPageObject.waitForVolumeLoad();
-
   await checkForScreenshot({
     page,
     maxDiffPixelRatio: 0.0075,