67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
package discord
|
|
|
|
import "testing"
|
|
|
|
func TestWrapTablesInCodeBlocks(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
in string
|
|
want string
|
|
}{
|
|
{
|
|
name: "no table",
|
|
in: "hello world\nno tables here",
|
|
want: "hello world\nno tables here",
|
|
},
|
|
{
|
|
name: "simple table",
|
|
in: "before\n| a | b |\n| - | - |\n| 1 | 2 |\nafter",
|
|
want: "before\n```\n| a | b |\n| - | - |\n| 1 | 2 |\n```\nafter",
|
|
},
|
|
{
|
|
name: "table already in code block",
|
|
in: "```\n| a | b |\n| 1 | 2 |\n```",
|
|
want: "```\n| a | b |\n| 1 | 2 |\n```",
|
|
},
|
|
{
|
|
name: "table at end of content with separator",
|
|
in: "text\n| x | y |\n| --- | --- |\n| 1 | 2 |",
|
|
want: "text\n```\n| x | y |\n| --- | --- |\n| 1 | 2 |\n```",
|
|
},
|
|
{
|
|
name: "pipe rows without separator not wrapped",
|
|
in: "text\n| x | y |\n| 1 | 2 |",
|
|
want: "text\n| x | y |\n| 1 | 2 |",
|
|
},
|
|
{
|
|
name: "multiple tables with separators",
|
|
in: "| a | b |\n| - | - |\n| 1 | 2 |\n\ntext\n| c | d |\n| --- | --- |\n| 3 | 4 |",
|
|
want: "```\n| a | b |\n| - | - |\n| 1 | 2 |\n```\n\ntext\n```\n| c | d |\n| --- | --- |\n| 3 | 4 |\n```",
|
|
},
|
|
{
|
|
name: "pipe in regular text not treated as table",
|
|
in: "use | for OR operations",
|
|
want: "use | for OR operations",
|
|
},
|
|
{
|
|
name: "table with code block after",
|
|
in: "| a | b |\n| - | - |\n| 1 | 2 |\n```go\nfmt.Println()\n```",
|
|
want: "```\n| a | b |\n| - | - |\n| 1 | 2 |\n```\n```go\nfmt.Println()\n```",
|
|
},
|
|
{
|
|
name: "aligned separator with colons",
|
|
in: "| left | center | right |\n| :--- | :---: | ---: |\n| a | b | c |",
|
|
want: "```\n| left | center | right |\n| :--- | :---: | ---: |\n| a | b | c |\n```",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := wrapTablesInCodeBlocks(tt.in)
|
|
if got != tt.want {
|
|
t.Errorf("wrapTablesInCodeBlocks():\n got: %q\nwant: %q", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|