Changed the location of the rewrite

Author: Cristian101

src/Magento/FunctionalTestingFramework/Module/MagentoWebDriver.php

@@ -599,4 +599,17 @@ public function saveScreenshot()
         $this->_saveScreenshot($this->pngReport = $outputDir . mb_strcut($filename, 0, 245, 'utf-8') . '.fail.png');
         $this->_savePageSource($this->htmlReport = $outputDir . mb_strcut($filename, 0, 244, 'utf-8') . '.fail.html');
     }
+
+    /**
+     * Go to a page and wait for ajax requests to finish
+     *
+     * @param string $page
+     * @throws \Exception
+     * @return void
+     */
+    public function amOnPage($page)
+    {
+        parent::amOnPage($page)
+        $this->waitForPageLoad();
+    }
 }

src/Magento/FunctionalTestingFramework/Util/TestGenerator.php

@@ -1898,19 +1898,4 @@ private function printRuleErrorToConsole($key, $tagName, $attributes)
         $message .= '" can be use for action "' . $tagName . "\".\n";
         print $message;
     }
-
-    /**
-     * Go to a page and wait for ajax requests to finish
-     *
-     * @param string $page
-     * @throws \Exception
-     * @return void
-     */
-    public function amOnPage($page)
-    {
-        $url = Uri::appendPath($this->config['url'], $page);
-        $this->debugSection('GET', $url);
-        $this->webDriver->get($url);
-        $this->waitForPageLoad();
-    }
 }