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) } }