From 7499147550110d24fa3a376bd811d9dd38971629 Mon Sep 17 00:00:00 2001 From: kegsay Date: Fri, 8 Apr 2022 10:12:30 +0100 Subject: Add test infrastructure code for dendrite unit/integ tests (#2331) * Add test infrastructure code for dendrite unit/integ tests Start re-enabling some syncapi storage tests in the process. * Linting * Add postgres service to unit tests * dendrite not syncv3 * Skip test which doesn't work * Linting * Add `jetstream.PrepareForTests` Co-authored-by: Neil Alexander --- setup/jetstream/nats.go | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'setup/jetstream') diff --git a/setup/jetstream/nats.go b/setup/jetstream/nats.go index 4e4fe7a2..1c8a89e8 100644 --- a/setup/jetstream/nats.go +++ b/setup/jetstream/nats.go @@ -13,12 +13,22 @@ import ( "github.com/sirupsen/logrus" natsserver "github.com/nats-io/nats-server/v2/server" + "github.com/nats-io/nats.go" natsclient "github.com/nats-io/nats.go" ) var natsServer *natsserver.Server var natsServerMutex sync.Mutex +func PrepareForTests() (*process.ProcessContext, nats.JetStreamContext, *nats.Conn) { + cfg := &config.Dendrite{} + cfg.Defaults(true) + cfg.Global.JetStream.InMemory = true + pc := process.NewProcessContext() + js, jc := Prepare(pc, &cfg.Global.JetStream) + return pc, js, jc +} + func Prepare(process *process.ProcessContext, cfg *config.JetStream) (natsclient.JetStreamContext, *natsclient.Conn) { // check if we need an in-process NATS Server if len(cfg.Addresses) != 0 { -- cgit v1.2.3