//go:build !ios // +build !ios package testutil import ( "context" "os" "time" log "github.com/sirupsen/logrus" "github.com/testcontainers/testcontainers-go" "github.com/testcontainers/testcontainers-go/modules/mysql" "github.com/testcontainers/testcontainers-go/modules/postgres" "github.com/testcontainers/testcontainers-go/wait" ) // CreateMysqlTestContainer creates a new MySQL container for testing. func CreateMysqlTestContainer() (func(), error) { ctx := context.Background() container, err := mysql.RunContainer(ctx, testcontainers.WithImage("mlsmaycon/warmed-mysql:8"), mysql.WithDatabase("testing"), mysql.WithUsername("testing"), mysql.WithPassword("testing"), testcontainers.WithWaitStrategy( wait.ForLog("/usr/sbin/mysqld: ready for connections"). WithOccurrence(1).WithStartupTimeout(15*time.Second).WithPollInterval(100*time.Millisecond), ), ) if err != nil { return nil, err } cleanUp := func() { if err = container.Terminate(ctx); err != nil { log.WithContext(ctx).Warnf("failed to stop container: %s", err) } } talksConn, err := container.ConnectionString(ctx) if err != nil { return nil, err } return cleanUp, os.Setenv("NETBIRD_STORE_ENGINE_MYSQL_DSN", talksConn) } // CreatePostgresTestContainer creates a new PostgreSQL container for testing. func CreatePostgresTestContainer() (func(), error) { ctx := context.Background() container, err := postgres.RunContainer(ctx, testcontainers.WithImage("postgres:16-alpine"), postgres.WithDatabase("netbird"), postgres.WithUsername("root"), postgres.WithPassword("netbird"), testcontainers.WithWaitStrategy( wait.ForLog("database system is ready to accept connections"). WithOccurrence(2).WithStartupTimeout(15*time.Second), ), ) if err != nil { return nil, err } cleanUp := func() { timeout := 2 * time.Second if err = container.Stop(ctx, &timeout); err != nil { log.WithContext(ctx).Warnf("failed to stop container: %s", err) } } talksConn, err := container.ConnectionString(ctx) if err != nil { return nil, err } return cleanUp, os.Setenv("NETBIRD_STORE_ENGINE_POSTGRES_DSN", talksConn) }