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

server/channels/app/file_helper_test.go

@@ -0,0 +1,207 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+
+package app
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/mattermost/mattermost/server/public/model"
+)
+
+func TestFilterInaccessibleFiles(t *testing.T) {
+	mainHelper.Parallel(t)
+	th := Setup(t)
+	th.App.Srv().SetLicense(model.NewTestLicense("cloud"))
+	err := th.App.Srv().Store().System().Save(&model.System{
+		Name:  model.SystemLastAccessibleFileTime,
+		Value: "2",
+	})
+	require.NoError(t, err)
+
+	getFileWithCreateAt := func(at int64) *model.FileInfo {
+		return &model.FileInfo{CreateAt: at}
+	}
+
+	t.Run("ascending order returns correct files", func(t *testing.T) {
+		fileList := &model.FileInfoList{
+			FileInfos: map[string]*model.FileInfo{
+				"file_a": getFileWithCreateAt(0),
+				"file_b": getFileWithCreateAt(1),
+				"file_c": getFileWithCreateAt(2),
+				"file_d": getFileWithCreateAt(3),
+				"file_e": getFileWithCreateAt(4),
+			},
+			Order: []string{"file_a", "file_b", "file_c", "file_d", "file_e"},
+		}
+		appErr := th.App.filterInaccessibleFiles(fileList, filterFileOptions{assumeSortedCreatedAt: true})
+
+		require.Nil(t, appErr)
+
+		assert.Equal(t, map[string]*model.FileInfo{
+			"file_c": getFileWithCreateAt(2),
+			"file_d": getFileWithCreateAt(3),
+			"file_e": getFileWithCreateAt(4),
+		}, fileList.FileInfos)
+
+		assert.Equal(t, []string{
+			"file_c",
+			"file_d",
+			"file_e",
+		}, fileList.Order)
+		assert.Equal(t, int64(1), fileList.FirstInaccessibleFileTime)
+	})
+
+	t.Run("descending order returns correct files", func(t *testing.T) {
+		fileList := &model.FileInfoList{
+			FileInfos: map[string]*model.FileInfo{
+				"file_a": getFileWithCreateAt(0),
+				"file_b": getFileWithCreateAt(1),
+				"file_c": getFileWithCreateAt(2),
+				"file_d": getFileWithCreateAt(3),
+				"file_e": getFileWithCreateAt(4),
+			},
+			Order: []string{"file_e", "file_d", "file_c", "file_b", "file_a"},
+		}
+		appErr := th.App.filterInaccessibleFiles(fileList, filterFileOptions{assumeSortedCreatedAt: true})
+
+		require.Nil(t, appErr)
+
+		assert.Equal(t, map[string]*model.FileInfo{
+			"file_c": getFileWithCreateAt(2),
+			"file_d": getFileWithCreateAt(3),
+			"file_e": getFileWithCreateAt(4),
+		}, fileList.FileInfos)
+
+		assert.Equal(t, []string{
+			"file_e",
+			"file_d",
+			"file_c",
+		}, fileList.Order)
+
+		assert.Equal(t, int64(1), fileList.FirstInaccessibleFileTime)
+	})
+
+	t.Run("handles mixed create at ordering correctly if correct options given", func(t *testing.T) {
+		fileList := &model.FileInfoList{
+			FileInfos: map[string]*model.FileInfo{
+				"file_a": getFileWithCreateAt(0),
+				"file_b": getFileWithCreateAt(1),
+				"file_c": getFileWithCreateAt(2),
+				"file_d": getFileWithCreateAt(3),
+				"file_e": getFileWithCreateAt(4),
+			},
+			Order: []string{"file_e", "file_b", "file_a", "file_d", "file_c"},
+		}
+		appErr := th.App.filterInaccessibleFiles(fileList, filterFileOptions{assumeSortedCreatedAt: false})
+
+		require.Nil(t, appErr)
+
+		assert.Equal(t, map[string]*model.FileInfo{
+			"file_c": getFileWithCreateAt(2),
+			"file_d": getFileWithCreateAt(3),
+			"file_e": getFileWithCreateAt(4),
+		}, fileList.FileInfos)
+
+		assert.Equal(t, []string{
+			"file_e",
+			"file_d",
+			"file_c",
+		}, fileList.Order)
+	})
+}
+
+func TestGetFilteredAccessibleFiles(t *testing.T) {
+	mainHelper.Parallel(t)
+	th := Setup(t)
+	th.App.Srv().SetLicense(model.NewTestLicense("cloud"))
+	err := th.App.Srv().Store().System().Save(&model.System{
+		Name:  model.SystemLastAccessibleFileTime,
+		Value: "2",
+	})
+	require.NoError(t, err)
+
+	getFileWithCreateAt := func(at int64) *model.FileInfo {
+		return &model.FileInfo{CreateAt: at}
+	}
+
+	t.Run("ascending order returns correct files", func(t *testing.T) {
+		files := []*model.FileInfo{getFileWithCreateAt(0), getFileWithCreateAt(1), getFileWithCreateAt(2), getFileWithCreateAt(3), getFileWithCreateAt(4)}
+		filteredFiles, firstInaccessibleFileTime, appErr := th.App.getFilteredAccessibleFiles(files, filterFileOptions{assumeSortedCreatedAt: true})
+
+		require.Nil(t, appErr)
+		assert.Equal(t, []*model.FileInfo{getFileWithCreateAt(2), getFileWithCreateAt(3), getFileWithCreateAt(4)}, filteredFiles)
+		assert.Equal(t, int64(1), firstInaccessibleFileTime)
+	})
+
+	t.Run("descending order returns correct files", func(t *testing.T) {
+		files := []*model.FileInfo{getFileWithCreateAt(4), getFileWithCreateAt(3), getFileWithCreateAt(2), getFileWithCreateAt(1), getFileWithCreateAt(0)}
+		filteredFiles, firstInaccessibleFileTime, appErr := th.App.getFilteredAccessibleFiles(files, filterFileOptions{assumeSortedCreatedAt: true})
+
+		require.Nil(t, appErr)
+		assert.Equal(t, []*model.FileInfo{getFileWithCreateAt(4), getFileWithCreateAt(3), getFileWithCreateAt(2)}, filteredFiles)
+		assert.Equal(t, int64(1), firstInaccessibleFileTime)
+	})
+
+	t.Run("handles mixed create at ordering correctly if correct options given", func(t *testing.T) {
+		files := []*model.FileInfo{getFileWithCreateAt(4), getFileWithCreateAt(1), getFileWithCreateAt(0), getFileWithCreateAt(3), getFileWithCreateAt(2)}
+		filteredFiles, _, appErr := th.App.getFilteredAccessibleFiles(files, filterFileOptions{assumeSortedCreatedAt: false})
+
+		require.Nil(t, appErr)
+		assert.Equal(t, []*model.FileInfo{getFileWithCreateAt(4), getFileWithCreateAt(3), getFileWithCreateAt(2)}, filteredFiles)
+	})
+}
+
+func TestIsInaccessibleFile(t *testing.T) {
+	mainHelper.Parallel(t)
+	th := Setup(t)
+	th.App.Srv().SetLicense(model.NewTestLicense("cloud"))
+	err := th.App.Srv().Store().System().Save(&model.System{
+		Name:  model.SystemLastAccessibleFileTime,
+		Value: "2",
+	})
+	require.NoError(t, err)
+
+	file := &model.FileInfo{CreateAt: 3}
+	firstInaccessibleFileTime, appErr := th.App.isInaccessibleFile(file)
+	require.Nil(t, appErr)
+	assert.Equal(t, int64(0), firstInaccessibleFileTime)
+
+	file = &model.FileInfo{CreateAt: 1}
+	firstInaccessibleFileTime, appErr = th.App.isInaccessibleFile(file)
+	require.Nil(t, appErr)
+	assert.Equal(t, int64(1), firstInaccessibleFileTime)
+}
+
+func TestRemoveInaccessibleContentFromFilesSlice(t *testing.T) {
+	mainHelper.Parallel(t)
+	th := Setup(t)
+	th.App.Srv().SetLicense(model.NewTestLicense("cloud"))
+	err := th.App.Srv().Store().System().Save(&model.System{
+		Name:  model.SystemLastAccessibleFileTime,
+		Value: "2",
+	})
+	require.NoError(t, err)
+
+	getFileWithCreateAt := func(at int64) *model.FileInfo {
+		return &model.FileInfo{CreateAt: at}
+	}
+
+	files := []*model.FileInfo{getFileWithCreateAt(4), getFileWithCreateAt(1), getFileWithCreateAt(0), getFileWithCreateAt(3), getFileWithCreateAt(2)}
+
+	_, appErr := th.App.removeInaccessibleContentFromFilesSlice(files)
+
+	require.Nil(t, appErr)
+	assert.Len(t, files, len(files))
+	for _, file := range files {
+		// Inaccessible files are archived
+		if file.CreateAt < 2 {
+			assert.True(t, file.Archived)
+		} else {
+			assert.False(t, file.Archived)
+		}
+	}
+}