fix: latest on nimbus

NethermindEth · Dec 1, 2024 · 3ebaf4a · 3ebaf4a
1 parent aa766fb
commit 3ebaf4a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/cli/actions/generation_test.go b/cli/actions/generation_test.go
@@ -418,6 +418,8 @@ func TestGenerateDockerCompose(t *testing.T) {
 				if tc.genData.LatestVersion {
 					if tc.genData.ConsensusClient.Modified {
 						assert.True(t, strings.HasSuffix(named.String(), tc.genData.ConsensusClient.Image))
+					} else if tc.genData.ConsensusClient.Name == "nimbus" {
+						assert.True(t, strings.HasSuffix(named.String(), ":multiarch-latest"))
 					} else {
 						assert.True(t, strings.HasSuffix(named.String(), ":latest"))
 					}
@@ -483,6 +485,8 @@ func TestGenerateDockerCompose(t *testing.T) {
 				if tc.genData.LatestVersion {
 					if tc.genData.ValidatorClient.Modified {
 						assert.True(t, strings.HasSuffix(named.String(), tc.genData.ValidatorClient.Image))
+					} else if tc.genData.ValidatorClient.Name == "nimbus" {
+						assert.True(t, strings.HasSuffix(named.String(), ":multiarch-latest"))
 					} else {
 						assert.True(t, strings.HasSuffix(named.String(), ":latest"))
 					}

diff --git a/internal/pkg/generate/generate_scripts.go b/internal/pkg/generate/generate_scripts.go
@@ -698,9 +698,16 @@ func joinIfNotEmpty(strs ...string) string {
 func imageOrEmpty(cls *clients.Client, latest bool) string {
 	if cls != nil {
 		if latest && !cls.Modified {
-			splits := strings.Split(cls.Image, ":")
-			splits[len(splits)-1] = "latest"
-			return strings.Join(splits, ":")
+			if cls.Name == "nimbus" {
+				splits := strings.Split(cls.Image, ":")
+				splits[len(splits)-1] = "multiarch-latest"
+				return strings.Join(splits, ":")
+
+			} else {
+				splits := strings.Split(cls.Image, ":")
+				splits[len(splits)-1] = "latest"
+				return strings.Join(splits, ":")
+			}
 		}
 		return cls.Image
 	}