diff --git a/internal/ui/commands/deploy.go b/internal/ui/commands/deploy.go index 82085cc..cf30d5d 100644 --- a/internal/ui/commands/deploy.go +++ b/internal/ui/commands/deploy.go @@ -707,8 +707,10 @@ func (m *DeployView) View() string { case StateCreatingApp: output.WriteString(formatStateLine(m.spinner.View(), "Creating app...", ui.ActiveStyle.Render)) output.WriteString("\n") - output.WriteString(formatStateLine("-", "Upload to Cerebrium", ui.PendingStyle.Render)) - output.WriteString("\n") + if !m.isPartnerService() { + output.WriteString(formatStateLine("-", "Upload to Cerebrium", ui.PendingStyle.Render)) + output.WriteString("\n") + } output.WriteString(formatStateLine("-", "Build app", ui.PendingStyle.Render)) output.WriteString("\n") diff --git a/internal/ui/commands/deploy_test.go b/internal/ui/commands/deploy_test.go index 1ef1b7c..ef702ff 100644 --- a/internal/ui/commands/deploy_test.go +++ b/internal/ui/commands/deploy_test.go @@ -1070,6 +1070,26 @@ func TestDeployView_View(t *testing.T) { view := model.View() assert.Contains(t, view, "Creating app") + assert.Contains(t, view, "Upload to Cerebrium") + }) + + t.Run("view during app creation omits upload step for partner services", func(t *testing.T) { + model := NewDeployView(t.Context(), DeployConfig{ + DisplayConfig: ui.DisplayConfig{ + IsInteractive: true, + DisableAnimation: false, + }, + Config: partnerConfig("partner-app"), + ProjectID: "test-project", + Client: apimock.NewMockClient(t), + }) + + model.state = StateCreatingApp + + view := model.View() + assert.Contains(t, view, "Creating app") + assert.Contains(t, view, "Build app") + assert.NotContains(t, view, "Upload to Cerebrium") }) t.Run("view during upload", func(t *testing.T) {