add test

Signed-off-by: wayner0628 <[email protected]>
flyteorg · Nov 26, 2024 · aa9ae72 · aa9ae72
1 parent b3dca2a
commit aa9ae72
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/flyteplugins/go/tasks/pluginmachinery/flytek8s/copilot_test.go b/flyteplugins/go/tasks/pluginmachinery/flytek8s/copilot_test.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/base64"
 	"reflect"
+	"strings"
 	"testing"
 	"time"
 
@@ -107,6 +108,12 @@ func TestFlyteCoPilotContainer(t *testing.T) {
 		assert.Error(t, err)
 		cfg.Memory = old
 	})
+
+	t.Run("sidecar-container-name-change", func(t *testing.T) {
+		c, err := FlyteCoPilotContainer(flyteSidecarContainerName, cfg, []string{"hello"})
+		assert.NoError(t, err)
+		assert.Equal(t, "uploader", strings.Split(c.Name, "-")[1])
+	})
 }
 
 func TestDownloadCommandArgs(t *testing.T) {