mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
106 lines
3.3 KiB
Go
106 lines
3.3 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package plugin_test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost-server/v5/model"
|
|
"github.com/mattermost/mattermost-server/v5/plugin"
|
|
"github.com/mattermost/mattermost-server/v5/plugin/plugintest"
|
|
"github.com/mattermost/mattermost-server/v5/plugin/plugintest/mock"
|
|
"github.com/mattermost/mattermost-server/v5/utils/fileutils"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestInstallPluginFromURL(t *testing.T) {
|
|
replace := true
|
|
|
|
t.Run("incompatible server version", func(t *testing.T) {
|
|
h := &plugin.HelpersImpl{}
|
|
api := &plugintest.API{}
|
|
api.On("GetServerVersion").Return("5.1.0")
|
|
h.API = api
|
|
|
|
_, err := h.InstallPluginFromURL("", true)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, "incompatible server version for plugin, minimum required version: 5.18.0, current version: 5.1.0", err.Error())
|
|
})
|
|
|
|
t.Run("error while parsing the download url", func(t *testing.T) {
|
|
h := &plugin.HelpersImpl{}
|
|
api := &plugintest.API{}
|
|
api.On("GetServerVersion").Return("5.19.0")
|
|
h.API = api
|
|
_, err := h.InstallPluginFromURL("http://%41:8080/", replace)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, "error while parsing url: parse http://%41:8080/: invalid URL escape \"%41\"", err.Error())
|
|
})
|
|
|
|
t.Run("errors out while downloading file", func(t *testing.T) {
|
|
h := &plugin.HelpersImpl{}
|
|
api := &plugintest.API{}
|
|
api.On("GetServerVersion").Return("5.19.0")
|
|
h.API = api
|
|
testServer := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
|
res.WriteHeader(http.StatusInternalServerError)
|
|
}))
|
|
defer testServer.Close()
|
|
url := testServer.URL
|
|
|
|
_, err := h.InstallPluginFromURL(url, replace)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, "received 500 status code while downloading plugin from server", err.Error())
|
|
})
|
|
|
|
t.Run("downloads the file successfully", func(t *testing.T) {
|
|
h := &plugin.HelpersImpl{}
|
|
api := &plugintest.API{}
|
|
api.On("GetServerVersion").Return("5.19.0")
|
|
h.API = api
|
|
path, _ := fileutils.FindDir("tests")
|
|
tarData, err := ioutil.ReadFile(filepath.Join(path, "testplugin.tar.gz"))
|
|
require.NoError(t, err)
|
|
expectedManifest := &model.Manifest{Id: "testplugin"}
|
|
api.On("InstallPlugin", mock.Anything, false).Return(expectedManifest, nil)
|
|
|
|
testServer := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
|
res.WriteHeader(http.StatusOK)
|
|
_, _ = res.Write(tarData)
|
|
}))
|
|
defer testServer.Close()
|
|
url := testServer.URL
|
|
|
|
manifest, err := h.InstallPluginFromURL(url, false)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "testplugin", manifest.Id)
|
|
})
|
|
|
|
t.Run("the url pointing to server is incorrect", func(t *testing.T) {
|
|
h := &plugin.HelpersImpl{}
|
|
api := &plugintest.API{}
|
|
api.On("GetServerVersion").Return("5.19.0")
|
|
h.API = api
|
|
testServer := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
|
res.WriteHeader(http.StatusNotFound)
|
|
}))
|
|
defer testServer.Close()
|
|
url := testServer.URL
|
|
|
|
_, err := h.InstallPluginFromURL(url, false)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, "received 404 status code while downloading plugin from server", err.Error())
|
|
})
|
|
}
|