Fix e2e tests

databricks · Dec 5, 2024 · 0160a72 · 0160a72
1 parent 2dd7e3c
commit 0160a72
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/packages/databricks-vscode/src/test/e2e/deploy_and_run_pipeline.e2e.ts b/packages/databricks-vscode/src/test/e2e/deploy_and_run_pipeline.e2e.ts
@@ -119,7 +119,7 @@ describe("Deploy and run pipeline", async function () {
         );
     });
 
-    it("should show expected datasets with schemas for the successful run", async () => {
+    it("should show expected datasets after the successful run", async () => {
         const datasetItems = await getResourceSubItems(
             resourceExplorerView,
             "Pipelines",
@@ -144,7 +144,11 @@ describe("Deploy and run pipeline", async function () {
         assert.strictEqual(datasets[0].description, "materialized view");
         assert.strictEqual(datasets[1].label, "test_view");
         assert.strictEqual(datasets[1].description, "view");
-        const schemaItems = await resourceExplorerView.openItem(
+    });
+
+    it("should show expected schema definitions for a dataset", async () => {
+        const schemaItems = await getResourceSubItems(
+            resourceExplorerView,
             "Pipelines",
             pipelineName,
             "Datasets",

diff --git a/packages/databricks-vscode/src/test/e2e/utils/dabsExplorerUtils.ts b/packages/databricks-vscode/src/test/e2e/utils/dabsExplorerUtils.ts
@@ -36,7 +36,7 @@ export async function getResourceSubItems(
     resourceExplorerView: CustomTreeSection,
     resourceType: "Workflows" | "Pipelines",
     resourceName: string,
-    subItemName: string
+    ...subItemNames: string[]
 ) {
     const resourceViewItem = await getResourceViewItem(
         resourceExplorerView,
@@ -50,7 +50,7 @@ export async function getResourceSubItems(
     return await resourceExplorerView.openItem(
         resourceType,
         await (await resourceViewItem!.elem).getText(),
-        subItemName
+        ...subItemNames
     );
 }