54 lines
1.5 KiB
Go
54 lines
1.5 KiB
Go
package guildstree
|
|
|
|
import (
|
|
"slices"
|
|
"testing"
|
|
|
|
"github.com/ayn2op/arikawa/v3/discord"
|
|
"github.com/ayn2op/tview/tree"
|
|
)
|
|
|
|
func TestModelMoveDMToFront(t *testing.T) {
|
|
t.Run("moves rendered channel to front", func(t *testing.T) {
|
|
firstID := discord.ChannelID(1)
|
|
secondID := discord.ChannelID(2)
|
|
thirdID := discord.ChannelID(3)
|
|
|
|
first := tree.NewNode("first").SetReference(firstID)
|
|
second := tree.NewNode("second").SetReference(secondID)
|
|
third := tree.NewNode("third").SetReference(thirdID)
|
|
dmRoot := tree.NewNode("Direct Messages").SetChildren([]*tree.Node{first, second, third})
|
|
m := Model{
|
|
dmRootNode: dmRoot,
|
|
nodes: map[discord.Snowflake]*tree.Node{
|
|
discord.Snowflake(firstID): first,
|
|
discord.Snowflake(secondID): second,
|
|
discord.Snowflake(thirdID): third,
|
|
},
|
|
}
|
|
|
|
m.moveDMToFront(thirdID)
|
|
|
|
want := []*tree.Node{third, first, second}
|
|
if !slices.Equal(dmRoot.Children(), want) {
|
|
t.Fatalf("children = %v, want %v", dmRoot.Children(), want)
|
|
}
|
|
})
|
|
|
|
t.Run("ignores unrendered channel", func(t *testing.T) {
|
|
firstID := discord.ChannelID(1)
|
|
first := tree.NewNode("first").SetReference(firstID)
|
|
dmRoot := tree.NewNode("Direct Messages").SetChildren([]*tree.Node{first})
|
|
m := Model{
|
|
dmRootNode: dmRoot,
|
|
nodes: map[discord.Snowflake]*tree.Node{},
|
|
}
|
|
|
|
m.moveDMToFront(discord.ChannelID(2))
|
|
|
|
children := dmRoot.Children()
|
|
if len(children) != 1 || children[0] != first {
|
|
t.Fatalf("children = %v, want the original DM node", children)
|
|
}
|
|
})
|
|
}
|