Karti Tripathi
a4e51f6412
- 27 base tools: channels, messaging, users, reactions, files, DMs - Team management: invite/remove users, get stats, list members - Slash commands: execute /remind, /poll, etc. - Webhook management: incoming and outgoing webhooks - System tools: server config, logs, bulk status updates - Channel admin: create, invite, leave, delete channels - Read-only mode for safe exploration - Dual token support (bot + PAT) for enhanced security - Apache 2.0 licensed
109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
package channel
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/karti-ai/mattermost-mcp-server/pkg/mattermost"
|
|
"github.com/mark3labs/mcp-go/mcp"
|
|
"github.com/mattermost/mattermost-server/v6/model"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSlimChannel(t *testing.T) {
|
|
c := &model.Channel{
|
|
Id: "channel123",
|
|
Name: "general",
|
|
DisplayName: "General",
|
|
Type: "O",
|
|
TeamId: "team456",
|
|
}
|
|
|
|
slim := SlimChannel(c)
|
|
assert.NotNil(t, slim)
|
|
assert.Equal(t, "channel123", slim["id"])
|
|
assert.Equal(t, "general", slim["name"])
|
|
assert.Equal(t, "General", slim["display_name"])
|
|
assert.Equal(t, "O", slim["type"])
|
|
assert.Equal(t, "team456", slim["team_id"])
|
|
}
|
|
|
|
func TestSlimChannel_Nil(t *testing.T) {
|
|
slim := SlimChannel(nil)
|
|
assert.Nil(t, slim)
|
|
}
|
|
|
|
func TestSlimChannel_DifferentTypes(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
chType model.ChannelType
|
|
expected string
|
|
}{
|
|
{"Open channel", model.ChannelTypeOpen, "O"},
|
|
{"Private channel", model.ChannelTypePrivate, "P"},
|
|
{"Direct message", model.ChannelTypeDirect, "D"},
|
|
{"Group message", model.ChannelTypeGroup, "G"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
c := &model.Channel{
|
|
Id: "channel123",
|
|
Name: "test-channel",
|
|
DisplayName: "Test Channel",
|
|
Type: tt.chType,
|
|
TeamId: "team456",
|
|
}
|
|
slim := SlimChannel(c)
|
|
assert.Equal(t, tt.expected, slim["type"])
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestToolRegistration(t *testing.T) {
|
|
tools := Tool.Tools()
|
|
assert.Len(t, tools, 5)
|
|
|
|
toolNames := make(map[string]bool)
|
|
for _, t := range tools {
|
|
toolNames[t.Tool.Name] = true
|
|
}
|
|
|
|
assert.True(t, toolNames[ListChannelsToolName], "ListChannels tool should be registered")
|
|
assert.True(t, toolNames[GetChannelByNameToolName], "GetChannelByName tool should be registered")
|
|
assert.True(t, toolNames[GetChannelInfoToolName], "GetChannelInfo tool should be registered")
|
|
assert.True(t, toolNames[GetUnreadCountToolName], "GetUnreadCount tool should be registered")
|
|
assert.True(t, toolNames[MarkChannelReadToolName], "MarkChannelRead tool should be registered")
|
|
}
|
|
|
|
func TestListChannelsFn_MissingTeamId(t *testing.T) {
|
|
req := mcp.CallToolRequest{
|
|
Params: mcp.CallToolParams{
|
|
Name: ListChannelsToolName,
|
|
Arguments: map[string]interface{}{},
|
|
},
|
|
}
|
|
|
|
result, err := ListChannelsFn(nil, req)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.True(t, result.IsError)
|
|
}
|
|
|
|
func TestListChannelsFn_ClientNotInitialized(t *testing.T) {
|
|
mattermost.SetGlobalClient(nil)
|
|
|
|
req := mcp.CallToolRequest{
|
|
Params: mcp.CallToolParams{
|
|
Name: ListChannelsToolName,
|
|
Arguments: map[string]interface{}{
|
|
"team_id": "team123",
|
|
},
|
|
},
|
|
}
|
|
|
|
result, err := ListChannelsFn(nil, req)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.True(t, result.IsError)
|
|
}
|