test(mssql): add test for WithInitSQL

Author: MattiasMTS

modules/mssql/mssql_test.go

@@ -1,13 +1,15 @@
 package mssql_test
 
 import (
+	"bytes"
 	"context"
 	"database/sql"
 	"testing"
 
+	_ "embed"
+
 	_ "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 +130,72 @@ func TestMSSQLServerWithInvalidPassword(t *testing.T) {
 	testcontainers.CleanupContainer(t, ctr)
 	require.NoError(t, err)
 }
+
+//go:embed testdata/seed.sql
+var seedSQLContent []byte
+
+// tests that a container can be created with a DDL script
+func TestMSSQLServerWithScriptsDDL(t *testing.T) {
+	const password = "MyCustom@Passw0rd"
+
+	// 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...)...,
+		)
+		testcontainers.CleanupContainer(t, ctr)
+		require.NoError(t, err)
+
+		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, len(want))
+
+		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 := t.Context()
+
+	t.Run("WithPassword/beforeWithScripts", func(t *testing.T) {
+		assertContainer(t, ctx,
+			mssql.WithPassword(password),
+			mssql.WithInitSQL(bytes.NewReader(seedSQLContent)),
+		)
+	})
+
+	t.Run("WithPassword/afterWithScripts", func(t *testing.T) {
+		assertContainer(t, ctx,
+			mssql.WithInitSQL(bytes.NewReader(seedSQLContent)),
+			mssql.WithPassword(password),
+		)
+	})
+}

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