telegabber/telegram/formatter/formatter_test.go

209 lines
4.7 KiB
Go
Raw Normal View History

package formatter
import (
"testing"
"github.com/zelenin/go-tdlib/client"
)
func TestNoFormatting(t *testing.T) {
markup := Format("abc\ndef", []*client.TextEntity{}, EntityToMarkdown)
if markup != "abc\ndef" {
t.Errorf("No formatting expected, but: %v", markup)
}
}
func TestFormattingSimple(t *testing.T) {
markup := Format("👙🐧🐖", []*client.TextEntity{
&client.TextEntity{
Offset: 2,
Length: 4,
Type: &client.TextEntityTypeBold{},
},
}, EntityToMarkdown)
if markup != "👙**🐧🐖**" {
t.Errorf("Wrong simple formatting: %v", markup)
}
}
func TestFormattingAdjacent(t *testing.T) {
markup := Format("a👙🐧🐖", []*client.TextEntity{
&client.TextEntity{
Offset: 3,
Length: 2,
Type: &client.TextEntityTypeItalic{},
},
&client.TextEntity{
Offset: 5,
Length: 2,
Type: &client.TextEntityTypeTextUrl{
Url: "https://narayana.im/",
},
},
}, EntityToMarkdown)
if markup != "a👙_🐧_[🐖](https://narayana.im/)" {
t.Errorf("Wrong adjacent formatting: %v", markup)
}
}
func TestFormattingAdjacentAndNested(t *testing.T) {
markup := Format("👙🐧🐖", []*client.TextEntity{
&client.TextEntity{
Offset: 0,
Length: 4,
Type: &client.TextEntityTypePre{},
},
&client.TextEntity{
Offset: 0,
Length: 2,
Type: &client.TextEntityTypeBold{},
},
&client.TextEntity{
Offset: 4,
Length: 2,
Type: &client.TextEntityTypeItalic{},
},
}, EntityToMarkdown)
if markup != "\n```\n**👙**🐧\n```\n_🐖_" {
t.Errorf("Wrong adjacent&nested formatting: %v", markup)
}
}
func TestRebalanceTwoZero(t *testing.T) {
s1 := InsertionStack{
&Insertion{Offset: 7},
&Insertion{Offset: 8},
}
s2 := InsertionStack{}
s1, s2 = s1.rebalance(s2, 7)
if !(len(s1) == 2 && len(s2) == 0 && s1[0].Offset == 7 && s1[1].Offset == 8) {
t.Errorf("Wrong rebalance 20: %#v %#v", s1, s2)
}
}
func TestRebalanceNeeded(t *testing.T) {
s1 := InsertionStack{
&Insertion{Offset: 7},
&Insertion{Offset: 8},
}
s2 := InsertionStack{
&Insertion{Offset: 10},
&Insertion{Offset: 9},
}
s1, s2 = s1.rebalance(s2, 9)
if !(len(s1) == 3 && len(s2) == 1 &&
s1[0].Offset == 7 && s1[1].Offset == 8 && s1[2].Offset == 9 &&
s2[0].Offset == 10) {
t.Errorf("Wrong rebalance when needed: %#v %#v", s1, s2)
}
}
func TestRebalanceNotNeeded(t *testing.T) {
s1 := InsertionStack{
&Insertion{Offset: 7},
&Insertion{Offset: 8},
}
s2 := InsertionStack{
&Insertion{Offset: 10},
&Insertion{Offset: 9},
}
s1, s2 = s1.rebalance(s2, 8)
if !(len(s1) == 2 && len(s2) == 2 &&
s1[0].Offset == 7 && s1[1].Offset == 8 &&
s2[0].Offset == 10 && s2[1].Offset == 9) {
t.Errorf("Wrong rebalance when not needed: %#v %#v", s1, s2)
}
}
func TestRebalanceLate(t *testing.T) {
s1 := InsertionStack{
&Insertion{Offset: 7},
&Insertion{Offset: 8},
}
s2 := InsertionStack{
&Insertion{Offset: 10},
&Insertion{Offset: 9},
}
s1, s2 = s1.rebalance(s2, 10)
if !(len(s1) == 4 && len(s2) == 0 &&
s1[0].Offset == 7 && s1[1].Offset == 8 &&
s1[2].Offset == 9 && s1[3].Offset == 10) {
t.Errorf("Wrong rebalance when late: %#v %#v", s1, s2)
}
}
func TestIteratorEmpty(t *testing.T) {
s := InsertionStack{}
g := s.NewIterator()
v := g()
if v != nil {
t.Errorf("Empty iterator should return nil but returned %#v", v)
}
}
func TestIterator(t *testing.T) {
s := InsertionStack{
&Insertion{Offset: 7},
&Insertion{Offset: 8},
}
g := s.NewIterator()
v := g()
if v == nil || v.Offset != 7 {
t.Errorf("Wrong insertion instead of 7: %#v", v)
}
v = g()
if v == nil || v.Offset != 8 {
t.Errorf("Wrong insertion instead of 8: %#v", v)
}
v = g()
if v != nil {
t.Errorf("nil should be returned after end, %#v instead", v)
}
v = g()
if v != nil {
t.Errorf("Further attempts should return nil too, %#v instead", v)
}
}
func TestSortEntities(t *testing.T) {
entities := []*client.TextEntity{
&client.TextEntity{
Offset: 3,
Length: 2,
},
&client.TextEntity{
Offset: 5,
Length: 2,
},
&client.TextEntity{
Offset: 7,
Length: 2,
},
&client.TextEntity{
Offset: 6,
Length: 1,
},
&client.TextEntity{
Offset: 5,
Length: 1,
},
}
entities = SortEntities(entities)
if !(len(entities) == 5 &&
entities[0].Offset == 3 && entities[0].Length == 2 &&
entities[1].Offset == 5 && entities[1].Length == 2 &&
entities[2].Offset == 5 && entities[2].Length == 1 &&
entities[3].Offset == 6 && entities[3].Length == 1 &&
entities[4].Offset == 7 && entities[4].Length == 2) {
t.Errorf("Wrong sorting order: %#v", entities)
}
}
func TestSortEmpty(t *testing.T) {
entities := []*client.TextEntity{}
entities = SortEntities(entities)
if len(entities) != 0 {
t.Errorf("Empty entities set sorting error: %#v", entities)
}
}