mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
167 lines
5.1 KiB
Go
167 lines
5.1 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package api4
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hanzoteam/server/server/v8/channels/utils/testutils"
|
|
)
|
|
|
|
func TestGetBrandImage(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
client := th.Client
|
|
|
|
_, resp, err := client.GetBrandImage(context.Background())
|
|
require.Error(t, err)
|
|
CheckNotFoundStatus(t, resp)
|
|
|
|
_, err = client.Logout(context.Background())
|
|
require.NoError(t, err)
|
|
_, resp, err = client.GetBrandImage(context.Background())
|
|
require.Error(t, err)
|
|
CheckNotFoundStatus(t, resp)
|
|
|
|
_, resp, err = th.SystemAdminClient.GetBrandImage(context.Background())
|
|
require.Error(t, err)
|
|
CheckNotFoundStatus(t, resp)
|
|
}
|
|
|
|
func TestUploadBrandImage(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
client := th.Client
|
|
|
|
data, err := testutils.ReadTestFile("test.png")
|
|
require.NoError(t, err)
|
|
|
|
resp, err := client.UploadBrandImage(context.Background(), data)
|
|
require.Error(t, err)
|
|
CheckForbiddenStatus(t, resp)
|
|
|
|
// status code returns either forbidden or unauthorized
|
|
// note: forbidden is set as default at Client4.SetProfileImage when request is terminated early by server
|
|
_, err = client.Logout(context.Background())
|
|
require.NoError(t, err)
|
|
resp, err = client.UploadBrandImage(context.Background(), data)
|
|
require.Error(t, err)
|
|
if resp.StatusCode == http.StatusForbidden {
|
|
CheckForbiddenStatus(t, resp)
|
|
} else if resp.StatusCode == http.StatusUnauthorized {
|
|
CheckUnauthorizedStatus(t, resp)
|
|
} else {
|
|
require.Fail(t, "Should have failed either forbidden or unauthorized")
|
|
}
|
|
|
|
resp, err = th.SystemAdminClient.UploadBrandImage(context.Background(), data)
|
|
require.NoError(t, err)
|
|
CheckCreatedStatus(t, resp)
|
|
}
|
|
|
|
func TestUploadBrandImageTwice(t *testing.T) {
|
|
th := Setup(t)
|
|
|
|
data, err := testutils.ReadTestFile("test.png")
|
|
require.NoError(t, err)
|
|
|
|
// First upload as system admin
|
|
resp, err := th.SystemAdminClient.UploadBrandImage(context.Background(), data)
|
|
require.NoError(t, err)
|
|
CheckCreatedStatus(t, resp)
|
|
|
|
// Verify the image exists and contents match what was uploaded
|
|
receivedImg, resp, err := th.SystemAdminClient.GetBrandImage(context.Background())
|
|
require.NoError(t, err)
|
|
require.NotNil(t, receivedImg)
|
|
require.Equal(t, http.StatusOK, resp.StatusCode)
|
|
require.NotEmpty(t, receivedImg, "Received image data should not be empty")
|
|
|
|
// Get the list of files in the brand directory
|
|
files, err := th.App.FileBackend().ListDirectory("brand/")
|
|
require.NoError(t, err)
|
|
require.Len(t, files, 1, "Expected only the original image file")
|
|
|
|
// ListDirectory returns paths with the directory prefix included
|
|
fileName := files[0]
|
|
fileName = strings.TrimPrefix(fileName, "brand/")
|
|
require.Equal(t, "image.png", fileName, "Expected the original image file")
|
|
|
|
// Second upload (which should back up the previous one)
|
|
data2, err := testutils.ReadTestFile("test.tiff")
|
|
require.NoError(t, err)
|
|
|
|
resp, err = th.SystemAdminClient.UploadBrandImage(context.Background(), data2)
|
|
require.NoError(t, err)
|
|
CheckCreatedStatus(t, resp)
|
|
|
|
// Get the list of files in the brand directory again
|
|
files, err = th.App.FileBackend().ListDirectory("brand/")
|
|
require.NoError(t, err)
|
|
|
|
// Should now have the new image.png and a backup with timestamp
|
|
require.Len(t, files, 2, "Expected the original and backup files")
|
|
|
|
// Check that one of the files is image.png
|
|
hasOriginal := false
|
|
hasBackup := false
|
|
for _, file := range files {
|
|
// ListDirectory returns paths with the directory prefix included
|
|
fileName := strings.TrimPrefix(file, "brand/")
|
|
|
|
if fileName == "image.png" {
|
|
hasOriginal = true
|
|
} else if strings.HasSuffix(fileName, ".png") && strings.Contains(fileName, "-") {
|
|
// Backup file should have a timestamp format like 2006-01-02T15:04:05.png
|
|
hasBackup = true
|
|
}
|
|
}
|
|
|
|
require.True(t, hasOriginal, "Original image.png file should exist")
|
|
require.True(t, hasBackup, "Backup image file should exist")
|
|
|
|
// Verify the new image is available through the API and matches what was uploaded
|
|
receivedImg2, resp, err := th.SystemAdminClient.GetBrandImage(context.Background())
|
|
require.NoError(t, err)
|
|
require.NotNil(t, receivedImg2)
|
|
require.Equal(t, http.StatusOK, resp.StatusCode)
|
|
require.NotEmpty(t, receivedImg2, "Received image data should not be empty")
|
|
}
|
|
|
|
func TestDeleteBrandImage(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
|
|
data, err := testutils.ReadTestFile("test.png")
|
|
require.NoError(t, err)
|
|
|
|
resp, err := th.SystemAdminClient.UploadBrandImage(context.Background(), data)
|
|
require.NoError(t, err)
|
|
CheckCreatedStatus(t, resp)
|
|
|
|
resp, err = th.Client.DeleteBrandImage(context.Background())
|
|
require.Error(t, err)
|
|
CheckForbiddenStatus(t, resp)
|
|
|
|
_, err = th.Client.Logout(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
resp, err = th.Client.DeleteBrandImage(context.Background())
|
|
require.Error(t, err)
|
|
CheckUnauthorizedStatus(t, resp)
|
|
|
|
resp, err = th.SystemAdminClient.DeleteBrandImage(context.Background())
|
|
require.NoError(t, err)
|
|
CheckOKStatus(t, resp)
|
|
|
|
resp, err = th.SystemAdminClient.DeleteBrandImage(context.Background())
|
|
require.Error(t, err)
|
|
CheckNotFoundStatus(t, resp)
|
|
}
|