Add time added route object model

4 files changed, 34 insertions, 0 deletions

diff --git a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/routes-object-model.ts b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/routes-object-model.ts
index d571eb3c53..3b629bebef 100644
--- a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/routes-object-model.ts
+++ b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/routes-object-model.ts
@@ -14,6 +14,7 @@ import { SelectLocationRouteObjectModel } from './select-location';
 import { SettingsRouteObjectModel } from './settings/settings-route-object-model';
 import { ShadowsocksSettingsRouteObjectModel } from './shadowsocks-settings';
 import { SplitTunnelingSettingsRouteObjectModel } from './split-tunneling-settings';
+import { TimeAddedRouteObjectModel } from './time-added';
 import { UdpOverTcpSettingsRouteObjectModel } from './udp-over-tcp-settings';
 import { UserInterfaceSettingsRouteObjectModel } from './user-interface-settings';
 import { VoucherSuccessRouteObjectModel } from './voucher-success';
@@ -27,6 +28,7 @@ export class RoutesObjectModel {
   readonly expired: ExpiredRouteObjectModel;
   readonly redeemVoucher: RedeemVoucherRouteObjectModel;
   readonly voucherSuccess: VoucherSuccessRouteObjectModel;
+  readonly timeAdded: TimeAddedRouteObjectModel;
   readonly settings: SettingsRouteObjectModel;
   readonly userInterfaceSettings: UserInterfaceSettingsRouteObjectModel;
   readonly selectLanguage: SelectLanguageRouteObjectModel;
@@ -48,6 +50,7 @@ export class RoutesObjectModel {
     this.expired = new ExpiredRouteObjectModel(page, utils);
     this.redeemVoucher = new RedeemVoucherRouteObjectModel(page, utils);
     this.voucherSuccess = new VoucherSuccessRouteObjectModel(page, utils);
+    this.timeAdded = new TimeAddedRouteObjectModel(page, utils);
     this.settings = new SettingsRouteObjectModel(page, utils);
     this.userInterfaceSettings = new UserInterfaceSettingsRouteObjectModel(page, utils);
     this.filter = new FilterRouteObjectModel(page, utils);
diff --git a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/index.ts b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/index.ts
new file mode 100644
index 0000000000..2d4afd7d8f
--- /dev/null
+++ b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/index.ts
@@ -0,0 +1,2 @@
+export * from './time-added-route-object-model';
+export * from './selectors';
diff --git a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/selectors.ts b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/selectors.ts
new file mode 100644
index 0000000000..6394901191
--- /dev/null
+++ b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/selectors.ts
@@ -0,0 +1,5 @@
+import { Page } from 'playwright';
+
+export const createSelectors = (page: Page) => ({
+  nextButton: () => page.getByRole('button', { name: 'Next' }),
+});
diff --git a/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/time-added-route-object-model.ts b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/time-added-route-object-model.ts
new file mode 100644
index 0000000000..5bad1862db
--- /dev/null
+++ b/desktop/packages/mullvad-vpn/test/e2e/route-object-models/time-added/time-added-route-object-model.ts
@@ -0,0 +1,24 @@
+import { Page } from 'playwright';
+
+import { RoutePath } from '../../../../src/shared/routes';
+import { TestUtils } from '../../utils';
+import { createSelectors } from './selectors';
+
+export class TimeAddedRouteObjectModel {
+  readonly selectors: ReturnType<typeof createSelectors>;
+
+  constructor(
+    private readonly page: Page,
+    private readonly utils: TestUtils,
+  ) {
+    this.selectors = createSelectors(this.page);
+  }
+
+  async waitForRoute() {
+    await this.utils.expectRoute(RoutePath.timeAdded);
+  }
+
+  async gotoNext() {
+    await this.selectors.nextButton().click();
+  }
+}