@@ -30,23 +30,24 @@ func TestUpgradeReplaceNode(cluster *shared.Cluster,
3030 }
3131
3232 awsClient := getAwsClient (cluster )
33+ resourceName := os .Getenv ("resource_name" )
3334
3435 // create and prepare the servers
3536 var newExternalServerIps , newPrivateServerIps []string
36- newExternalServerIps , newPrivateServerIps = createAndPrepNodes (awsClient , cluster , server )
37+ newExternalServerIps , newPrivateServerIps = createAndPrepNodes (awsClient , cluster , server , resourceName )
3738
3839 serverLeaderIP := cluster .ServerIPs [0 ]
3940 token , err := shared .FetchToken (cluster .Config .Product , serverLeaderIP )
4041 Expect (err ).NotTo (HaveOccurred (), err )
4142
4243 serverErr := nodeReplaceServers (cluster , awsClient , serverLeaderIP , token ,
43- version , channel , newExternalServerIps , newPrivateServerIps )
44+ version , channel , resourceName , newExternalServerIps , newPrivateServerIps )
4445 Expect (serverErr ).NotTo (HaveOccurred (), serverErr )
4546 shared .LogLevel ("info" , "Server control plane nodes replaced with ips: %s\n " , newExternalServerIps )
4647
4748 // replace agents only if exists.
4849 if len (cluster .AgentIPs ) > 0 {
49- nodeReplaceAgents (cluster , awsClient , version , channel , serverLeaderIP , token )
50+ nodeReplaceAgents (cluster , awsClient , version , channel , serverLeaderIP , token , resourceName )
5051 }
5152 // delete the last remaining server = leader.
5253 delErr := deleteRemainServer (serverLeaderIP , awsClient )
@@ -184,7 +185,7 @@ func k3sServerSCP(cluster *shared.Cluster, ip string) error {
184185func nodeReplaceServers (
185186 cluster * shared.Cluster ,
186187 a * aws.Client ,
187- serverLeaderIp , token , version , channel string ,
188+ serverLeaderIp , token , version , channel , resourceName string ,
188189 newExternalServerIps , newPrivateServerIps []string ,
189190) error {
190191 if token == "" {
@@ -213,7 +214,7 @@ func nodeReplaceServers(
213214 }
214215
215216 shared .LogLevel ("info" , "Proceeding to update kubeconfig file to point to new first server join %s\n " , newFirstServerIP )
216- kubeConfigUpdated , kbCfgErr := shared .UpdateKubeConfig (newFirstServerIP , cluster . ResourceName , cluster .Config .Product )
217+ kubeConfigUpdated , kbCfgErr := shared .UpdateKubeConfig (newFirstServerIP , resourceName , cluster .Config .Product )
217218 if kbCfgErr != nil {
218219 return shared .ReturnLogError ("error updating kubeconfig: %w with ip: %s" , kbCfgErr , newFirstServerIP )
219220 }
@@ -521,9 +522,10 @@ func nodeReplaceAgents(
521522 version ,
522523 channel ,
523524 serverLeaderIp ,
524- token string ,
525+ token ,
526+ resourceName string ,
525527) {
526- newExternalAgentIps , newPrivateAgentIps := createAndPrepNodes (awsClient , cluster , agent )
528+ newExternalAgentIps , newPrivateAgentIps := createAndPrepNodes (awsClient , cluster , agent , resourceName )
527529
528530 agentErr := replaceAgents (cluster , awsClient , serverLeaderIp , token , version , channel ,
529531 newExternalAgentIps , newPrivateAgentIps )
@@ -728,10 +730,10 @@ func getNodeNames(cluster *shared.Cluster, resourceName, nodeType string) []stri
728730 return nodeNames
729731}
730732
731- func createAndPrepNodes (awsClient * aws.Client , cluster * shared.Cluster , nodeType string ) (
733+ func createAndPrepNodes (awsClient * aws.Client , cluster * shared.Cluster , nodeType , resourceName string ) (
732734 newExternalIps []string , newPrivateIps []string ) {
733735 // create aws ec2 instances
734- names := getNodeNames (cluster , cluster . ResourceName , nodeType )
736+ names := getNodeNames (cluster , resourceName , nodeType )
735737 newExternalIps , newPrivateIps , instanceIds , createErr := awsClient .CreateInstances (names ... )
736738 Expect (createErr ).NotTo (HaveOccurred (), createErr )
737739 shared .LogLevel ("debug" , "Created %s nodes with public ips: %s and ids: %s\n " ,
0 commit comments