mirror of
https://github.com/Azure/cosmos-explorer.git
synced 2026-01-10 13:08:20 +00:00
add user defined function playwright test
This commit is contained in:
@@ -24,6 +24,7 @@ export async function deleteUserDefinedFunction(databaseId: string, collectionId
|
|||||||
} else {
|
} else {
|
||||||
await client().database(databaseId).container(collectionId).scripts.userDefinedFunction(id).delete();
|
await client().database(databaseId).container(collectionId).scripts.userDefinedFunction(id).delete();
|
||||||
}
|
}
|
||||||
|
logConsoleProgress(`Successfully deleted user defined function ${id}`);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
handleError(error, "DeleteUserDefinedFunction", `Error while deleting user defined function ${id}`);
|
handleError(error, "DeleteUserDefinedFunction", `Error while deleting user defined function ${id}`);
|
||||||
throw error;
|
throw error;
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ import { UserDefinedFunctionDefinition } from "@azure/cosmos";
|
|||||||
import { Label, TextField } from "@fluentui/react";
|
import { Label, TextField } from "@fluentui/react";
|
||||||
import { FluentProvider, webDarkTheme, webLightTheme } from "@fluentui/react-components";
|
import { FluentProvider, webDarkTheme, webLightTheme } from "@fluentui/react-components";
|
||||||
import { KeyboardAction } from "KeyboardShortcuts";
|
import { KeyboardAction } from "KeyboardShortcuts";
|
||||||
|
import { logConsoleInfo } from "Utils/NotificationConsoleUtils";
|
||||||
import { ValidCosmosDbIdDescription, ValidCosmosDbIdInputPattern } from "Utils/ValidationUtils";
|
import { ValidCosmosDbIdDescription, ValidCosmosDbIdInputPattern } from "Utils/ValidationUtils";
|
||||||
import { useThemeStore } from "hooks/useTheme";
|
import { useThemeStore } from "hooks/useTheme";
|
||||||
import React, { Component } from "react";
|
import React, { Component } from "react";
|
||||||
@@ -170,6 +171,7 @@ export default class UserDefinedFunctionTabContent extends Component<
|
|||||||
startKey,
|
startKey,
|
||||||
);
|
);
|
||||||
this.props.editorState(ViewModels.ScriptEditorState.existingNoEdits);
|
this.props.editorState(ViewModels.ScriptEditorState.existingNoEdits);
|
||||||
|
logConsoleInfo(`Sucessfully created user defined function ${createdResource.id}`);
|
||||||
}
|
}
|
||||||
} catch (createError) {
|
} catch (createError) {
|
||||||
this.props.isExecutionError(true);
|
this.props.isExecutionError(true);
|
||||||
|
|||||||
74
test/sql/scripts/userDefinedFunction.spec.ts
Normal file
74
test/sql/scripts/userDefinedFunction.spec.ts
Normal file
@@ -0,0 +1,74 @@
|
|||||||
|
import { expect, test } from "@playwright/test";
|
||||||
|
import { CommandBarButton, DataExplorer, ONE_MINUTE_MS, TestAccount } from "../../fx";
|
||||||
|
import { createTestSQLContainer, TestContainerContext } from "../../testData";
|
||||||
|
|
||||||
|
test.describe("User Defined Functions", () => {
|
||||||
|
let context: TestContainerContext = null!;
|
||||||
|
let explorer: DataExplorer = null!;
|
||||||
|
const udfBody = `function extractDocumentId(doc) {
|
||||||
|
return {
|
||||||
|
id: doc.id
|
||||||
|
};
|
||||||
|
}`;
|
||||||
|
|
||||||
|
test.beforeAll("Create Test Database", async () => {
|
||||||
|
context = await createTestSQLContainer(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
test.beforeEach("Open container", async ({ page }) => {
|
||||||
|
explorer = await DataExplorer.open(page, TestAccount.SQL);
|
||||||
|
});
|
||||||
|
|
||||||
|
test.afterAll("Delete Test Database", async () => {
|
||||||
|
await context?.dispose();
|
||||||
|
});
|
||||||
|
|
||||||
|
test("Add, execute, and delete user defined function", async ({ page }) => {
|
||||||
|
// Open container context menu and click New UDF
|
||||||
|
const containerNode = await explorer.waitForContainerNode(context.database.id, context.container.id);
|
||||||
|
await containerNode.openContextMenu();
|
||||||
|
await containerNode.contextMenuItem("New UDF").click();
|
||||||
|
|
||||||
|
// Assign UDF id
|
||||||
|
const udfIdTextBox = explorer.frame.getByLabel("User Defined Function Id");
|
||||||
|
const udfName: string = "extractDocumentId";
|
||||||
|
await udfIdTextBox.fill(udfName);
|
||||||
|
|
||||||
|
// Create UDF body that extracts the document id from a document
|
||||||
|
const udfBodyTextArea = explorer.frame.getByTestId("EditorReact/Host/Loaded");
|
||||||
|
await udfBodyTextArea.click();
|
||||||
|
|
||||||
|
// Clear existing content
|
||||||
|
const isMac: boolean = process.platform === "darwin";
|
||||||
|
await page.keyboard.press(isMac ? "Meta+A" : "Control+A");
|
||||||
|
await page.keyboard.press("Backspace");
|
||||||
|
|
||||||
|
await page.keyboard.type(udfBody);
|
||||||
|
|
||||||
|
// Save changes
|
||||||
|
const saveButton = explorer.commandBarButton(CommandBarButton.Save);
|
||||||
|
await expect(saveButton).toBeEnabled();
|
||||||
|
await saveButton.click();
|
||||||
|
await expect(explorer.getConsoleMessage()).toContainText(`Sucessfully created user defined function ${udfName}`, {
|
||||||
|
timeout: ONE_MINUTE_MS,
|
||||||
|
});
|
||||||
|
|
||||||
|
// Delete UDF
|
||||||
|
await containerNode.expand();
|
||||||
|
const udfsNode = await explorer.waitForNode(
|
||||||
|
`${context.database.id}/${context.container.id}/User Defined Functions`,
|
||||||
|
);
|
||||||
|
await udfsNode.expand();
|
||||||
|
const udfNode = await explorer.waitForNode(
|
||||||
|
`${context.database.id}/${context.container.id}/User Defined Functions/${udfName}`,
|
||||||
|
);
|
||||||
|
await udfNode.openContextMenu();
|
||||||
|
await udfNode.contextMenuItem("Delete User Defined Function").click();
|
||||||
|
const deleteUserDefinedFunctionButton = explorer.frame.getByTestId("DialogButton:Delete");
|
||||||
|
await deleteUserDefinedFunctionButton.click();
|
||||||
|
|
||||||
|
await expect(explorer.getConsoleMessage()).toContainText(`Successfully deleted user defined function ${udfName}`, {
|
||||||
|
timeout: ONE_MINUTE_MS,
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user