mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
efaa6264cc
It was a good decision in hindsight to keep the public module as 0.x because this would have been a breaking change again. https://mattermost.atlassian.net/browse/MM-53032 ```release-note Changed the Go module path from github.com/mattermost/mattermost-server/server/v8 to github.com/mattermost/mattermost/server/v8. For the public facing module, it's path is also changed from github.com/mattermost/mattermost-server/server/public to github.com/mattermost/mattermost/server/public ```
211 lines
6.2 KiB
Go
211 lines
6.2 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package api4
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/v8/channels/utils/fileutils"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestListExports(t *testing.T) {
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
t.Run("no permissions", func(t *testing.T) {
|
|
exports, _, err := th.Client.ListExports(context.Background())
|
|
require.Error(t, err)
|
|
CheckErrorID(t, err, "api.context.permissions.app_error")
|
|
require.Nil(t, exports)
|
|
})
|
|
|
|
th.TestForSystemAdminAndLocal(t, func(t *testing.T, c *model.Client4) {
|
|
exports, _, err := c.ListExports(context.Background())
|
|
require.NoError(t, err)
|
|
require.Empty(t, exports)
|
|
}, "no exports")
|
|
|
|
dataDir, found := fileutils.FindDir("data")
|
|
require.True(t, found)
|
|
|
|
th.TestForSystemAdminAndLocal(t, func(t *testing.T, c *model.Client4) {
|
|
exportDir := filepath.Join(dataDir, *th.App.Config().ExportSettings.Directory)
|
|
err := os.Mkdir(exportDir, 0700)
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(exportDir)
|
|
|
|
f, err := os.Create(filepath.Join(exportDir, "export.zip"))
|
|
require.NoError(t, err)
|
|
f.Close()
|
|
|
|
exports, _, err := c.ListExports(context.Background())
|
|
require.NoError(t, err)
|
|
require.Len(t, exports, 1)
|
|
require.Equal(t, exports[0], "export.zip")
|
|
}, "expected exports")
|
|
|
|
th.TestForSystemAdminAndLocal(t, func(t *testing.T, c *model.Client4) {
|
|
value := *th.App.Config().ExportSettings.Directory
|
|
th.App.UpdateConfig(func(cfg *model.Config) { *cfg.ExportSettings.Directory = value + "new" })
|
|
defer th.App.UpdateConfig(func(cfg *model.Config) { *cfg.ExportSettings.Directory = value })
|
|
|
|
exportDir := filepath.Join(dataDir, value+"new")
|
|
err := os.Mkdir(exportDir, 0700)
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(exportDir)
|
|
|
|
exports, _, err := c.ListExports(context.Background())
|
|
require.NoError(t, err)
|
|
require.Empty(t, exports)
|
|
|
|
f, err := os.Create(filepath.Join(exportDir, "export.zip"))
|
|
require.NoError(t, err)
|
|
f.Close()
|
|
|
|
exports, _, err = c.ListExports(context.Background())
|
|
require.NoError(t, err)
|
|
require.Len(t, exports, 1)
|
|
require.Equal(t, "export.zip", exports[0])
|
|
}, "change export directory")
|
|
}
|
|
|
|
func TestDeleteExport(t *testing.T) {
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
t.Run("no permissions", func(t *testing.T) {
|
|
_, err := th.Client.DeleteExport(context.Background(), "export.zip")
|
|
require.Error(t, err)
|
|
CheckErrorID(t, err, "api.context.permissions.app_error")
|
|
})
|
|
|
|
dataDir, found := fileutils.FindDir("data")
|
|
require.True(t, found)
|
|
exportDir := filepath.Join(dataDir, *th.App.Config().ExportSettings.Directory)
|
|
|
|
th.TestForSystemAdminAndLocal(t, func(t *testing.T, c *model.Client4) {
|
|
err := os.Mkdir(exportDir, 0700)
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(exportDir)
|
|
exportName := "export.zip"
|
|
f, err := os.Create(filepath.Join(exportDir, exportName))
|
|
require.NoError(t, err)
|
|
f.Close()
|
|
|
|
exports, _, err := c.ListExports(context.Background())
|
|
require.NoError(t, err)
|
|
require.Len(t, exports, 1)
|
|
require.Equal(t, exports[0], exportName)
|
|
|
|
_, err = c.DeleteExport(context.Background(), exportName)
|
|
require.NoError(t, err)
|
|
|
|
exports, _, err = c.ListExports(context.Background())
|
|
require.NoError(t, err)
|
|
require.Empty(t, exports)
|
|
|
|
// verify idempotence
|
|
_, err = c.DeleteExport(context.Background(), exportName)
|
|
require.NoError(t, err)
|
|
}, "successfully delete export")
|
|
}
|
|
|
|
func TestDownloadExport(t *testing.T) {
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
t.Run("no permissions", func(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
n, _, err := th.Client.DownloadExport(context.Background(), "export.zip", &buf, 0)
|
|
require.Error(t, err)
|
|
CheckErrorID(t, err, "api.context.permissions.app_error")
|
|
require.Zero(t, n)
|
|
})
|
|
|
|
dataDir, found := fileutils.FindDir("data")
|
|
require.True(t, found)
|
|
exportDir := filepath.Join(dataDir, *th.App.Config().ExportSettings.Directory)
|
|
|
|
th.TestForSystemAdminAndLocal(t, func(t *testing.T, c *model.Client4) {
|
|
var buf bytes.Buffer
|
|
n, _, err := c.DownloadExport(context.Background(), "export.zip", &buf, 0)
|
|
require.Error(t, err)
|
|
CheckErrorID(t, err, "api.export.export_not_found.app_error")
|
|
require.Zero(t, n)
|
|
}, "not found")
|
|
|
|
th.TestForSystemAdminAndLocal(t, func(t *testing.T, c *model.Client4) {
|
|
err := os.Mkdir(exportDir, 0700)
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(exportDir)
|
|
|
|
data := randomBytes(t, 1024*1024)
|
|
var buf bytes.Buffer
|
|
exportName := "export.zip"
|
|
err = os.WriteFile(filepath.Join(exportDir, exportName), data, 0600)
|
|
require.NoError(t, err)
|
|
|
|
n, _, err := c.DownloadExport(context.Background(), exportName, &buf, 0)
|
|
require.NoError(t, err)
|
|
require.Equal(t, len(data), int(n))
|
|
require.Equal(t, data, buf.Bytes())
|
|
}, "full download")
|
|
|
|
th.TestForSystemAdminAndLocal(t, func(t *testing.T, c *model.Client4) {
|
|
err := os.Mkdir(exportDir, 0700)
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(exportDir)
|
|
|
|
data := randomBytes(t, 1024*1024)
|
|
var buf bytes.Buffer
|
|
exportName := "export.zip"
|
|
err = os.WriteFile(filepath.Join(exportDir, exportName), data, 0600)
|
|
require.NoError(t, err)
|
|
|
|
offset := 1024 * 512
|
|
n, _, err := c.DownloadExport(context.Background(), exportName, &buf, int64(offset))
|
|
require.NoError(t, err)
|
|
require.Equal(t, len(data)-offset, int(n))
|
|
require.Equal(t, data[offset:], buf.Bytes())
|
|
}, "download with offset")
|
|
}
|
|
|
|
func BenchmarkDownloadExport(b *testing.B) {
|
|
th := Setup(b)
|
|
defer th.TearDown()
|
|
|
|
dataDir, found := fileutils.FindDir("data")
|
|
require.True(b, found)
|
|
exportDir := filepath.Join(dataDir, *th.App.Config().ExportSettings.Directory)
|
|
|
|
err := os.Mkdir(exportDir, 0700)
|
|
require.NoError(b, err)
|
|
defer os.RemoveAll(exportDir)
|
|
|
|
exportName := "export.zip"
|
|
f, err := os.Create(filepath.Join(exportDir, exportName))
|
|
require.NoError(b, err)
|
|
f.Close()
|
|
|
|
err = os.Truncate(filepath.Join(exportDir, exportName), 1024*1024*1024)
|
|
require.NoError(b, err)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
outFilePath := filepath.Join(dataDir, fmt.Sprintf("export%d.zip", i))
|
|
outFile, _ := os.Create(outFilePath)
|
|
th.SystemAdminClient.DownloadExport(context.Background(), exportName, outFile, 0)
|
|
outFile.Close()
|
|
os.Remove(outFilePath)
|
|
}
|
|
}
|