mirror of
https://github.com/matrix-org/dendrite
synced 2024-12-14 23:32:45 +00:00
34 lines
933 B
Go
34 lines
933 B
Go
package testrig
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/nats-io/nats.go"
|
|
|
|
"github.com/matrix-org/dendrite/roomserver/api"
|
|
"github.com/matrix-org/dendrite/setup/base"
|
|
"github.com/matrix-org/dendrite/setup/jetstream"
|
|
)
|
|
|
|
func MustPublishMsgs(t *testing.T, jsctx nats.JetStreamContext, msgs ...*nats.Msg) {
|
|
t.Helper()
|
|
for _, msg := range msgs {
|
|
if _, err := jsctx.PublishMsg(msg); err != nil {
|
|
t.Fatalf("MustPublishMsgs: failed to publish message: %s", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func NewOutputEventMsg(t *testing.T, base *base.BaseDendrite, roomID string, update api.OutputEvent) *nats.Msg {
|
|
t.Helper()
|
|
msg := nats.NewMsg(base.Cfg.Global.JetStream.Prefixed(jetstream.OutputRoomEvent))
|
|
msg.Header.Set(jetstream.RoomEventType, string(update.Type))
|
|
msg.Header.Set(jetstream.RoomID, roomID)
|
|
var err error
|
|
msg.Data, err = json.Marshal(update)
|
|
if err != nil {
|
|
t.Fatalf("failed to marshal update: %s", err)
|
|
}
|
|
return msg
|
|
}
|