x

itest/lnd_test.go

@@ -127,6 +127,12 @@ func TestLightningNetworkDaemon(t *testing.T) {
 			ht := harnessTest.Subtest(t1)
 			ht.SetTestName(testCase.Name)
 
+			postgresPort := 6432
+			if testCasesRunTranche != nil {
+				postgresPort += int(*testCasesRunTranche)
+			}
+			ht.SetPostgresPort(postgresPort)
+
 			ht.RunTestCase(testCase)
 		})
 

lntest/harness.go

@@ -469,6 +469,10 @@ func (h *HarnessTest) SetTestName(name string) {
 	h.manager.currentTestCase = cleanTestCaseName
 }
 
+func (h *HarnessTest) SetPostgresPort(port int) {
+	h.manager.postgresPort = port
+}
+
 // NewNode creates a new node and asserts its creation. The node is guaranteed
 // to have finished its initialization and all its subservers are started.
 func (h *HarnessTest) NewNode(name string,

lntest/harness_node_manager.go

@@ -46,6 +46,9 @@ type nodeManager struct {
 
 	// feeServiceURL is the url of the fee service.
 	feeServiceURL string
+
+	// postgresPort is the port of the postgres instance.
+	postgresPort int
 }
 
 // newNodeManager creates a new node manager instance.
@@ -86,6 +89,7 @@ func (nm *nodeManager) newNode(t *testing.T, name string, extraArgs []string,
 		LndBinary:         nm.lndBinary,
 		NetParams:         miner.HarnessNetParams,
 		SkipUnlock:        noAuth,
+		PostgresPort:      nm.postgresPort,
 	}
 
 	node, err := node.NewHarnessNode(t, cfg)

lntest/node/config.go

@@ -159,6 +159,10 @@ type BaseNodeConfig struct {
 	// postgresDBName is the name of the postgres database where lnd data
 	// is stored in.
 	postgresDBName string
+
+	// postgresPort is the port where the postgres database is listening
+	// on.
+	PostgresPort int
 }
 
 func (cfg BaseNodeConfig) P2PAddr() string {

lntest/node/harness_node.go

@@ -39,9 +39,6 @@ const (
 	// release of announcements by AuthenticatedGossiper to the network.
 	trickleDelay = 50
 
-	postgresDsn = "postgres://postgres:postgres@localhost:" +
-		"6432/%s?sslmode=disable"
-
 	// commitInterval specifies the maximum interval the graph database
 	// will wait between attempting to flush a batch of modifications to
 	// disk(db.batch-commit-interval).
@@ -123,11 +120,11 @@ func NewHarnessNode(t *testing.T, cfg *BaseNodeConfig) (*HarnessNode, error) {
 	var dbName string
 	if cfg.DBBackend == BackendPostgres {
 		var err error
-		dbName, err = createTempPgDB()
+		dbName, err = createTempPgDB(cfg.PostgresPort)
 		if err != nil {
 			return nil, err
 		}
-		cfg.PostgresDsn = postgresDatabaseDsn(dbName)
+		cfg.PostgresDsn = postgresDatabaseDsn(dbName, cfg.PostgresPort)
 	}
 
 	cfg.OriginalExtraArgs = cfg.ExtraArgs
@@ -837,8 +834,8 @@ func (hn *HarnessNode) BackupDB() error {
 		// Backup database.
 		backupDBName := hn.Cfg.postgresDBName + "_backup"
 		err := executePgQuery(
-			"CREATE DATABASE " + backupDBName + " WITH TEMPLATE " +
-				hn.Cfg.postgresDBName,
+			"CREATE DATABASE "+backupDBName+" WITH TEMPLATE "+
+				hn.Cfg.postgresDBName, hn.Cfg.PostgresPort,
 		)
 		if err != nil {
 			return err
@@ -868,14 +865,15 @@ func (hn *HarnessNode) RestoreDB() error {
 		// Restore database.
 		backupDBName := hn.Cfg.postgresDBName + "_backup"
 		err := executePgQuery(
-			"DROP DATABASE " + hn.Cfg.postgresDBName,
+			"DROP DATABASE "+hn.Cfg.postgresDBName,
+			hn.Cfg.PostgresPort,
 		)
 		if err != nil {
 			return err
 		}
 		err = executePgQuery(
-			"ALTER DATABASE " + backupDBName + " RENAME TO " +
-				hn.Cfg.postgresDBName,
+			"ALTER DATABASE "+backupDBName+" RENAME TO "+
+				hn.Cfg.postgresDBName, hn.Cfg.PostgresPort,
 		)
 		if err != nil {
 			return err
@@ -915,12 +913,15 @@ func (hn *HarnessNode) UpdateGlobalPolicy(policy *lnrpc.RoutingPolicy) {
 	hn.RPC.UpdateChannelPolicy(updateFeeReq)
 }
 
-func postgresDatabaseDsn(dbName string) string {
-	return fmt.Sprintf(postgresDsn, dbName)
+func postgresDatabaseDsn(dbName string, port int) string {
+	postgresDsn := "postgres://postgres:postgres@localhost:" +
+		"%d/%s?sslmode=disable"
+
+	return fmt.Sprintf(postgresDsn, port, dbName)
 }
 
 // createTempPgDB creates a temp postgres database.
-func createTempPgDB() (string, error) {
+func createTempPgDB(port int) (string, error) {
 	// Create random database name.
 	randBytes := make([]byte, 8)
 	_, err := rand.Read(randBytes)
@@ -930,7 +931,7 @@ func createTempPgDB() (string, error) {
 	dbName := "itest_" + hex.EncodeToString(randBytes)
 
 	// Create database.
-	err = executePgQuery("CREATE DATABASE " + dbName)
+	err = executePgQuery("CREATE DATABASE "+dbName, port)
 	if err != nil {
 		return "", err
 	}
@@ -939,10 +940,10 @@ func createTempPgDB() (string, error) {
 }
 
 // executePgQuery executes a SQL statement in a postgres db.
-func executePgQuery(query string) error {
+func executePgQuery(query string, port int) error {
 	pool, err := pgxpool.Connect(
 		context.Background(),
-		postgresDatabaseDsn("postgres"),
+		postgresDatabaseDsn("postgres", port),
 	)
 	if err != nil {
 		return fmt.Errorf("unable to connect to database: %w", err)

scripts/build_postgres.sh

@@ -13,7 +13,7 @@ fi
 echo "Starting new postgres container"
 
 for ((i=0; i<PROCESSES; i++)); do 
-    port=$((5432 + i))
+    port=$((6432 + i))
     container_name="lnd-postgres-$i"
 
     echo "Starting postgres container $container_name on port $port"