test(mssql): add test for WithScripts and GetSQLCmdPath

Author: MattiasMTS

modules/mssql/mssql_test.go

@@ -3,11 +3,11 @@ package mssql_test
 import (
 	"context"
 	"database/sql"
+	"path/filepath"
 	"testing"
 
 	_ "github.com/microsoft/go-mssqldb"
 	"github.com/stretchr/testify/require"
-
 	"github.com/testcontainers/testcontainers-go"
 	"github.com/testcontainers/testcontainers-go/modules/mssql"
 	"github.com/testcontainers/testcontainers-go/wait"
@@ -128,3 +128,125 @@ func TestMSSQLServerWithInvalidPassword(t *testing.T) {
 	testcontainers.CleanupContainer(t, ctr)
 	require.NoError(t, err)
 }
+
+// tests that a container can be created with a DDL script
+func TestMSSQLServerWithScriptsDDL(t *testing.T) {
+	const password = "MyCustom@Passw0rd"
+	seedFile := filepath.Join("testdata", "seed.sql")
+
+	tests := []struct {
+		name  string
+		input []testcontainers.ContainerCustomizer
+	}{
+		{
+			name: "WithPassword/beforeWithScripts",
+			input: []testcontainers.ContainerCustomizer{
+				mssql.WithPassword(password),
+				mssql.WithScripts(seedFile),
+			},
+		},
+		{
+			name: "WithPassword/afterWithScripts",
+			input: []testcontainers.ContainerCustomizer{
+				mssql.WithScripts(seedFile),
+				mssql.WithPassword(password),
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			ctx := context.Background()
+
+			options := append([]testcontainers.ContainerCustomizer{
+				mssql.WithAcceptEULA(),
+			}, tt.input...)
+
+			ctr, err := mssql.Run(ctx, "mcr.microsoft.com/mssql/server:2022-CU17-ubuntu-22.04", options...)
+			require.NoError(t, err)
+			testcontainers.CleanupContainer(t, ctr)
+
+			connectionString, err := ctr.ConnectionString(ctx)
+			require.NoError(t, err)
+
+			db, err := sql.Open("sqlserver", connectionString)
+			require.NoError(t, err)
+			defer db.Close()
+
+			err = db.PingContext(ctx)
+			require.NoError(t, err)
+
+			rows, err := db.QueryContext(ctx, "SELECT * FROM pizza_palace.pizzas")
+			require.NoError(t, err)
+			defer rows.Close()
+
+			// test data structure, must match the seed.sql file
+			type Pizza struct {
+				ID            int
+				ToppingName   string
+				Deliciousness string
+			}
+
+			want := []Pizza{
+				{1, "Pineapple", "Controversial but tasty"},
+				{2, "Pepperoni", "Classic never fails"},
+			}
+			got := make([]Pizza, 0)
+
+			for rows.Next() {
+				var p Pizza
+				err := rows.Scan(&p.ID, &p.ToppingName, &p.Deliciousness)
+				require.NoError(t, err)
+				got = append(got, p)
+			}
+
+			require.EqualValues(t, want, got)
+		})
+	}
+}
+
+// tests that the correct path to the sqlcmd executable is returned based on the image tag
+func TestGetSQLCmdPath(t *testing.T) {
+	tests := []struct {
+		name  string
+		input string
+		want  string
+	}{
+		{
+			name:  "2019-CU14/old-version",
+			input: "mcr.microsoft.com/mssql/server:2019-CU14-ubuntu-20.04",
+			want:  "/opt/mssql-tools/bin/sqlcmd",
+		},
+		{
+			name:  "2022-latest/new-version",
+			input: "mcr.microsoft.com/mssql/server:2022-latest",
+			want:  "/opt/mssql-tools18/bin/sqlcmd",
+		},
+		{
+			name:  "2019-latest/old-version",
+			input: "mcr.microsoft.com/mssql/server:2019-latest",
+			want:  "/opt/mssql-tools/bin/sqlcmd",
+		},
+		{
+			name:  "2022-CU14/new-version",
+			input: "mcr.microsoft.com/mssql/server:2022-CU14-ubuntu-22.04",
+			want:  "/opt/mssql-tools18/bin/sqlcmd",
+		},
+		{
+			name:  "2022-CU17/new-version",
+			input: "mcr.microsoft.com/mssql/server:2022-CU17-ubuntu-22.04",
+			want:  "/opt/mssql-tools18/bin/sqlcmd",
+		},
+		{
+			name:  "2025-CU14/new-version",
+			input: "mcr.microsoft.com/mssql/server:2025-CU14-ubuntu-22.04",
+			want:  "/opt/mssql-tools18/bin/sqlcmd",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := mssql.GetSQLCmdPath(tt.input)
+			require.Equal(t, tt.want, got)
+		})
+	}
+}

modules/mssql/testdata/seed.sql

@@ -0,0 +1,14 @@
+CREATE SCHEMA pizza_palace;
+GO
+
+CREATE TABLE pizza_palace.pizzas (
+    ID INT PRIMARY KEY IDENTITY,
+    ToppingName NVARCHAR(100),
+    Deliciousness NVARCHAR(100) UNIQUE
+);
+GO
+
+INSERT INTO pizza_palace.pizzas (ToppingName, Deliciousness) VALUES
+    ('Pineapple', 'Controversial but tasty'),
+    ('Pepperoni', 'Classic never fails')
+GO