dendrite/roomserver/internal/input/input_test.go

package input_test

import (
	"context"
	"os"
	"testing"
	"time"

	"github.com/matrix-org/dendrite/internal/caching"
	"github.com/matrix-org/dendrite/internal/sqlutil"
	"github.com/matrix-org/dendrite/roomserver/api"
	"github.com/matrix-org/dendrite/roomserver/internal/input"
	"github.com/matrix-org/dendrite/roomserver/storage"
	"github.com/matrix-org/dendrite/setup/base"
	"github.com/matrix-org/dendrite/setup/config"
	"github.com/matrix-org/dendrite/test/testrig"
	"github.com/matrix-org/gomatrixserverlib"
	"github.com/nats-io/nats.go"
)

var js nats.JetStreamContext
var jc *nats.Conn

func TestMain(m *testing.M) {
	var b *base.BaseDendrite
	b, js, jc = testrig.Base(nil)
	code := m.Run()
	b.ShutdownDendrite()
	b.WaitForComponentsToFinish()
	os.Exit(code)
}

func TestSingleTransactionOnInput(t *testing.T) {
	deadline, _ := t.Deadline()
	if max := time.Now().Add(time.Second * 3); deadline.After(max) {
		deadline = max
	}
	ctx, cancel := context.WithDeadline(context.Background(), deadline)
	defer cancel()

	event, err := gomatrixserverlib.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, gomatrixserverlib.RoomVersionV6,
	)
	if err != nil {
		t.Fatal(err)
	}
	in := api.InputRoomEvent{
		Kind:  api.KindOutlier, // don't panic if we generate an output event
		Event: event.Headered(gomatrixserverlib.RoomVersionV6),
	}
	cm := sqlutil.NewConnectionManager()
	db, err := storage.Open(
		context.Background(), cm,
		&config.DatabaseOptions{
			ConnectionString:   "",
			MaxOpenConnections: 1,
			MaxIdleConnections: 1,
		},
		caching.NewRistrettoCache(8*1024*1024, time.Hour, caching.DisableMetrics),
	)
	if err != nil {
		t.Logf("PostgreSQL not available (%s), skipping", err)
		t.SkipNow()
	}
	inputter := &input.Inputer{
		DB:         db,
		JetStream:  js,
		NATSClient: jc,
	}
	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)
	}
}