discodoc/doc_test.go

91 lines
2.0 KiB
Go

package main
import "testing"
func TestParseQuery(t *testing.T) {
cases := []struct {
name string
query string
module string
parts []string
}{
{
name: "stdlib basic",
query: "strings",
module: "strings",
parts: nil,
},
{
name: "stdlib type",
query: "strings.Split",
module: "strings",
parts: []string{"split"},
},
{
name: "stdlib method",
query: "strings.Builder.Grow",
module: "strings",
parts: []string{"builder", "grow"},
},
{
name: "stdlib redirect basic",
query: "json",
module: "encoding/json",
parts: nil,
},
{
name: "stdlib redirect type",
query: "json.Unmarshal",
module: "encoding/json",
parts: []string{"unmarshal"},
},
{
name: "stdlib redirect method",
query: "json.NewDecoder.Decode",
module: "encoding/json",
parts: []string{"newdecoder", "decode"},
},
{
name: "custom basic",
query: "github.com/golang/go",
module: "github.com/golang/go",
parts: nil,
},
{
name: "custom type",
query: "github.com/bwmarrin/discordgo.Session",
module: "github.com/bwmarrin/discordgo",
parts: []string{"session"},
},
{
name: "custom method",
query: "github.com/bwmarrin/discordgo.Session.AddHandler",
module: "github.com/bwmarrin/discordgo",
parts: []string{"session", "addhandler"},
},
{
name: "custom method with space",
query: "github.com/bwmarrin/discordgo Session AddHandler",
module: "github.com/bwmarrin/discordgo",
parts: []string{"session", "addhandler"},
},
}
for _, c := range cases {
t.Run(c.name, func(t *testing.T) {
module, parts := parseQuery(c.query)
if module != c.module {
t.Errorf("INVALID MODULE:\nGOT:%s\nEXPECTED:%s", module, c.module)
}
if len(parts) != len(c.parts) {
t.Errorf("INVALID PARTS:\nGOT:%v\nEXPECTED:%v", parts, c.parts)
}
for i, part := range parts {
if part != c.parts[i] {
t.Errorf("INVALID PARTS(%d):\nGOT:%v\nEXPECTED:%v", i, part, c.parts[i])
}
}
})
}
}