MySQL add isRoot parameter to executeQuery (#882)

Author: jeengbe

packages/modules/mysql/src/mysql-container.test.ts

@@ -99,5 +99,19 @@ describe("MySqlContainer", () => {
 
     await container.stop();
   });
+
+  it("should execute a query as root user", async () => {
+    const container = await new MySqlContainer().withUsername("customUsername").start();
+
+    // Test non-root user
+    const queryResult = await container.executeQuery("SELECT CURRENT_USER() as user");
+    expect(queryResult).toEqual(expect.stringContaining("user\ncustomUsername"));
+
+    // Test root user
+    const rootQueryResult = await container.executeQuery("SELECT CURRENT_USER() as user", [], true);
+    expect(rootQueryResult).toEqual(expect.stringContaining("user\nroot"));
+
+    await container.stop();
+  });
   // }
 });

packages/modules/mysql/src/mysql-container.ts

@@ -94,14 +94,14 @@ export class StartedMySqlContainer extends AbstractStartedContainer {
     return url.toString();
   }
 
-  public async executeQuery(query: string, additionalFlags: string[] = []): Promise<string> {
+  public async executeQuery(query: string, additionalFlags: string[] = [], isRoot = false): Promise<string> {
     const result = await this.startedTestContainer.exec([
       "mysql",
       "-h",
       "127.0.0.1",
       "-u",
-      this.username,
-      `-p${this.userPassword}`,
+      isRoot ? "root" : this.username,
+      `-p${isRoot ? this.getRootPassword() : this.getUserPassword()}`,
       "-e",
       `${query};`,
       ...additionalFlags,