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,116 @@ 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")
+
+	// assertContainer contains the logic for asserting the test
+	assertContainer := func(t *testing.T, ctx context.Context, options ...testcontainers.ContainerCustomizer) {
+		ctr, err := mssql.Run(ctx,
+			"mcr.microsoft.com/mssql/server:2022-CU17-ubuntu-22.04",
+			append([]testcontainers.ContainerCustomizer{mssql.WithAcceptEULA()}, 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()
+
+		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)
+	}
+
+	ctx := context.Background()
+
+	t.Run("WithPassword/beforeWithScripts", func(t *testing.T) {
+		assertContainer(t, ctx,
+			mssql.WithPassword(password),
+			mssql.WithScripts(seedFile),
+		)
+	})
+
+	t.Run("WithPassword/afterWithScripts", func(t *testing.T) {
+		assertContainer(t, ctx,
+			mssql.WithScripts(seedFile),
+			mssql.WithPassword(password),
+		)
+	})
+}
+
+// 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