mirror of
https://github.com/mattermost/mattermost.git
synced 2026-02-20 16:31:10 -05:00
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package plugintest_test
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/mattermost-server/model"
|
|
"github.com/mattermost/mattermost-server/plugin"
|
|
"github.com/mattermost/mattermost-server/plugin/plugintest"
|
|
)
|
|
|
|
type HelloUserPlugin struct {
|
|
plugin.MattermostPlugin
|
|
}
|
|
|
|
func (p *HelloUserPlugin) ServeHTTP(context *plugin.Context, w http.ResponseWriter, r *http.Request) {
|
|
userId := r.Header.Get("Mattermost-User-Id")
|
|
user, err := p.API.GetUser(userId)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
p.API.LogError(err.Error())
|
|
return
|
|
}
|
|
|
|
fmt.Fprintf(w, "Welcome back, %s!", user.Username)
|
|
}
|
|
|
|
func Example() {
|
|
t := &testing.T{}
|
|
user := &model.User{
|
|
Id: model.NewId(),
|
|
Username: "billybob",
|
|
}
|
|
|
|
api := &plugintest.API{}
|
|
api.On("GetUser", user.Id).Return(user, nil)
|
|
defer api.AssertExpectations(t)
|
|
|
|
p := &HelloUserPlugin{}
|
|
p.SetAPI(api)
|
|
|
|
w := httptest.NewRecorder()
|
|
r := httptest.NewRequest("GET", "/", nil)
|
|
r.Header.Add("Mattermost-User-Id", user.Id)
|
|
p.ServeHTTP(&plugin.Context{}, w, r)
|
|
body, err := ioutil.ReadAll(w.Result().Body)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "Welcome back, billybob!", string(body))
|
|
}
|