types_test.go (view raw)
1package tgbotapi
2
3import (
4 "testing"
5 "time"
6)
7
8func TestUserStringWith(t *testing.T) {
9 user := User{
10 ID: 0,
11 FirstName: "Test",
12 LastName: "Test",
13 UserName: "",
14 LanguageCode: "en",
15 IsBot: false,
16 }
17
18 if user.String() != "Test Test" {
19 t.Fail()
20 }
21}
22
23func TestUserStringWithUserName(t *testing.T) {
24 user := User{
25 ID: 0,
26 FirstName: "Test",
27 LastName: "Test",
28 UserName: "@test",
29 LanguageCode: "en",
30 }
31
32 if user.String() != "@test" {
33 t.Fail()
34 }
35}
36
37func TestMessageTime(t *testing.T) {
38 message := Message{Date: 0}
39
40 date := time.Unix(0, 0)
41 if message.Time() != date {
42 t.Fail()
43 }
44}
45
46func TestMessageIsCommandWithCommand(t *testing.T) {
47 message := Message{Text: "/command"}
48 message.Entities = []MessageEntity{{Type: "bot_command", Offset: 0, Length: 8}}
49
50 if !message.IsCommand() {
51 t.Fail()
52 }
53}
54
55func TestIsCommandWithText(t *testing.T) {
56 message := Message{Text: "some text"}
57
58 if message.IsCommand() {
59 t.Fail()
60 }
61}
62
63func TestIsCommandWithEmptyText(t *testing.T) {
64 message := Message{Text: ""}
65
66 if message.IsCommand() {
67 t.Fail()
68 }
69}
70
71func TestCommandWithCommand(t *testing.T) {
72 message := Message{Text: "/command"}
73 message.Entities = []MessageEntity{{Type: "bot_command", Offset: 0, Length: 8}}
74
75 if message.Command() != "command" {
76 t.Fail()
77 }
78}
79
80func TestCommandWithEmptyText(t *testing.T) {
81 message := Message{Text: ""}
82
83 if message.Command() != "" {
84 t.Fail()
85 }
86}
87
88func TestCommandWithNonCommand(t *testing.T) {
89 message := Message{Text: "test text"}
90
91 if message.Command() != "" {
92 t.Fail()
93 }
94}
95
96func TestCommandWithBotName(t *testing.T) {
97 message := Message{Text: "/command@testbot"}
98 message.Entities = []MessageEntity{{Type: "bot_command", Offset: 0, Length: 16}}
99
100 if message.Command() != "command" {
101 t.Fail()
102 }
103}
104
105func TestCommandWithAtWithBotName(t *testing.T) {
106 message := Message{Text: "/command@testbot"}
107 message.Entities = []MessageEntity{{Type: "bot_command", Offset: 0, Length: 16}}
108
109 if message.CommandWithAt() != "command@testbot" {
110 t.Fail()
111 }
112}
113
114func TestMessageCommandArgumentsWithArguments(t *testing.T) {
115 message := Message{Text: "/command with arguments"}
116 message.Entities = []MessageEntity{{Type: "bot_command", Offset: 0, Length: 8}}
117 if message.CommandArguments() != "with arguments" {
118 t.Fail()
119 }
120}
121
122func TestMessageCommandArgumentsWithMalformedArguments(t *testing.T) {
123 message := Message{Text: "/command-without argument space"}
124 message.Entities = []MessageEntity{{Type: "bot_command", Offset: 0, Length: 8}}
125 if message.CommandArguments() != "without argument space" {
126 t.Fail()
127 }
128}
129
130func TestMessageCommandArgumentsWithoutArguments(t *testing.T) {
131 message := Message{Text: "/command"}
132 if message.CommandArguments() != "" {
133 t.Fail()
134 }
135}
136
137func TestMessageCommandArgumentsForNonCommand(t *testing.T) {
138 message := Message{Text: "test text"}
139 if message.CommandArguments() != "" {
140 t.Fail()
141 }
142}
143
144func TestMessageEntityParseURLGood(t *testing.T) {
145 entity := MessageEntity{URL: "https://www.google.com"}
146
147 if _, err := entity.ParseURL(); err != nil {
148 t.Fail()
149 }
150}
151
152func TestMessageEntityParseURLBad(t *testing.T) {
153 entity := MessageEntity{URL: ""}
154
155 if _, err := entity.ParseURL(); err == nil {
156 t.Fail()
157 }
158}
159
160func TestChatIsPrivate(t *testing.T) {
161 chat := Chat{ID: 10, Type: "private"}
162
163 if !chat.IsPrivate() {
164 t.Fail()
165 }
166}
167
168func TestChatIsGroup(t *testing.T) {
169 chat := Chat{ID: 10, Type: "group"}
170
171 if !chat.IsGroup() {
172 t.Fail()
173 }
174}
175
176func TestChatIsChannel(t *testing.T) {
177 chat := Chat{ID: 10, Type: "channel"}
178
179 if !chat.IsChannel() {
180 t.Fail()
181 }
182}
183
184func TestChatIsSuperGroup(t *testing.T) {
185 chat := Chat{ID: 10, Type: "supergroup"}
186
187 if !chat.IsSuperGroup() {
188 t.Fail()
189 }
190}
191
192func TestMessageEntityIsMention(t *testing.T) {
193 entity := MessageEntity{Type: "mention"}
194
195 if !entity.IsMention() {
196 t.Fail()
197 }
198}
199
200func TestMessageEntityIsHashtag(t *testing.T) {
201 entity := MessageEntity{Type: "hashtag"}
202
203 if !entity.IsHashtag() {
204 t.Fail()
205 }
206}
207
208func TestMessageEntityIsBotCommand(t *testing.T) {
209 entity := MessageEntity{Type: "bot_command"}
210
211 if !entity.IsCommand() {
212 t.Fail()
213 }
214}
215
216func TestMessageEntityIsUrl(t *testing.T) {
217 entity := MessageEntity{Type: "url"}
218
219 if !entity.IsURL() {
220 t.Fail()
221 }
222}
223
224func TestMessageEntityIsEmail(t *testing.T) {
225 entity := MessageEntity{Type: "email"}
226
227 if !entity.IsEmail() {
228 t.Fail()
229 }
230}
231
232func TestMessageEntityIsBold(t *testing.T) {
233 entity := MessageEntity{Type: "bold"}
234
235 if !entity.IsBold() {
236 t.Fail()
237 }
238}
239
240func TestMessageEntityIsItalic(t *testing.T) {
241 entity := MessageEntity{Type: "italic"}
242
243 if !entity.IsItalic() {
244 t.Fail()
245 }
246}
247
248func TestMessageEntityIsCode(t *testing.T) {
249 entity := MessageEntity{Type: "code"}
250
251 if !entity.IsCode() {
252 t.Fail()
253 }
254}
255
256func TestMessageEntityIsPre(t *testing.T) {
257 entity := MessageEntity{Type: "pre"}
258
259 if !entity.IsPre() {
260 t.Fail()
261 }
262}
263
264func TestMessageEntityIsTextLink(t *testing.T) {
265 entity := MessageEntity{Type: "text_link"}
266
267 if !entity.IsTextLink() {
268 t.Fail()
269 }
270}
271
272func TestFileLink(t *testing.T) {
273 file := File{FilePath: "test/test.txt"}
274
275 if file.Link("token") != "https://api.telegram.org/file/bottoken/test/test.txt" {
276 t.Fail()
277 }
278}
279
280// Ensure all configs are sendable
281var (
282 _ Chattable = AnimationConfig{}
283 _ Chattable = AudioConfig{}
284 _ Chattable = BanChatMemberConfig{}
285 _ Chattable = CallbackConfig{}
286 _ Chattable = ChatActionConfig{}
287 _ Chattable = ChatAdministratorsConfig{}
288 _ Chattable = ChatInfoConfig{}
289 _ Chattable = ChatInviteLinkConfig{}
290 _ Chattable = CloseConfig{}
291 _ Chattable = ContactConfig{}
292 _ Chattable = CopyMessageConfig{}
293 _ Chattable = CreateChatInviteLinkConfig{}
294 _ Chattable = DeleteChatPhotoConfig{}
295 _ Chattable = DeleteChatStickerSetConfig{}
296 _ Chattable = DeleteMessageConfig{}
297 _ Chattable = DeleteMyCommandsConfig{}
298 _ Chattable = DeleteWebhookConfig{}
299 _ Chattable = DocumentConfig{}
300 _ Chattable = EditChatInviteLinkConfig{}
301 _ Chattable = EditMessageCaptionConfig{}
302 _ Chattable = EditMessageLiveLocationConfig{}
303 _ Chattable = EditMessageMediaConfig{}
304 _ Chattable = EditMessageReplyMarkupConfig{}
305 _ Chattable = EditMessageTextConfig{}
306 _ Chattable = FileConfig{}
307 _ Chattable = ForwardConfig{}
308 _ Chattable = GameConfig{}
309 _ Chattable = GetChatMemberConfig{}
310 _ Chattable = GetGameHighScoresConfig{}
311 _ Chattable = InlineConfig{}
312 _ Chattable = InvoiceConfig{}
313 _ Chattable = KickChatMemberConfig{}
314 _ Chattable = LeaveChatConfig{}
315 _ Chattable = LocationConfig{}
316 _ Chattable = LogOutConfig{}
317 _ Chattable = MediaGroupConfig{}
318 _ Chattable = MessageConfig{}
319 _ Chattable = PhotoConfig{}
320 _ Chattable = PinChatMessageConfig{}
321 _ Chattable = PreCheckoutConfig{}
322 _ Chattable = PromoteChatMemberConfig{}
323 _ Chattable = RestrictChatMemberConfig{}
324 _ Chattable = RevokeChatInviteLinkConfig{}
325 _ Chattable = SendPollConfig{}
326 _ Chattable = SetChatDescriptionConfig{}
327 _ Chattable = SetChatPhotoConfig{}
328 _ Chattable = SetChatTitleConfig{}
329 _ Chattable = SetGameScoreConfig{}
330 _ Chattable = ShippingConfig{}
331 _ Chattable = StickerConfig{}
332 _ Chattable = StopMessageLiveLocationConfig{}
333 _ Chattable = StopPollConfig{}
334 _ Chattable = UnbanChatMemberConfig{}
335 _ Chattable = UnpinChatMessageConfig{}
336 _ Chattable = UpdateConfig{}
337 _ Chattable = UserProfilePhotosConfig{}
338 _ Chattable = VenueConfig{}
339 _ Chattable = VideoConfig{}
340 _ Chattable = VideoNoteConfig{}
341 _ Chattable = VoiceConfig{}
342 _ Chattable = WebhookConfig{}
343)
344
345// Ensure all Fileable types are correct.
346var (
347 _ Fileable = (*PhotoConfig)(nil)
348 _ Fileable = (*AudioConfig)(nil)
349 _ Fileable = (*DocumentConfig)(nil)
350 _ Fileable = (*StickerConfig)(nil)
351 _ Fileable = (*VideoConfig)(nil)
352 _ Fileable = (*AnimationConfig)(nil)
353 _ Fileable = (*VideoNoteConfig)(nil)
354 _ Fileable = (*VoiceConfig)(nil)
355 _ Fileable = (*SetChatPhotoConfig)(nil)
356 _ Fileable = (*EditMessageMediaConfig)(nil)
357 _ Fileable = (*SetChatPhotoConfig)(nil)
358 _ Fileable = (*UploadStickerConfig)(nil)
359 _ Fileable = (*NewStickerSetConfig)(nil)
360 _ Fileable = (*AddStickerConfig)(nil)
361 _ Fileable = (*MediaGroupConfig)(nil)
362 _ Fileable = (*WebhookConfig)(nil)
363 _ Fileable = (*SetStickerSetThumbConfig)(nil)
364)
365
366// Ensure all RequestFileData types are correct.
367var (
368 _ RequestFileData = (*FilePath)(nil)
369 _ RequestFileData = (*FileBytes)(nil)
370 _ RequestFileData = (*FileReader)(nil)
371 _ RequestFileData = (*FileURL)(nil)
372 _ RequestFileData = (*FileID)(nil)
373 _ RequestFileData = (*fileAttach)(nil)
374)