snapshot of https://github.com/mattermost/mattermost.git (master, 2026-09-16; blobs > 131072B omitted)

server/channels/store/sqlstore/migration_000217_test.go

@@ -0,0 +1,121 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+
+package sqlstore
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/mattermost/mattermost/server/public/model"
+	"github.com/mattermost/mattermost/server/public/shared/mlog"
+)
+
+func TestMigration000217(t *testing.T) {
+	logger := mlog.CreateTestLogger(t)
+
+	settings, err := makeSqlSettings(model.DatabaseDriverPostgres)
+	if err != nil {
+		t.Skip(err)
+	}
+
+	store, err := New(*settings, logger, nil)
+	require.NoError(t, err)
+	defer store.Close()
+
+	master := store.GetMaster()
+
+	upSQL := readMigrationSQL(t, "000217_resize_message_columns.up.sql")
+	downSQL := readMigrationSQL(t, "000217_resize_message_columns.down.sql")
+
+	type tableCol struct {
+		table  string
+		column string
+	}
+	targets := []tableCol{
+		{"posts", "message"},
+		{"drafts", "message"},
+		{"scheduledposts", "message"},
+		{"temporaryposts", "message"},
+	}
+
+	colLength := func(t *testing.T, table, column string) int {
+		t.Helper()
+		var length int
+		require.NoError(t, master.Get(&length, fmt.Sprintf(`
+			SELECT COALESCE(character_maximum_length, 0)
+			FROM information_schema.columns
+			WHERE table_name = '%s' AND column_name = '%s'
+		`, table, column)))
+		return length
+	}
+
+	setColLength := func(t *testing.T, table, column string, size int) {
+		t.Helper()
+		_, alterErr := master.ExecNoTimeout(fmt.Sprintf("ALTER TABLE %s ALTER COLUMN %s TYPE VARCHAR(%d)", table, column, size))
+		require.NoError(t, alterErr)
+	}
+
+	restoreTo := func(t *testing.T, size int) {
+		t.Helper()
+		for _, tc := range targets {
+			_, alterErr := master.ExecNoTimeout(fmt.Sprintf("ALTER TABLE %s ALTER COLUMN %s TYPE VARCHAR(%d)", tc.table, tc.column, size))
+			require.NoError(t, alterErr)
+		}
+	}
+
+	t.Run("NormalUpThenDown", func(t *testing.T) {
+		t.Cleanup(func() { restoreTo(t, 1048576) })
+
+		for _, tc := range targets {
+			setColLength(t, tc.table, tc.column, 65535)
+		}
+
+		_, err := master.ExecNoTimeout(upSQL)
+		require.NoError(t, err, "up migration should succeed")
+
+		for _, tc := range targets {
+			assert.Equal(t, 1048576, colLength(t, tc.table, tc.column), "%s.%s after up migration", tc.table, tc.column)
+		}
+
+		_, err = master.ExecNoTimeout(downSQL)
+		require.NoError(t, err, "down migration should succeed")
+
+		for _, tc := range targets {
+			assert.Equal(t, 65535, colLength(t, tc.table, tc.column), "%s.%s after down migration", tc.table, tc.column)
+		}
+	})
+
+	t.Run("UpSkipsWhenAlreadyLarger", func(t *testing.T) {
+		t.Cleanup(func() { restoreTo(t, 1048576) })
+
+		for _, tc := range targets {
+			setColLength(t, tc.table, tc.column, 2097152)
+		}
+
+		_, err := master.ExecNoTimeout(upSQL)
+		require.NoError(t, err, "up migration should succeed even when columns are already larger")
+
+		for _, tc := range targets {
+			assert.Equal(t, 2097152, colLength(t, tc.table, tc.column), "%s.%s should be unchanged by up migration", tc.table, tc.column)
+		}
+	})
+
+	t.Run("DownSkipsWhenLargerThanTarget", func(t *testing.T) {
+		t.Cleanup(func() { restoreTo(t, 1048576) })
+
+		for _, tc := range targets {
+			setColLength(t, tc.table, tc.column, 2097152)
+		}
+
+		_, err := master.ExecNoTimeout(downSQL)
+		require.NoError(t, err, "down migration should succeed even when columns exceed the target size")
+
+		for _, tc := range targets {
+			assert.Equal(t, 2097152, colLength(t, tc.table, tc.column), "%s.%s should be unchanged by down migration", tc.table, tc.column)
+		}
+	})
+}