feat: add nats server on integration tests

formancehq · Oct 16, 2024 · eb0a819 · eb0a819
1 parent f7ce827
commit eb0a819
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/pkg/testserver/server.go b/pkg/testserver/server.go
@@ -3,6 +3,7 @@ package testserver
 import (
 	"context"
 	"fmt"
+	"github.com/formancehq/go-libs/publish"
 	"io"
 	"net/http"
 	"strings"
@@ -30,6 +31,7 @@ type T interface {
 
 type Configuration struct {
 	PostgresConfiguration bunconnect.ConnectionOptions
+	NatsURL string
 	Output                io.Writer
 	Debug                 bool
 }
@@ -75,6 +77,14 @@ func (s *Server) Start() {
 			fmt.Sprint(s.configuration.PostgresConfiguration.ConnMaxIdleTime),
 		)
 	}
+	if s.configuration.NatsURL != "" {
+		args = append(
+			args,
+			"--" + publish.PublisherNatsEnabledFlag,
+			"--" + publish.PublisherNatsURLFlag, s.configuration.NatsURL,
+		)
+	}
+
 	if s.configuration.Debug {
 		args = append(args, "--"+service.DebugFlag)
 	}

diff --git a/test/e2e/environment_test.go b/test/e2e/environment_test.go
@@ -4,6 +4,7 @@ package test_suite
 
 import (
 	"encoding/json"
+	"github.com/formancehq/go-libs/testing/platform/natstesting"
 	"os"
 
 	"github.com/formancehq/go-libs/logging"
@@ -17,12 +18,14 @@ import (
 var (
 	dockerPool = NewDeferred[*docker.Pool]()
 	pgServer   = NewDeferred[*PostgresServer]()
+	natsServer = NewDeferred[*natstesting.NatsServer]()
 	debug      = os.Getenv("DEBUG") == "true"
 	logger     = logging.NewDefaultLogger(GinkgoWriter, debug, false)
 )
 
 type ParallelExecutionContext struct {
 	PostgresServer *PostgresServer
+	NatsServer *natstesting.NatsServer
 }
 
 var _ = SynchronizedBeforeSuite(func() []byte {
@@ -38,13 +41,18 @@ var _ = SynchronizedBeforeSuite(func() []byte {
 			WithPGCrypto(),
 		)
 	})
+	natsServer.LoadAsync(func() *natstesting.NatsServer {
+		By("Initializing nats server")
+		return natstesting.CreateServer(GinkgoT(), debug, logger)
+	})
 
 	By("Waiting services alive")
-	Wait(pgServer)
+	Wait(pgServer, natsServer)
 	By("All services ready.")
 
 	data, err := json.Marshal(ParallelExecutionContext{
 		PostgresServer: pgServer.GetValue(),
+		NatsServer:     natsServer.GetValue(),
 	})
 	Expect(err).To(BeNil())
 

diff --git a/test/e2e/integration_test.go b/test/e2e/integration_test.go
@@ -33,6 +33,7 @@ var _ = Context("Ledger integration tests", func() {
 			PostgresConfiguration: db.GetValue().ConnectionOptions(),
 			Output:                GinkgoWriter,
 			Debug:                 debug,
+			NatsURL:               natsServer.GetValue().URL,
 		}
 	})
 	When("starting the service", func() {