#466@trivial: Adds unit tests for waitForNavigation().

capricorn86 · Jan 10, 2024 · c6ca530 · c6ca530
1 parent f245b8b
commit c6ca530
Show file tree

Hide file tree

Showing 4 changed files with 106 additions and 0 deletions.
diff --git a/packages/happy-dom/test/browser/BrowserFrame.test.ts b/packages/happy-dom/test/browser/BrowserFrame.test.ts
@@ -12,6 +12,7 @@ import BrowserNavigationCrossOriginPolicyEnum from '../../src/browser/enums/Brow
 import BrowserFrameFactory from '../../src/browser/utilities/BrowserFrameFactory';
 import BrowserErrorCaptureEnum from '../../src/browser/enums/BrowserErrorCaptureEnum';
 import Headers from '../../src/fetch/Headers';
+import IHTMLAnchorElement from '../../src/nodes/html-anchor-element/IHTMLAnchorElement';
 
 describe('BrowserFrame', () => {
 	afterEach(() => {
@@ -135,6 +136,41 @@ describe('BrowserFrame', () => {
 		});
 	});
 
+	describe('waitForNavigation()', () => {
+		it('Waits page to have been navigated.', async () => {
+			let count = 0;
+			vi.spyOn(Fetch.prototype, 'send').mockImplementation(function (): Promise<IResponse> {
+				count++;
+				return Promise.resolve(<IResponse>{
+					text: () =>
+						new Promise((resolve) =>
+							setTimeout(
+								() =>
+									resolve(
+										count === 1 ? '<a href="http://localhost:3000/navigated/">' : '<b>Navigated</b>'
+									),
+								1
+							)
+						)
+				});
+			});
+
+			const browser = new Browser();
+			const page = browser.newPage();
+
+			await page.mainFrame.goto('http://localhost:3000', {
+				referrer: 'http://localhost:3000/referrer',
+				referrerPolicy: 'no-referrer-when-downgrade'
+			});
+
+			(<IHTMLAnchorElement>page.mainFrame.document.querySelector('a')).click();
+
+			await page.mainFrame.waitForNavigation();
+
+			expect(page.mainFrame.document.querySelector('b')?.textContent).toBe('Navigated');
+		});
+	});
+
 	describe('abort()', () => {
 		it('Aborts all ongoing operations.', async () => {
 			const browser = new Browser();

diff --git a/packages/happy-dom/test/browser/BrowserPage.test.ts b/packages/happy-dom/test/browser/BrowserPage.test.ts
@@ -137,6 +137,23 @@ describe('BrowserPage', () => {
 		});
 	});
 
+	describe('waitForNavigation()', () => {
+		it('Waits page to have been navigated.', async () => {
+			const browser = new Browser();
+			const page = browser.newPage();
+			let isCalled = false;
+
+			vi.spyOn(page.mainFrame, 'waitForNavigation').mockImplementation((): Promise<void> => {
+				isCalled = true;
+				return Promise.resolve();
+			});
+
+			await page.waitForNavigation();
+
+			expect(isCalled).toBe(true);
+		});
+	});
+
 	describe('abort()', () => {
 		it('Aborts all ongoing operations.', async () => {
 			const browser = new Browser();

diff --git a/packages/happy-dom/test/browser/detached-browser/DetachedBrowserFrame.test.ts b/packages/happy-dom/test/browser/detached-browser/DetachedBrowserFrame.test.ts
@@ -13,6 +13,7 @@ import BrowserNavigationCrossOriginPolicyEnum from '../../../src/browser/enums/B
 import BrowserFrameFactory from '../../../src/browser/utilities/BrowserFrameFactory';
 import BrowserErrorCaptureEnum from '../../../src/browser/enums/BrowserErrorCaptureEnum';
 import Headers from '../../../src/fetch/Headers';
+import IHTMLAnchorElement from '../../../src/nodes/html-anchor-element/IHTMLAnchorElement';
 
 describe('DetachedBrowserFrame', () => {
 	afterEach(() => {
@@ -145,6 +146,41 @@ describe('DetachedBrowserFrame', () => {
 		});
 	});
 
+	describe('waitForNavigation()', () => {
+		it('Waits page to have been navigated.', async () => {
+			let count = 0;
+			vi.spyOn(Fetch.prototype, 'send').mockImplementation(function (): Promise<IResponse> {
+				count++;
+				return Promise.resolve(<IResponse>{
+					text: () =>
+						new Promise((resolve) =>
+							setTimeout(
+								() =>
+									resolve(
+										count === 1 ? '<a href="http://localhost:3000/navigated/">' : '<b>Navigated</b>'
+									),
+								1
+							)
+						)
+				});
+			});
+
+			const browser = new DetachedBrowser(BrowserWindow);
+			const page = browser.newPage();
+
+			await page.mainFrame.goto('http://localhost:3000', {
+				referrer: 'http://localhost:3000/referrer',
+				referrerPolicy: 'no-referrer-when-downgrade'
+			});
+
+			(<IHTMLAnchorElement>page.mainFrame.document.querySelector('a')).click();
+
+			await page.mainFrame.waitForNavigation();
+
+			expect(page.mainFrame.document.querySelector('b')?.textContent).toBe('Navigated');
+		});
+	});
+
 	describe('abort()', () => {
 		it('Aborts all ongoing operations.', async () => {
 			const browser = new DetachedBrowser(BrowserWindow);

diff --git a/packages/happy-dom/test/browser/detached-browser/DetachedBrowserPage.test.ts b/packages/happy-dom/test/browser/detached-browser/DetachedBrowserPage.test.ts
@@ -147,6 +147,23 @@ describe('DetachedBrowserPage', () => {
 		});
 	});
 
+	describe('waitForNavigation()', () => {
+		it('Waits page to have been navigated.', async () => {
+			const browser = new DetachedBrowser(BrowserWindow);
+			const page = browser.newPage();
+			let isCalled = false;
+
+			vi.spyOn(page.mainFrame, 'waitForNavigation').mockImplementation((): Promise<void> => {
+				isCalled = true;
+				return Promise.resolve();
+			});
+
+			await page.waitForNavigation();
+
+			expect(isCalled).toBe(true);
+		});
+	});
+
 	describe('abort()', () => {
 		it('Aborts all ongoing operations.', async () => {
 			const browser = new DetachedBrowser(BrowserWindow);