@@ -38,7 +38,9 @@ test('Render works correctly', async ({ page, sharedCodeServer, tempDir }) => {
3838
3939 // Check if the model is rendered by check if "`oi`.`order_id` AS `order_id`," is in the window
4040 await expect ( page . locator ( 'text="marketing"."customer_id" AS' ) ) . toBeVisible ( )
41- await expect ( page . locator ( 'text=sushi.customers (rendered)' ) ) . toBeVisible ( )
41+ await expect (
42+ page . getByRole ( 'tab' , { name : 'sushi.customers (rendered)' } ) ,
43+ ) . toBeVisible ( )
4244} )
4345
4446test ( 'Render works correctly with model without a description' , async ( {
@@ -74,7 +76,9 @@ test('Render works correctly with model without a description', async ({
7476
7577 // Check if the model is rendered correctly
7678 await expect ( page . locator ( 'text="orders"."id" AS "id",' ) ) . toBeVisible ( )
77- await expect ( page . locator ( 'text=sushi.latest_order (rendered)' ) ) . toBeVisible ( )
79+ await expect (
80+ page . getByRole ( 'tab' , { name : 'sushi.latest_order (rendered)' } ) ,
81+ ) . toBeVisible ( )
7882} )
7983
8084test ( 'Render works correctly with every rendered model opening a new tab' , async ( {
@@ -104,7 +108,9 @@ test('Render works correctly with every rendered model opening a new tab', async
104108 await runCommand ( page , 'Render Model' )
105109
106110 // Check if the model is rendered correctly
107- await expect ( page . locator ( 'text=sushi.latest_order (rendered)' ) ) . toBeVisible ( )
111+ await expect (
112+ page . getByRole ( 'tab' , { name : 'sushi.latest_order (rendered)' } ) ,
113+ ) . toBeVisible ( )
108114
109115 // Open the customers model
110116 await page
@@ -117,8 +123,12 @@ test('Render works correctly with every rendered model opening a new tab', async
117123 await runCommand ( page , 'Render Model' )
118124
119125 // Assert both tabs exist
120- await expect ( page . locator ( 'text=sushi.latest_order (rendered)' ) ) . toBeVisible ( )
121- await expect ( page . locator ( 'text=sushi.customers (rendered)' ) ) . toBeVisible ( )
126+ await expect (
127+ page . getByRole ( 'tab' , { name : 'sushi.latest_order (rendered)' } ) ,
128+ ) . toBeVisible ( )
129+ await expect (
130+ page . getByRole ( 'tab' , { name : 'sushi.customers (rendered)' } ) ,
131+ ) . toBeVisible ( )
122132} )
123133
124134test ( 'Render shows model picker when no active editor is open' , async ( {
@@ -146,7 +156,9 @@ test('Render shows model picker when no active editor is open', async ({
146156 await page . locator ( 'text=sushi.customers' ) . click ( )
147157
148158 // Verify the rendered model is shown
149- await expect ( page . locator ( 'text=sushi.customers (rendered)' ) ) . toBeVisible ( {
159+ await expect (
160+ page . getByRole ( 'tab' , { name : 'sushi.customers (rendered)' } ) ,
161+ ) . toBeVisible ( {
150162 timeout : 2_000 ,
151163 } )
152164} )
0 commit comments