Accept extra args to build the connection string

testcontainers · Mar 23, 2023 · 4fab6c6 · 4fab6c6
1 parent 2c5f020
commit 4fab6c6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/modules/mysql/mysql.go b/modules/mysql/mysql.go
@@ -79,7 +79,7 @@ func StartContainer(ctx context.Context, opts ...MySQLContainerOption) (*MySQLCo
 	return &MySQLContainer{container, username, password, database}, nil
 }
 
-func (c *MySQLContainer) ConnectionString(ctx context.Context) (string, error) {
+func (c *MySQLContainer) ConnectionString(ctx context.Context, args ...string) (string, error) {
 	containerPort, err := c.MappedPort(ctx, "3306/tcp")
 	if err != nil {
 		return "", err
@@ -90,7 +90,15 @@ func (c *MySQLContainer) ConnectionString(ctx context.Context) (string, error) {
 		return "", err
 	}
 
-	connectionString := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", c.username, c.password, host, containerPort.Port(), c.database)
+	extraArgs := ""
+	if len(args) > 0 {
+		extraArgs = strings.Join(args, "&")
+	}
+	if extraArgs != "" {
+		extraArgs = "?" + extraArgs
+	}
+
+	connectionString := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s%s", c.username, c.password, host, containerPort.Port(), c.database, extraArgs)
 	return connectionString, nil
 }
 

diff --git a/modules/mysql/mysql_test.go b/modules/mysql/mysql_test.go
@@ -26,7 +26,7 @@ func TestMySQL(t *testing.T) {
 	})
 
 	// perform assertions
-	connectionString, _ := container.ConnectionString(ctx)
+	connectionString, _ := container.ConnectionString(ctx, "tls=skip-verify")
 
 	db, err := sql.Open("mysql", connectionString)
 	if err != nil {