gotgbot/formatting_test.go at v2 · PaulSonOfLars/gotgbot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package gotgbot

import (
	"reflect"
	"testing"
)

func Test_entitiesToRichBlocks(t *testing.T) {
	type args struct {
		text     string
		entities []MessageEntity
	}
	tests := []struct {
		name string
		args args
		want []RichBlock
	}{
		{
			name: "no formatting",
			args: args{
				text:     "hello, this is some content",
				entities: nil,
			},
			want: []RichBlock{
				RichBlockParagraph{Text: RichTextString("hello, this is some content")},
			},
		}, {
			name: "basic italic formatting",
			args: args{
				text: "hello there",
				entities: []MessageEntity{{
					Type:   "italic",
					Offset: 0,
					Length: 5,
				}},
			},
			want: []RichBlock{
				RichBlockParagraph{Text: RichTextArray{
					RichTextItalic{Text: RichTextString("hello")},
					RichTextString(" there")},
				},
			},
		}, {
			name: "basic username formatting",
			args: args{
				text: "@hello there",
				entities: []MessageEntity{{
					Type:   "mention",
					Offset: 0,
					Length: 6,
				}},
			},
			want: []RichBlock{
				RichBlockParagraph{Text: RichTextArray{
					RichTextMention{Text: RichTextString("@hello"), Username: "hello"},
					RichTextString(" there")},
				},
			},
		}, {
			name: "single pre block",
			args: args{
				text: "hello",
				entities: []MessageEntity{{
					Type:   "pre",
					Offset: 0,
					Length: 5,
				}},
			},
			want: []RichBlock{
				RichBlockPreformatted{Text: RichTextString("hello")},
			},
		}, {
			name: "combined pre and italic",
			args: args{
				text: "hello there",
				entities: []MessageEntity{{
					Type:   "pre",
					Offset: 0,
					Length: 5,
				}, {
					Type:   "italic",
					Offset: 6,
					Length: 5,
				}},
			},
			want: []RichBlock{
				RichBlockPreformatted{Text: RichTextString("hello")},
				RichBlockParagraph{Text: RichTextArray{RichTextString(" "), RichTextItalic{Text: RichTextString("there")}}},
			},
		}, {
			name: "nested blockquote",
			args: args{
				text: "hello there you",
				entities: []MessageEntity{{
					Type:   "blockquote",
					Offset: 0,
					Length: 15,
				}, {
					Type:   "italic",
					Offset: 6,
					Length: 5,
				}},
			},
			want: []RichBlock{
				RichBlockBlockQuotation{Blocks: []RichBlock{
					RichBlockParagraph{Text: RichTextArray{
						RichTextString("hello "),
						RichTextItalic{Text: RichTextString("there")},
						RichTextString(" you"),
					}},
				}},
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := entitiesToRichBlocks(tt.args.text, tt.args.entities)
			if len(got) != len(tt.want) {
				t.Errorf("incorrect number of entities got=%d, want=%d", len(got), len(tt.want))
			}

			for idx, want := range tt.want {
				if want.GetType() != got[idx].GetType() {
					t.Errorf("incorrect type got=%s, want=%s", got[idx].GetType(), want.GetType())
				}
				if !reflect.DeepEqual(got[idx], want) {
					t.Errorf("incorrect entity got=%+v, want=%+v", got[idx], want)
				}
			}
		})
	}
}