mirror of
https://github.com/sartoopjj/thefeed.git
synced 2026-05-19 11:24:36 +03:00
Sarto
103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
package server
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/sartoopjj/thefeed/internal/protocol"
|
|
)
|
|
|
|
func TestParsePublicMessages(t *testing.T) {
|
|
body := []byte(`
|
|
<html><body>
|
|
<div class="tgme_widget_message" data-post="testchan/101">
|
|
<a class="tgme_widget_message_date"><time datetime="2026-03-30T04:45:00+00:00"></time></a>
|
|
<div class="tgme_widget_message_text">hello<br/>world</div>
|
|
</div>
|
|
<div class="tgme_widget_message" data-post="testchan/105">
|
|
<a class="tgme_widget_message_date"><time datetime="2026-03-30T04:50:00+00:00"></time></a>
|
|
<div class="tgme_widget_message_photo_wrap"></div>
|
|
</div>
|
|
<div class="tgme_widget_message" data-post="testchan/106">
|
|
<a class="tgme_widget_message_date"><time datetime="2026-03-30T04:51:00+00:00"></time></a>
|
|
<a class="tgme_widget_message_photo_wrap" href="https://t.me/testchan/106"></a>
|
|
<div class="tgme_widget_message_text">photo caption</div>
|
|
</div>
|
|
</body></html>
|
|
`)
|
|
|
|
msgs, err := parsePublicMessages(body)
|
|
if err != nil {
|
|
t.Fatalf("parsePublicMessages: %v", err)
|
|
}
|
|
if len(msgs) != 3 {
|
|
t.Fatalf("len(msgs) = %d, want 3", len(msgs))
|
|
}
|
|
// Photo with caption (newest first)
|
|
if msgs[0].ID != 106 {
|
|
t.Fatalf("msgs[0].ID = %d, want 106", msgs[0].ID)
|
|
}
|
|
want := protocol.MediaImage + "\n" + "photo caption"
|
|
if msgs[0].Text != want {
|
|
t.Fatalf("msgs[0].Text = %q, want %q", msgs[0].Text, want)
|
|
}
|
|
// Photo only
|
|
if msgs[1].ID != 105 {
|
|
t.Fatalf("msgs[1].ID = %d, want 105", msgs[1].ID)
|
|
}
|
|
if msgs[1].Text != protocol.MediaImage {
|
|
t.Fatalf("msgs[1].Text = %q, want %q", msgs[1].Text, protocol.MediaImage)
|
|
}
|
|
// Text only
|
|
if msgs[2].ID != 101 {
|
|
t.Fatalf("msgs[2].ID = %d, want 101", msgs[2].ID)
|
|
}
|
|
if msgs[2].Text != "hello\nworld" {
|
|
t.Fatalf("msgs[2].Text = %q, want %q", msgs[2].Text, "hello\nworld")
|
|
}
|
|
}
|
|
|
|
func TestParsePublicMessagesNoLimit(t *testing.T) {
|
|
body := []byte(`
|
|
<html><body>
|
|
<div class="tgme_widget_message" data-post="testchan/1"><div class="tgme_widget_message_text">one</div></div>
|
|
<div class="tgme_widget_message" data-post="testchan/2"><div class="tgme_widget_message_text">two</div></div>
|
|
<div class="tgme_widget_message" data-post="testchan/3"><div class="tgme_widget_message_text">three</div></div>
|
|
</body></html>
|
|
`)
|
|
|
|
msgs, err := parsePublicMessages(body)
|
|
if err != nil {
|
|
t.Fatalf("parsePublicMessages: %v", err)
|
|
}
|
|
if len(msgs) != 3 {
|
|
t.Fatalf("len(msgs) = %d, want 3", len(msgs))
|
|
}
|
|
if msgs[0].ID != 3 || msgs[1].ID != 2 || msgs[2].ID != 1 {
|
|
t.Fatalf("got ids %d,%d,%d want 3,2,1", msgs[0].ID, msgs[1].ID, msgs[2].ID)
|
|
}
|
|
}
|
|
|
|
func TestMergeMessages(t *testing.T) {
|
|
old := []protocol.Message{
|
|
{ID: 100, Timestamp: 1000, Text: "old100"},
|
|
{ID: 99, Timestamp: 999, Text: "old99"},
|
|
}
|
|
newMsgs := []protocol.Message{
|
|
{ID: 101, Timestamp: 1001, Text: "new101"},
|
|
{ID: 100, Timestamp: 1000, Text: "edited100"},
|
|
}
|
|
merged := mergeMessages(old, newMsgs)
|
|
if len(merged) != 3 {
|
|
t.Fatalf("len(merged) = %d, want 3", len(merged))
|
|
}
|
|
if merged[0].ID != 101 {
|
|
t.Fatalf("merged[0].ID = %d, want 101", merged[0].ID)
|
|
}
|
|
if merged[1].Text != "edited100" {
|
|
t.Fatalf("merged[1].Text = %q, want edited100", merged[1].Text)
|
|
}
|
|
if merged[2].ID != 99 {
|
|
t.Fatalf("merged[2].ID = %d, want 99", merged[2].ID)
|
|
}
|
|
}
|