mirror of
https://github.com/matrix-org/dendrite.git
synced 2026-01-22 05:23:09 -06:00
Fix sqlite unit test parallelization issues
This commit is contained in:
parent
703d4371f5
commit
91126e6efa
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -56,6 +56,7 @@ dendrite.yaml
|
||||||
|
|
||||||
# Database files
|
# Database files
|
||||||
*.db
|
*.db
|
||||||
|
*.db-journal
|
||||||
|
|
||||||
# Log files
|
# Log files
|
||||||
*.log*
|
*.log*
|
||||||
|
|
|
||||||
|
|
@ -67,9 +67,11 @@ func Setup(
|
||||||
servers federationAPI.ServersInRoomProvider,
|
servers federationAPI.ServersInRoomProvider,
|
||||||
producer *producers.SyncAPIProducer,
|
producer *producers.SyncAPIProducer,
|
||||||
) {
|
) {
|
||||||
|
if cfg.Matrix.Metrics.Enabled {
|
||||||
prometheus.MustRegister(
|
prometheus.MustRegister(
|
||||||
internal.PDUCountTotal, internal.EDUCountTotal,
|
internal.PDUCountTotal, internal.EDUCountTotal,
|
||||||
)
|
)
|
||||||
|
}
|
||||||
|
|
||||||
v2keysmux := keyMux.PathPrefix("/v2").Subrouter()
|
v2keysmux := keyMux.PathPrefix("/v2").Subrouter()
|
||||||
v1fedmux := fedMux.PathPrefix("/v1").Subrouter()
|
v1fedmux := fedMux.PathPrefix("/v1").Subrouter()
|
||||||
|
|
|
||||||
|
|
@ -546,7 +546,8 @@ type sendContent struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestHandleSend(t *testing.T) {
|
func TestHandleSend(t *testing.T) {
|
||||||
base, close := testrig.CreateBaseDendrite(t, test.DBTypeSQLite)
|
test.WithAllDatabases(t, func(t *testing.T, dbType test.DBType) {
|
||||||
|
base, close := testrig.CreateBaseDendrite(t, dbType)
|
||||||
defer close()
|
defer close()
|
||||||
|
|
||||||
fedMux := mux.NewRouter().SkipClean(true).PathPrefix(httputil.PublicFederationPathPrefix).Subrouter().UseEncodedPath()
|
fedMux := mux.NewRouter().SkipClean(true).PathPrefix(httputil.PublicFederationPathPrefix).Subrouter().UseEncodedPath()
|
||||||
|
|
@ -556,6 +557,9 @@ func TestHandleSend(t *testing.T) {
|
||||||
SigningIdentity: gomatrixserverlib.SigningIdentity{
|
SigningIdentity: gomatrixserverlib.SigningIdentity{
|
||||||
ServerName: "remote",
|
ServerName: "remote",
|
||||||
},
|
},
|
||||||
|
Metrics: config.Metrics{
|
||||||
|
Enabled: false,
|
||||||
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
fedapi := fedAPI.NewInternalAPI(base, nil, nil, nil, nil, true)
|
fedapi := fedAPI.NewInternalAPI(base, nil, nil, nil, nil, true)
|
||||||
|
|
@ -590,4 +594,5 @@ func TestHandleSend(t *testing.T) {
|
||||||
|
|
||||||
res := w.Result()
|
res := w.Result()
|
||||||
assert.Equal(t, 200, res.StatusCode)
|
assert.Equal(t, 200, res.StatusCode)
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -104,7 +104,8 @@ func createSendRelayTxnHTTPRequest(serverName gomatrixserverlib.ServerName, txnI
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCreateRelayPublicRoutes(t *testing.T) {
|
func TestCreateRelayPublicRoutes(t *testing.T) {
|
||||||
base, close := testrig.CreateBaseDendrite(t, test.DBTypeSQLite)
|
test.WithAllDatabases(t, func(t *testing.T, dbType test.DBType) {
|
||||||
|
base, close := testrig.CreateBaseDendrite(t, dbType)
|
||||||
defer close()
|
defer close()
|
||||||
|
|
||||||
relayAPI := relayapi.NewRelayInternalAPI(base, nil, nil, nil, nil)
|
relayAPI := relayapi.NewRelayInternalAPI(base, nil, nil, nil, nil)
|
||||||
|
|
@ -149,4 +150,5 @@ func TestCreateRelayPublicRoutes(t *testing.T) {
|
||||||
t.Fatalf("%s: got HTTP %d want %d", tc.name, w.Code, tc.wantCode)
|
t.Fatalf("%s: got HTTP %d want %d", tc.name, w.Code, tc.wantCode)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
|
||||||
17
test/db.go
17
test/db.go
|
|
@ -19,10 +19,12 @@ import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"math/rand"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"os/user"
|
"os/user"
|
||||||
"testing"
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/lib/pq"
|
"github.com/lib/pq"
|
||||||
)
|
)
|
||||||
|
|
@ -100,15 +102,16 @@ func currentUser() string {
|
||||||
// Returns the connection string to use and a close function which must be called when the test finishes.
|
// Returns the connection string to use and a close function which must be called when the test finishes.
|
||||||
// Calling this function twice will return the same database, which will have data from previous tests
|
// Calling this function twice will return the same database, which will have data from previous tests
|
||||||
// unless close() is called.
|
// unless close() is called.
|
||||||
// TODO: namespace for concurrent package tests
|
|
||||||
func PrepareDBConnectionString(t *testing.T, dbType DBType) (connStr string, close func()) {
|
func PrepareDBConnectionString(t *testing.T, dbType DBType) (connStr string, close func()) {
|
||||||
if dbType == DBTypeSQLite {
|
if dbType == DBTypeSQLite {
|
||||||
// this will be made in the current working directory which namespaces concurrent package runs correctly
|
rand.Seed(time.Now().UnixNano())
|
||||||
dbname := "dendrite_test.db"
|
randBytes := make([]byte, 32)
|
||||||
return fmt.Sprintf("file:%s", dbname), func() {
|
rand.Read(randBytes)
|
||||||
err := os.Remove(dbname)
|
dbName := fmt.Sprintf("dendrite_test_%s.db", hex.EncodeToString(randBytes[:16]))
|
||||||
|
return fmt.Sprintf("file:%s", dbName), func() {
|
||||||
|
err := os.Remove(dbName)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("failed to cleanup sqlite db '%s': %s", dbname, err)
|
t.Fatalf("failed to cleanup sqlite db '%s': %s", dbName, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -176,7 +179,7 @@ func WithAllDatabases(t *testing.T, testFn func(t *testing.T, db DBType)) {
|
||||||
for dbName, dbType := range dbs {
|
for dbName, dbType := range dbs {
|
||||||
dbt := dbType
|
dbt := dbType
|
||||||
t.Run(dbName, func(tt *testing.T) {
|
t.Run(dbName, func(tt *testing.T) {
|
||||||
//tt.Parallel()
|
tt.Parallel()
|
||||||
testFn(tt, dbt)
|
testFn(tt, dbt)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -69,6 +69,22 @@ func CreateBaseDendrite(t *testing.T, dbType test.DBType) (*base.BaseDendrite, f
|
||||||
Monolithic: false, // because we need a database per component
|
Monolithic: false, // because we need a database per component
|
||||||
})
|
})
|
||||||
cfg.Global.ServerName = "test"
|
cfg.Global.ServerName = "test"
|
||||||
|
connStr, _ := test.PrepareDBConnectionString(t, dbType)
|
||||||
|
cfg.FederationAPI.Database.ConnectionString = config.DataSource(connStr)
|
||||||
|
connStr, _ = test.PrepareDBConnectionString(t, dbType)
|
||||||
|
cfg.KeyServer.Database.ConnectionString = config.DataSource(connStr)
|
||||||
|
connStr, _ = test.PrepareDBConnectionString(t, dbType)
|
||||||
|
cfg.MSCs.Database.ConnectionString = config.DataSource(connStr)
|
||||||
|
connStr, _ = test.PrepareDBConnectionString(t, dbType)
|
||||||
|
cfg.MediaAPI.Database.ConnectionString = config.DataSource(connStr)
|
||||||
|
connStr, _ = test.PrepareDBConnectionString(t, dbType)
|
||||||
|
cfg.RoomServer.Database.ConnectionString = config.DataSource(connStr)
|
||||||
|
connStr, _ = test.PrepareDBConnectionString(t, dbType)
|
||||||
|
cfg.SyncAPI.Database.ConnectionString = config.DataSource(connStr)
|
||||||
|
connStr, _ = test.PrepareDBConnectionString(t, dbType)
|
||||||
|
cfg.UserAPI.AccountDatabase.ConnectionString = config.DataSource(connStr)
|
||||||
|
connStr, _ = test.PrepareDBConnectionString(t, dbType)
|
||||||
|
cfg.RelayAPI.Database.ConnectionString = config.DataSource(connStr)
|
||||||
// use a distinct prefix else concurrent postgres/sqlite runs will clash since NATS will use
|
// use a distinct prefix else concurrent postgres/sqlite runs will clash since NATS will use
|
||||||
// the file system event with InMemory=true :(
|
// the file system event with InMemory=true :(
|
||||||
cfg.Global.JetStream.TopicPrefix = fmt.Sprintf("Test_%d_", dbType)
|
cfg.Global.JetStream.TopicPrefix = fmt.Sprintf("Test_%d_", dbType)
|
||||||
|
|
@ -82,6 +98,7 @@ func CreateBaseDendrite(t *testing.T, dbType test.DBType) (*base.BaseDendrite, f
|
||||||
cfg.RoomServer.Database.ConnectionString,
|
cfg.RoomServer.Database.ConnectionString,
|
||||||
cfg.SyncAPI.Database.ConnectionString,
|
cfg.SyncAPI.Database.ConnectionString,
|
||||||
cfg.UserAPI.AccountDatabase.ConnectionString,
|
cfg.UserAPI.AccountDatabase.ConnectionString,
|
||||||
|
cfg.RelayAPI.Database.ConnectionString,
|
||||||
}
|
}
|
||||||
for _, fileURI := range dbFiles {
|
for _, fileURI := range dbFiles {
|
||||||
path := strings.TrimPrefix(string(fileURI), "file:")
|
path := strings.TrimPrefix(string(fileURI), "file:")
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue