package telegram import ( "testing" "time" "dev.narayana.im/narayana/telegabber/config" "dev.narayana.im/narayana/telegabber/persistence" "github.com/zelenin/go-tdlib/client" ) const testTimeFormat string = "15:04 02/01/2006" func TestOnlineStatus(t *testing.T) { c := Client{ DelayedStatuses: make(map[int64]*DelayedStatus), } show, status, presenceType := (&c).userStatusToText(client.UserStatus(&client.UserStatusOnline{}), 0) if show != "" || status != "Online" || presenceType != "" { t.Errorf("Wrong online status: %v, %v", show, status) } } func TestOnlineRecently(t *testing.T) { show, status, presenceType := (&Client{}).userStatusToText(client.UserStatus(&client.UserStatusRecently{}), 0) if show != "dnd" || status != "Last seen recently" || presenceType != "" { t.Errorf("Wrong recently status: %v, %v", show, status) } } func TestOnlineOfflineAway(t *testing.T) { location := time.FixedZone("UTC+1", 3600) timestamp := time.Now().In(location).Unix() - 3599 tm := time.Unix(timestamp, 0).In(location) c := &Client{ Session: &persistence.Session{ Timezone: "+01:00", }, } show, status, presenceType := c.userStatusToText(client.UserStatus(&client.UserStatusOffline{WasOnline: int32(timestamp)}), 0) trueStatus := "Last seen at " + tm.Format(testTimeFormat) if show != "away" || status != trueStatus || presenceType != "" { t.Errorf("Wrong away status: %v, %v, should be %v", show, status, trueStatus) } } func TestOnlineOfflineXa(t *testing.T) { timestamp := time.Now().Unix() - 3601 tm := time.Unix(timestamp, 0).UTC() c := &Client{ Session: &persistence.Session{}, } show, status, presenceType := c.userStatusToText(client.UserStatus(&client.UserStatusOffline{WasOnline: int32(timestamp)}), 0) trueStatus := "Last seen at " + tm.Format(testTimeFormat) if show != "xa" || status != trueStatus || presenceType != "" { t.Errorf("Wrong xa status: %v, %v, should be %v", show, status, trueStatus) } } func TestFormatMessageOneline(t *testing.T) { message := client.Message{ Id: 42, Content: &client.MessageText{ Text: &client.FormattedText{ Text: "tist", }, }, } text := (&Client{}).formatMessage(0, 0, true, &message) if text != "42 | | tist" { t.Errorf("Wrong oneline message formatting: %v", text) } } func TestFormatMessageMultiline(t *testing.T) { message := client.Message{ Id: 42, Content: &client.MessageText{ Text: &client.FormattedText{ Text: "tist\nziz", }, }, } text := (&Client{}).formatMessage(0, 0, true, &message) if text != "42 | | tist" { t.Errorf("Wrong multiline message formatting: %v", text) } } func TestFormatMessageOnelinePreview(t *testing.T) { message := client.Message{ Id: 42, Date: 1200000000, Content: &client.MessageText{ Text: &client.FormattedText{ Text: "tist", }, }, } c := &Client{ Session: &persistence.Session{}, } text := c.formatMessage(0, 0, false, &message) if text != "42 | | 10 Jan 2008 21:20:00 | tist" { t.Errorf("Wrong oneline preview message formatting: %v", text) } } func TestFormatMessageMultilinePreview(t *testing.T) { message := client.Message{ Id: 42, Date: 1200000000, Content: &client.MessageText{ Text: &client.FormattedText{ Text: "tist\nziz", }, }, } c := &Client{ Session: &persistence.Session{}, } text := c.formatMessage(0, 0, false, &message) if text != "42 | | 10 Jan 2008 21:20:00 | tist\nziz" { t.Errorf("Wrong multiline preview message formatting: %v", text) } } func TestFormatFile(t *testing.T) { file := client.File{ Size: 23899, Local: &client.LocalFile{ Path: "c:/Documents and Settings/blabla.jpg", }, Remote: &client.RemoteFile{ UniqueId: "tist", }, } c := Client{ content: &config.TelegramContentConfig{ Link: "localhvost", }, } content, link := c.formatFile(&file, false) if content != ". (23 kbytes) | " { t.Errorf("Wrong file label: %v", content) } if link != "" { t.Errorf("Wrong file link: %v", link) } } func TestFormatPreview(t *testing.T) { file := client.File{ Size: 23899, Local: &client.LocalFile{ Path: "c:/pron/smokovnica_vhsrip.mov", }, Remote: &client.RemoteFile{ UniqueId: "aZaZaZ", }, } c := Client{ content: &config.TelegramContentConfig{ Link: "localhvost", }, } content, link := c.formatFile(&file, true) if content != "" { t.Errorf("Wrong preview label: %v", content) } if link != "" { t.Errorf("Wrong preview link: %v", link) } } func TestMessageToTextSticker(t *testing.T) { sticker := client.Message{ Content: &client.MessageSticker{ Sticker: &client.Sticker{ Emoji: "💩", }, }, } text := (&Client{}).messageToText(&sticker, false) if text != "💩" { t.Errorf("Not poop") } } func TestMessageToTextGroup(t *testing.T) { group := client.Message{ Content: &client.MessageBasicGroupChatCreate{}, } text := (&Client{}).messageToText(&group, false) if text != "has created chat" { t.Errorf("Who created the group?") } } func TestMessageToTextSupergroup(t *testing.T) { supergroup := client.Message{ Content: &client.MessageSupergroupChatCreate{}, } text := (&Client{}).messageToText(&supergroup, false) if text != "has created chat" { t.Errorf("Who created the supergroup?") } } func TestMessageChatJoin(t *testing.T) { join := client.Message{ Content: &client.MessageChatJoinByLink{}, } text := (&Client{}).messageToText(&join, false) if text != "joined chat via invite link" { t.Errorf("Non-joined") } } func TestMessageChatAddNoMembers(t *testing.T) { add := client.Message{ Content: &client.MessageChatAddMembers{}, } text := (&Client{}).messageToText(&add, false) if text != "invited " { t.Errorf("Invited someone anyway") } } func TestMessageChatChangeTitle(t *testing.T) { title := client.Message{ Content: &client.MessageChatChangeTitle{ Title: "Anime", }, } text := (&Client{}).messageToText(&title, false) if text != "chat title set to: Anime" { t.Errorf("How to patch KDE2 for FreeBSD?") } } func TestMessageLocation(t *testing.T) { location := client.Message{ Content: &client.MessageLocation{ Location: &client.Location{ Latitude: 50.8, Longitude: 42.0167, }, }, } text := (&Client{}).messageToText(&location, false) if text != "coordinates: 50.8,42.0167 | https://www.google.com/maps/search/50.8,42.0167/" { t.Errorf("Excuse me, I'm lost") } } func TestMessagePhoto(t *testing.T) { photo := client.Message{ Content: &client.MessagePhoto{ Caption: &client.FormattedText{ Text: "tist", }, }, } text := (&Client{}).messageToText(&photo, false) if text != "tist" { t.Errorf("Wrong photo label") } } func TestMessageAudio(t *testing.T) { audio := client.Message{ Content: &client.MessageAudio{ Caption: &client.FormattedText{ Text: "tist", }, }, } text := (&Client{}).messageToText(&audio, false) if text != "tist" { t.Errorf("Wrong audio label") } } func TestMessageVideo(t *testing.T) { video := client.Message{ Content: &client.MessageVideo{ Caption: &client.FormattedText{ Text: "tist", }, }, } text := (&Client{}).messageToText(&video, false) if text != "tist" { t.Errorf("Wrong video label") } } func TestMessageDocument(t *testing.T) { document := client.Message{ Content: &client.MessageDocument{ Caption: &client.FormattedText{ Text: "tist", }, }, } text := (&Client{}).messageToText(&document, false) if text != "tist" { t.Errorf("Wrong document label") } } func TestMessageText(t *testing.T) { textMessage := client.Message{ Content: &client.MessageText{ Text: &client.FormattedText{ Text: "tist", }, }, } text := (&Client{}).messageToText(&textMessage, false) if text != "tist" { t.Errorf("Wrong text message") } } func TestMessageVoice(t *testing.T) { voice := client.Message{ Content: &client.MessageVoiceNote{ Caption: &client.FormattedText{ Text: "tist", }, }, } text := (&Client{}).messageToText(&voice, false) if text != "tist" { t.Errorf("Wrong voice label") } } func TestMessageVideoNote(t *testing.T) { videoNote := client.Message{ Content: &client.MessageVideoNote{}, } text := (&Client{}).messageToText(&videoNote, false) if text != "" { t.Errorf("Wrong video note label") } } func TestMessageAnimation(t *testing.T) { animation := client.Message{ Content: &client.MessageAnimation{ Caption: &client.FormattedText{ Text: "tist", }, }, } text := (&Client{}).messageToText(&animation, false) if text != "tist" { t.Errorf("Wrong animation label") } } func TestMessageUnknown(t *testing.T) { unknown := client.Message{ Content: &client.MessageExpiredPhoto{}, } text := (&Client{}).messageToText(&unknown, false) if text != "unknown message (messageExpiredPhoto)" { t.Errorf("Wrong label for unknown message") } } func TestMessageToPrefix1(t *testing.T) { message := client.Message{ Id: 42, IsOutgoing: true, ForwardInfo: &client.MessageForwardInfo{ Origin: &client.MessageForwardOriginHiddenUser{ SenderName: "ziz", }, }, } prefix := (&Client{Session: &persistence.Session{}}).messageToPrefix(&message, "", "", nil) if prefix != "➡ 42 | fwd: ziz" { t.Errorf("Wrong prefix: %v", prefix) } } func TestMessageToPrefix2(t *testing.T) { message := client.Message{ Id: 56, ForwardInfo: &client.MessageForwardInfo{ Origin: &client.MessageForwardOriginChannel{ AuthorSignature: "zaz", }, }, } prefix := (&Client{Session: &persistence.Session{}}).messageToPrefix(&message, "y.jpg", "", nil) if prefix != "⬅ 56 | fwd: (zaz) | preview: y.jpg" { t.Errorf("Wrong prefix: %v", prefix) } } func TestMessageToPrefix3(t *testing.T) { message := client.Message{ Id: 56, ForwardInfo: &client.MessageForwardInfo{ Origin: &client.MessageForwardOriginChannel{ AuthorSignature: "zuz", }, }, } prefix := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "a.jpg", nil) if prefix != "< 56 | fwd: (zuz) | file: a.jpg" { t.Errorf("Wrong prefix: %v", prefix) } } func TestMessageToPrefix4(t *testing.T) { message := client.Message{ Id: 23, IsOutgoing: true, } prefix := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "", nil) if prefix != "> 23" { t.Errorf("Wrong prefix: %v", prefix) } } func TestMessageToPrefix5(t *testing.T) { message := client.Message{ Id: 560, ForwardInfo: &client.MessageForwardInfo{ Origin: &client.MessageForwardOriginChat{ AuthorSignature: "zyz", }, }, } prefix := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "h.jpg", "a.jpg", nil) if prefix != "< 560 | fwd: (zyz) | preview: h.jpg | file: a.jpg" { t.Errorf("Wrong prefix: %v", prefix) } } func TestMessageToPrefix6(t *testing.T) { message := client.Message{ Id: 23, IsOutgoing: true, ReplyToMessageId: 42, } reply := client.Message{ Id: 42, Content: &client.MessageText{ Text: &client.FormattedText{ Text: "tist", }, }, } prefix := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "", &reply) if prefix != "> 23 | reply: 42 | | tist" { t.Errorf("Wrong prefix: %v", prefix) } } func GetSenderIdEmpty(t *testing.T) { message := client.Message{} senderId := (&Client{}).getSenderId(&message) if senderId != 0 { t.Errorf("Wrong sender id: %v", senderId) } } func GetSenderIdUser(t *testing.T) { message := client.Message{ SenderId: &client.MessageSenderUser{ UserId: 42, }, } senderId := (&Client{}).getSenderId(&message) if senderId != 42 { t.Errorf("Wrong sender id: %v", senderId) } } func GetSenderIdChat(t *testing.T) { message := client.Message{ SenderId: &client.MessageSenderChat{ ChatId: -42, }, } senderId := (&Client{}).getSenderId(&message) if senderId != -42 { t.Errorf("Wrong sender id: %v", senderId) } }