Update modules/yugabytedb/yugabytedb.go

Co-authored-by: Steven Hartland <[email protected]>
testcontainers · Oct 15, 2024 · a51e4dd · a51e4dd
1 parent 6b09e71
commit a51e4dd
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/modules/yugabytedb/yugabytedb.go b/modules/yugabytedb/yugabytedb.go
@@ -224,14 +224,17 @@ func (y *YugabyteDBContainer) YCQLConfigureClusterConfig(ctx context.Context, cf
 	)
 
 	for _, env := range inspect.Config.Env {
-		parts := strings.SplitN(env, "=", 2)
-		switch parts[0] {
+		key, val, found := strings.Cut(env, "=")
+		if !found {
+			continue
+		}
+		switch key {
 		case ycqlKeyspaceEnv:
-			keyspace = parts[1]
+			keyspace = val
 		case ycqlUserNameEnv:
-			user = parts[1]
+			user = val
 		case ycqlPasswordEnv:
-			password = parts[1]
+			password = val
 		}
 	}