mirror of
https://github.com/1f349/dendrite.git
synced 2024-11-22 03:31:41 +00:00
f5b3144dc3
Requires https://github.com/matrix-org/gomatrixserverlib/pull/376 This has numerous upsides: - Less type casting to `*Event` is required. - Making Dendrite work with `PDU` interfaces means we can swap out Event impls more easily. - Tests which represent weird event shapes are easier to write. Part of a series of refactors on GMSL.
72 lines
2.6 KiB
Go
72 lines
2.6 KiB
Go
package input_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/matrix-org/dendrite/internal/caching"
|
|
"github.com/matrix-org/dendrite/internal/sqlutil"
|
|
"github.com/matrix-org/dendrite/roomserver"
|
|
"github.com/matrix-org/dendrite/roomserver/api"
|
|
"github.com/matrix-org/dendrite/roomserver/internal/input"
|
|
"github.com/matrix-org/dendrite/roomserver/types"
|
|
"github.com/matrix-org/dendrite/setup/jetstream"
|
|
"github.com/matrix-org/dendrite/test"
|
|
"github.com/matrix-org/dendrite/test/testrig"
|
|
"github.com/matrix-org/gomatrixserverlib"
|
|
)
|
|
|
|
func TestSingleTransactionOnInput(t *testing.T) {
|
|
test.WithAllDatabases(t, func(t *testing.T, dbType test.DBType) {
|
|
cfg, processCtx, close := testrig.CreateConfig(t, dbType)
|
|
defer close()
|
|
cm := sqlutil.NewConnectionManager(processCtx, cfg.Global.DatabaseOptions)
|
|
|
|
natsInstance := &jetstream.NATSInstance{}
|
|
js, jc := natsInstance.Prepare(processCtx, &cfg.Global.JetStream)
|
|
caches := caching.NewRistrettoCache(8*1024*1024, time.Hour, caching.DisableMetrics)
|
|
rsAPI := roomserver.NewInternalAPI(processCtx, cfg, cm, natsInstance, caches, caching.DisableMetrics)
|
|
rsAPI.SetFederationAPI(nil, nil)
|
|
|
|
deadline, _ := t.Deadline()
|
|
if max := time.Now().Add(time.Second * 3); deadline.Before(max) {
|
|
deadline = max
|
|
}
|
|
ctx, cancel := context.WithDeadline(processCtx.Context(), deadline)
|
|
defer cancel()
|
|
|
|
event, err := gomatrixserverlib.MustGetRoomVersion(gomatrixserverlib.RoomVersionV6).NewEventFromTrustedJSON(
|
|
[]byte(`{"auth_events":[],"content":{"creator":"@neilalexander:dendrite.matrix.org","room_version":"6"},"depth":1,"hashes":{"sha256":"jqOqdNEH5r0NiN3xJtj0u5XUVmRqq9YvGbki1wxxuuM"},"origin":"dendrite.matrix.org","origin_server_ts":1644595362726,"prev_events":[],"prev_state":[],"room_id":"!jSZZRknA6GkTBXNP:dendrite.matrix.org","sender":"@neilalexander:dendrite.matrix.org","signatures":{"dendrite.matrix.org":{"ed25519:6jB2aB":"bsQXO1wketf1OSe9xlndDIWe71W9KIundc6rBw4KEZdGPW7x4Tv4zDWWvbxDsG64sS2IPWfIm+J0OOozbrWIDw"}},"state_key":"","type":"m.room.create"}`),
|
|
false,
|
|
)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
in := api.InputRoomEvent{
|
|
Kind: api.KindOutlier, // don't panic if we generate an output event
|
|
Event: &types.HeaderedEvent{PDU: event},
|
|
}
|
|
|
|
inputter := &input.Inputer{
|
|
JetStream: js,
|
|
NATSClient: jc,
|
|
Cfg: &cfg.RoomServer,
|
|
}
|
|
res := &api.InputRoomEventsResponse{}
|
|
inputter.InputRoomEvents(
|
|
ctx,
|
|
&api.InputRoomEventsRequest{
|
|
InputRoomEvents: []api.InputRoomEvent{in},
|
|
Asynchronous: false,
|
|
},
|
|
res,
|
|
)
|
|
// If we fail here then it's because we've hit the test deadline,
|
|
// so we probably deadlocked
|
|
if err := res.Err(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
}
|