From 11b126cb66dc4f1155de34db0505379ff91bdd86 Mon Sep 17 00:00:00 2001
From: Jason Weill <jweill@amazon.com>
Date: Wed, 1 Dec 2021 16:16:55 -0800
Subject: [PATCH] Updates UI test to expect new title

---
 ui-tests/test/notebook.spec.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/ui-tests/test/notebook.spec.ts b/ui-tests/test/notebook.spec.ts
index 7028b72fe..f99e4edc2 100644
--- a/ui-tests/test/notebook.spec.ts
+++ b/ui-tests/test/notebook.spec.ts
@@ -34,11 +34,12 @@ test.describe('Notebook', () => {
     const notebook = `${tmpPath}/${NOTEBOOK}`;
     await page.goto(`notebooks/${notebook}`);
 
-    // Click on the title
-    await page.click('text="example.ipynb"');
+    // Click on the title (with .ipynb extension stripped)
+    await page.click('text="example"');
 
     // Rename in the input dialog
     const newName = 'test.ipynb';
+    const newNameStripped = 'test';
     await page.fill(
       `//div[normalize-space(.)='File Path${notebook}New Name']/input`,
       newName
@@ -52,6 +53,6 @@ test.describe('Notebook', () => {
 
     // Check the URL contains the new name
     const url = page.url();
-    expect(url).toContain(newName);
+    expect(url).toContain(newNameStripped);
   });
 });