Add test to ensure search input is focused on load in select location

3 files changed, 10 insertions, 0 deletions

diff --git a/desktop/packages/mullvad-vpn/test/e2e/mocked/select-location/select-location.spec.ts b/desktop/packages/mullvad-vpn/test/e2e/mocked/select-location/select-location.spec.ts
index 02a1a403bc..e15c73efa4 100644
--- a/desktop/packages/mullvad-vpn/test/e2e/mocked/select-location/select-location.spec.ts
+++ b/desktop/packages/mullvad-vpn/test/e2e/mocked/select-location/select-location.spec.ts
@@ -36,6 +36,11 @@ test.describe('Select location', () => {
     await page.close();
   });
 
+  test('Should focus search input on load', async () => {
+    const input = routes.selectLocation.getSearchInput();
+    await expect(input).toBeFocused();
+  });
+
   test.describe('Multihop enabled', () => {
     test.beforeAll(async () => {
       await helpers.updateMockSettings({
diff --git a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/select-location/select-location-route-object-model.ts b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/select-location/select-location-route-object-model.ts
index 7d29b03700..a9881d3a24 100644
--- a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/select-location/select-location-route-object-model.ts
+++ b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/select-location/select-location-route-object-model.ts
@@ -28,6 +28,10 @@ export class SelectLocationRouteObjectModel {
     return this.selectors.exitButton();
   }
 
+  getSearchInput() {
+    return this.selectors.searchInput();
+  }
+
   getRelaysMatching(relayNames: string[]) {
     return this.selectors.relaysMatching(relayNames);
   }
diff --git a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/select-location/selectors.ts b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/select-location/selectors.ts
index 689401ad63..fa52e2a184 100644
--- a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/select-location/selectors.ts
+++ b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/select-location/selectors.ts
@@ -10,4 +10,5 @@ export const createSelectors = (page: Page) => ({
   expandAccordionButton: (label: string) => page.getByLabel(`Expand ${label}`),
   relaysMatching: (relayNames: string[]) =>
     page.getByRole('button', { name: new RegExp(relayNames.join('|')) }),
+  searchInput: () => page.getByPlaceholder('Search for...'),
 });