51 lines
931 B
Go
51 lines
931 B
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
func TestDBChannel_SaveMessage(t *testing.T) {
|
|
dsn := "postgresql://postgres:xixi0202@10.1.1.3:5432/telegram_resource?sslmode=disable"
|
|
if err := InitDB(dsn); err != nil {
|
|
t.Error(err)
|
|
}
|
|
db.Exec(`truncate channel_messages`)
|
|
|
|
msg := &ChannelMessage{
|
|
ID: 1,
|
|
GroupID: 1,
|
|
Message: "Hello",
|
|
Medias: ChannelMessageMedia{
|
|
MsgID: 1,
|
|
AssetID: 1,
|
|
Photo: lo.ToPtr("photo"),
|
|
},
|
|
PublishAt: time.Now(),
|
|
}
|
|
|
|
msg1 := &ChannelMessage{
|
|
ID: 1,
|
|
GroupID: 1,
|
|
Message: "Hello",
|
|
Medias: ChannelMessageMedia{
|
|
MsgID: 2,
|
|
AssetID: 3,
|
|
Photo: lo.ToPtr("Man"),
|
|
},
|
|
PublishAt: time.Now(),
|
|
}
|
|
|
|
c := NewDBChannel(123, "test", "hello")
|
|
if err := c.SaveMessage(context.Background(), msg); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := c.SaveMessage(context.Background(), msg1); err != nil {
|
|
t.Logf("%+v", err)
|
|
}
|
|
}
|