fix locator

grafana · Jan 27, 2025 · db03dba · db03dba
1 parent bc43151
commit db03dba
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/packages/plugin-e2e/src/models/components/DataSourcePicker.ts b/packages/plugin-e2e/src/models/components/DataSourcePicker.ts
@@ -15,7 +15,9 @@ export class DataSourcePicker extends GrafanaPage {
    * Sets the data source picker to the provided name
    */
   async set(name: string) {
-    let datasourcePicker = this.ctx.page.getByTestId(this.ctx.selectors.components.DataSourcePicker.inputV2);
+    let datasourcePicker = (this.root || this.ctx.page).getByTestId(
+      this.ctx.selectors.components.DataSourcePicker.inputV2
+    );
 
     if (semver.lt(this.ctx.grafanaVersion, '10.1.0')) {
       datasourcePicker = this.getByGrafanaSelector(this.ctx.selectors.components.DataSourcePicker.container, {