wehub-resource-sync
e071084ebe
govulncheck / govulncheck (push) Has been cancelled
Lint / golangci-lint (push) Has been cancelled
Run Tests / Unit Tests (push) Has been cancelled
Run Tests / Etcd Integration Tests (push) Has been cancelled
Harness (E2E) / Harnesses (mock LLM) (push) Has been cancelled
Harness (E2E) / Provider harnesses (live LLM conformance) (push) Has been cancelled
104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
package natsjs_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
nserver "github.com/nats-io/nats-server/v2/server"
|
|
)
|
|
|
|
func getFreeLocalhostAddress() string {
|
|
l, _ := net.Listen("tcp", "127.0.0.1:0")
|
|
defer l.Close()
|
|
return l.Addr().String()
|
|
}
|
|
|
|
func natsServer(ctx context.Context, t *testing.T, opts *nserver.Options) {
|
|
t.Helper()
|
|
|
|
// Report errors with Errorf (not Fatalf/require), which are safe to
|
|
// call from this non-test goroutine; Fatalf/FailNow are not.
|
|
server, err := nserver.NewServer(opts)
|
|
if err != nil {
|
|
t.Errorf("nats: new server: %v", err)
|
|
return
|
|
}
|
|
|
|
server.SetLoggerV2(
|
|
NewLogWrapper(),
|
|
true, true, false,
|
|
)
|
|
|
|
// first start NATS
|
|
go server.Start()
|
|
if !server.ReadyForConnections(time.Second * 10) {
|
|
t.Errorf("NATS server not ready")
|
|
return
|
|
}
|
|
|
|
// Manage the JetStream store dir ourselves rather than via t.TempDir.
|
|
// t.TempDir registers a RemoveAll that runs when the test ends, which
|
|
// races this goroutine's shutdown — the server can still be releasing
|
|
// JetStream files, leaving the dir non-empty ("directory not empty").
|
|
// Remove it here instead, only after the server has fully stopped.
|
|
storeDir, err := os.MkdirTemp("", "nats-js")
|
|
if err != nil {
|
|
t.Errorf("nats: temp dir: %v", err)
|
|
return
|
|
}
|
|
defer os.RemoveAll(storeDir)
|
|
|
|
// second start JetStream
|
|
if err := server.EnableJetStream(&nserver.JetStreamConfig{StoreDir: filepath.Join(storeDir, "nats-js")}); err != nil {
|
|
t.Errorf("nats: enable jetstream: %v", err)
|
|
return
|
|
}
|
|
|
|
<-ctx.Done()
|
|
|
|
server.Shutdown()
|
|
server.WaitForShutdown()
|
|
}
|
|
|
|
func NewLogWrapper() *LogWrapper {
|
|
return &LogWrapper{}
|
|
}
|
|
|
|
type LogWrapper struct {
|
|
}
|
|
|
|
// Noticef logs a notice statement.
|
|
func (l *LogWrapper) Noticef(format string, v ...interface{}) {
|
|
fmt.Printf(format+"\n", v...)
|
|
}
|
|
|
|
// Warnf logs a warning statement.
|
|
func (l *LogWrapper) Warnf(format string, v ...interface{}) {
|
|
fmt.Printf(format+"\n", v...)
|
|
}
|
|
|
|
// Fatalf logs a fatal statement.
|
|
func (l *LogWrapper) Fatalf(format string, v ...interface{}) {
|
|
fmt.Printf(format+"\n", v...)
|
|
}
|
|
|
|
// Errorf logs an error statement.
|
|
func (l *LogWrapper) Errorf(format string, v ...interface{}) {
|
|
fmt.Printf(format+"\n", v...)
|
|
}
|
|
|
|
// Debugf logs a debug statement.
|
|
func (l *LogWrapper) Debugf(format string, v ...interface{}) {
|
|
fmt.Printf(format+"\n", v...)
|
|
}
|
|
|
|
// Tracef logs a trace statement.
|
|
func (l *LogWrapper) Tracef(format string, v ...interface{}) {
|
|
fmt.Printf(format+"\n", v...)
|
|
}
|