reload only chainlink node dbs when performing restart of CL nodes

Author: iljapavlovs

framework/components/postgres/postgres.go

@@ -3,14 +3,16 @@ package postgres
 import (
 	"context"
 	"fmt"
+	"os"
+	"strings"
+	"time"
+
 	"github.com/docker/docker/api/types/container"
 	"github.com/docker/go-connections/nat"
-	"github.com/smartcontractkit/chainlink-testing-framework/framework"
 	"github.com/testcontainers/testcontainers-go"
 	tcwait "github.com/testcontainers/testcontainers-go/wait"
-	"os"
-	"strings"
-	"time"
+
+	"github.com/smartcontractkit/chainlink-testing-framework/framework"
 )
 
 const (
@@ -20,10 +22,12 @@ const (
 	ExposedStaticPort = 13000
 	Database          = "chainlink"
 	DBVolumeName      = "postgresql_data"
+	DBContainerLabel  = "postgresql"
 )
 
 type Input struct {
 	Image      string  `toml:"image" validate:"required"`
+	Name       string  `toml:"name"`
 	Port       int     `toml:"port"`
 	VolumeName string  `toml:"volume_name"`
 	Databases  int     `toml:"databases"`
@@ -40,7 +44,7 @@ func NewPostgreSQL(in *Input) (*Output, error) {
 	ctx := context.Background()
 
 	bindPort := fmt.Sprintf("%s/tcp", Port)
-	containerName := framework.DefaultTCName("postgresql")
+	containerName := framework.DefaultTCName(fmt.Sprintf("%s-%s", DBContainerLabel, in.Name))
 
 	var sqlCommands []string
 	for i := 0; i <= in.Databases; i++ {

framework/components/simple_node_set/node_set.go

@@ -2,19 +2,22 @@ package simple_node_set
 
 import (
 	"fmt"
+	"slices"
+	"strings"
+	"sync"
+
+	"golang.org/x/sync/errgroup"
+
 	"github.com/smartcontractkit/chainlink-testing-framework/framework"
 	"github.com/smartcontractkit/chainlink-testing-framework/framework/components/blockchain"
 	"github.com/smartcontractkit/chainlink-testing-framework/framework/components/clnode"
 	"github.com/smartcontractkit/chainlink-testing-framework/framework/components/postgres"
-	"golang.org/x/sync/errgroup"
-	"slices"
-	"strings"
-	"sync"
 )
 
 const (
 	DefaultHTTPPortStaticRangeStart = 10000
 	DefaultP2PStaticRangeStart      = 12000
+	CLNodeDBContainerLabel          = "clnode"
 )
 
 // Input is a node set configuration input
@@ -75,6 +78,7 @@ func printURLs(out *Output) {
 
 func sharedDBSetup(in *Input, bcOut *blockchain.Output) (*Output, error) {
 	in.DbInput.Databases = in.Nodes
+	in.DbInput.Name = CLNodeDBContainerLabel
 	dbOut, err := postgres.NewPostgreSQL(in.DbInput)
 	if err != nil {
 		return nil, err

framework/components/simple_node_set/reload.go

@@ -1,15 +1,19 @@
 package simple_node_set
 
 import (
+	"fmt"
+	"time"
+
 	"github.com/smartcontractkit/chainlink-testing-framework/framework/chaos"
 	"github.com/smartcontractkit/chainlink-testing-framework/framework/components/blockchain"
-	"time"
+	"github.com/smartcontractkit/chainlink-testing-framework/framework/components/postgres"
 )
 
 // UpgradeNodeSet updates nodes configuration TOML files
 // this API is discouraged, however, you can use it if nodes require restart or configuration updates, temporarily!
 func UpgradeNodeSet(in *Input, bc *blockchain.Output, wait time.Duration) (*Output, error) {
-	_, err := chaos.ExecPumba("rm --volumes=false re2:node.*|postgresql.*", wait)
+	clNodeDBContainerNameRegex := fmt.Sprintf("%s-%s.*", postgres.DBContainerLabel, CLNodeDBContainerLabel)
+	_, err := chaos.ExecPumba(fmt.Sprintf("rm --volumes=false re2:node.*|%s", clNodeDBContainerNameRegex), wait)
 	if err != nil {
 		return nil, err
 	}