mirror of
https://github.com/1f349/dendrite.git
synced 2024-12-23 08:44:11 +00:00
236b16aa6c
* Add very basic syncapi tests * Add a way to inject jetstream messages * implement add_state_ids * bugfixes * Unbreak tests * Remove now un-needed API call * Linting
36 lines
906 B
Go
36 lines
906 B
Go
package test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/matrix-org/dendrite/roomserver/api"
|
|
"github.com/matrix-org/dendrite/setup/base"
|
|
"github.com/matrix-org/dendrite/setup/jetstream"
|
|
"github.com/nats-io/nats.go"
|
|
)
|
|
|
|
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.Msg{
|
|
Subject: base.Cfg.Global.JetStream.Prefixed(jetstream.OutputRoomEvent),
|
|
Header: nats.Header{},
|
|
}
|
|
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
|
|
}
|