Skip to content

Commit 72b451d

Browse files
marceljkmarceljk
committed
Add nil pointer checks and tests for the outputResult functions
1 parent 6213bb5 commit 72b451d

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

73 files changed

+1400
-108
lines changed

‎internal/cmd/beta/server/backup/create/create.go‎

Lines changed: 17 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@ import (
1212
"github.com/stackitcloud/stackit-cli/internal/pkg/flags"
1313
"github.com/stackitcloud/stackit-cli/internal/pkg/globalflags"
1414
"github.com/stackitcloud/stackit-cli/internal/pkg/print"
15+
iaasClient "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client"
16+
iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils"
1517
"github.com/stackitcloud/stackit-cli/internal/pkg/services/serverbackup/client"
1618
"github.com/stackitcloud/stackit-cli/internal/pkg/utils"
1719

@@ -65,6 +67,17 @@ func NewCmd(p *print.Printer) *cobra.Command {
6567
return err
6668
}
6769

70+
serverLabel := model.ServerId
71+
// Get server name
72+
if iaasApiClient, err := iaasClient.ConfigureClient(p); err == nil {
73+
serverName, err := iaasUtils.GetServerName(ctx, iaasApiClient, model.ProjectId, model.ServerId)
74+
if err != nil {
75+
p.Debug(print.ErrorLevel, "get server name: %v", err)
76+
} else {
77+
serverLabel = serverName
78+
}
79+
}
80+
6881
if !model.AssumeYes {
6982
prompt := fmt.Sprintf("Are you sure you want to create a Backup for server %s?", model.ServerId)
7083
err = p.PromptForConfirmation(prompt)
@@ -83,7 +96,7 @@ func NewCmd(p *print.Printer) *cobra.Command {
8396
return fmt.Errorf("create Server Backup: %w", err)
8497
}
8598

86-
return outputResult(p, model, resp)
99+
return outputResult(p, model.OutputFormat, serverLabel, *resp)
87100
},
88101
}
89102
configureFlags(cmd)
@@ -140,8 +153,8 @@ func buildRequest(ctx context.Context, model *inputModel, apiClient *serverbacku
140153
return req, nil
141154
}
142155

143-
func outputResult(p *print.Printer, model *inputModel, resp *serverbackup.BackupJob) error {
144-
switch model.OutputFormat {
156+
func outputResult(p *print.Printer, outputFormat, serverLabel string, resp serverbackup.BackupJob) error {
157+
switch outputFormat {
145158
case print.JSONOutputFormat:
146159
details, err := json.MarshalIndent(resp, "", " ")
147160
if err != nil {
@@ -159,7 +172,7 @@ func outputResult(p *print.Printer, model *inputModel, resp *serverbackup.Backup
159172

160173
return nil
161174
default:
162-
p.Outputf("Triggered creation of server backup for server %s. Backup ID: %s\n", model.ServerId, utils.PtrString(resp.Id))
175+
p.Outputf("Triggered creation of server backup for server %s. Backup ID: %s\n", serverLabel, utils.PtrString(resp.Id))
163176
return nil
164177
}
165178
}

‎internal/cmd/beta/server/backup/create/create_test.go‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,3 +208,31 @@ func TestBuildRequest(t *testing.T) {
208208
})
209209
}
210210
}
211+
212+
func TestOutputResult(t *testing.T) {
213+
type args struct {
214+
outputFormat string
215+
serverLabel string
216+
resp serverbackup.BackupJob
217+
}
218+
tests := []struct {
219+
name string
220+
args args
221+
wantErr bool
222+
}{
223+
{
224+
name: "empty",
225+
args: args{},
226+
wantErr: false,
227+
},
228+
}
229+
p := print.NewPrinter()
230+
p.Cmd = NewCmd(p)
231+
for _, tt := range tests {
232+
t.Run(tt.name, func(t *testing.T) {
233+
if err := outputResult(p, tt.args.outputFormat, tt.args.serverLabel, tt.args.resp); (err != nil) != tt.wantErr {
234+
t.Errorf("outputResult() error = %v, wantErr %v", err, tt.wantErr)
235+
}
236+
})
237+
}
238+
}

‎internal/cmd/beta/server/backup/describe/describe.go‎

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"context"
55
"encoding/json"
66
"fmt"
7+
"strconv"
78

89
"github.com/goccy/go-yaml"
910
"github.com/stackitcloud/stackit-cli/internal/pkg/args"
@@ -64,7 +65,7 @@ func NewCmd(p *print.Printer) *cobra.Command {
6465
return fmt.Errorf("read server backup: %w", err)
6566
}
6667

67-
return outputResult(p, model.OutputFormat, resp)
68+
return outputResult(p, model.OutputFormat, *resp)
6869
},
6970
}
7071
configureFlags(cmd)
@@ -109,7 +110,7 @@ func buildRequest(ctx context.Context, model *inputModel, apiClient *serverbacku
109110
return req
110111
}
111112

112-
func outputResult(p *print.Printer, outputFormat string, backup *serverbackup.Backup) error {
113+
func outputResult(p *print.Printer, outputFormat string, backup serverbackup.Backup) error {
113114
switch outputFormat {
114115
case print.JSONOutputFormat:
115116
details, err := json.MarshalIndent(backup, "", " ")
@@ -145,7 +146,11 @@ func outputResult(p *print.Printer, outputFormat string, backup *serverbackup.Ba
145146
lastRestored := utils.PtrStringDefault(backup.LastRestoredAt, "")
146147
table.AddRow("LAST RESTORED AT", lastRestored)
147148
table.AddSeparator()
148-
table.AddRow("VOLUME BACKUPS", len(*backup.VolumeBackups))
149+
volBackups := ""
150+
if backups := backup.VolumeBackups; backups != nil {
151+
volBackups = strconv.Itoa(len(*backups))
152+
}
153+
table.AddRow("VOLUME BACKUPS", volBackups)
149154
table.AddSeparator()
150155

151156
err := table.Display(p)

‎internal/cmd/beta/server/backup/describe/describe_test.go‎

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,3 +209,38 @@ func TestBuildRequest(t *testing.T) {
209209
})
210210
}
211211
}
212+
213+
func TestOutputResult(t *testing.T) {
214+
type args struct {
215+
outputFormat string
216+
backup serverbackup.Backup
217+
}
218+
tests := []struct {
219+
name string
220+
args args
221+
wantErr bool
222+
}{
223+
{
224+
name: "empty",
225+
args: args{},
226+
wantErr: false,
227+
},
228+
{
229+
name: "output format json",
230+
args: args{
231+
outputFormat: print.JSONOutputFormat,
232+
backup: serverbackup.Backup{},
233+
},
234+
wantErr: false,
235+
},
236+
}
237+
p := print.NewPrinter()
238+
p.Cmd = NewCmd(p)
239+
for _, tt := range tests {
240+
t.Run(tt.name, func(t *testing.T) {
241+
if err := outputResult(p, tt.args.outputFormat, tt.args.backup); (err != nil) != tt.wantErr {
242+
t.Errorf("outputResult() error = %v, wantErr %v", err, tt.wantErr)
243+
}
244+
})
245+
}
246+
}

‎internal/cmd/beta/server/backup/disable/disable.go‎

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@ import (
1010
"github.com/stackitcloud/stackit-cli/internal/pkg/flags"
1111
"github.com/stackitcloud/stackit-cli/internal/pkg/globalflags"
1212
"github.com/stackitcloud/stackit-cli/internal/pkg/print"
13+
iaasClient "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client"
14+
iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils"
1315
"github.com/stackitcloud/stackit-cli/internal/pkg/services/serverbackup/client"
1416
serverbackupUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/serverbackup/utils"
1517

@@ -50,17 +52,28 @@ func NewCmd(p *print.Printer) *cobra.Command {
5052
return err
5153
}
5254

55+
serverLabel := model.ServerId
56+
// Get server name
57+
if iaasApiClient, err := iaasClient.ConfigureClient(p); err == nil {
58+
serverName, err := iaasUtils.GetServerName(ctx, iaasApiClient, model.ProjectId, model.ServerId)
59+
if err != nil {
60+
p.Debug(print.ErrorLevel, "get server name: %v", err)
61+
} else {
62+
serverLabel = serverName
63+
}
64+
}
65+
5366
canDisable, err := serverbackupUtils.CanDisableBackupService(ctx, apiClient, model.ProjectId, model.ServerId)
5467
if err != nil {
5568
return err
5669
}
5770
if !canDisable {
58-
p.Info("Cannot disable backup service for server %s - existing backups or existing backup schedules found\n", model.ServerId)
71+
p.Info("Cannot disable backup service for server %s - existing backups or existing backup schedules found\n", serverLabel)
5972
return nil
6073
}
6174

6275
if !model.AssumeYes {
63-
prompt := fmt.Sprintf("Are you sure you want to disable the backup service for server %s?", model.ServerId)
76+
prompt := fmt.Sprintf("Are you sure you want to disable the backup service for server %s?", serverLabel)
6477
err = p.PromptForConfirmation(prompt)
6578
if err != nil {
6679
return err
@@ -74,7 +87,7 @@ func NewCmd(p *print.Printer) *cobra.Command {
7487
return fmt.Errorf("disable server backup service: %w", err)
7588
}
7689

77-
p.Info("Disabled Server Backup service for server %s\n", model.ServerId)
90+
p.Info("Disabled Server Backup service for server %s\n", serverLabel)
7891
return nil
7992
},
8093
}

‎internal/cmd/beta/server/backup/enable/enable.go‎

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@ import (
1111
"github.com/stackitcloud/stackit-cli/internal/pkg/flags"
1212
"github.com/stackitcloud/stackit-cli/internal/pkg/globalflags"
1313
"github.com/stackitcloud/stackit-cli/internal/pkg/print"
14+
iaasClient "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client"
15+
iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils"
1416
"github.com/stackitcloud/stackit-cli/internal/pkg/services/serverbackup/client"
1517

1618
"github.com/spf13/cobra"
@@ -50,8 +52,19 @@ func NewCmd(p *print.Printer) *cobra.Command {
5052
return err
5153
}
5254

55+
serverLabel := model.ServerId
56+
// Get server name
57+
if iaasApiClient, err := iaasClient.ConfigureClient(p); err == nil {
58+
serverName, err := iaasUtils.GetServerName(ctx, iaasApiClient, model.ProjectId, model.ServerId)
59+
if err != nil {
60+
p.Debug(print.ErrorLevel, "get server name: %v", err)
61+
} else {
62+
serverLabel = serverName
63+
}
64+
}
65+
5366
if !model.AssumeYes {
54-
prompt := fmt.Sprintf("Are you sure you want to enable the Server Backup service for server %s?", model.ServerId)
67+
prompt := fmt.Sprintf("Are you sure you want to enable the Server Backup service for server %s?", serverLabel)
5568
err = p.PromptForConfirmation(prompt)
5669
if err != nil {
5770
return err
@@ -67,7 +80,7 @@ func NewCmd(p *print.Printer) *cobra.Command {
6780
}
6881
}
6982

70-
p.Info("Enabled backup service for server %s\n", model.ServerId)
83+
p.Info("Enabled backup service for server %s\n", serverLabel)
7184
return nil
7285
},
7386
}

‎internal/cmd/beta/server/backup/list/list.go‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,10 @@ func outputResult(p *print.Printer, outputFormat string, backups []serverbackup.
150150
s := backups[i]
151151

152152
lastRestored := utils.PtrStringDefault(s.LastRestoredAt, "")
153+
var volBackups int
154+
if s.VolumeBackups != nil {
155+
volBackups = len(*s.VolumeBackups)
156+
}
153157
table.AddRow(
154158
utils.PtrString(s.Id),
155159
utils.PtrString(s.Name),
@@ -158,7 +162,7 @@ func outputResult(p *print.Printer, outputFormat string, backups []serverbackup.
158162
utils.PtrString(s.CreatedAt),
159163
utils.PtrString(s.ExpireAt),
160164
lastRestored,
161-
len(*s.VolumeBackups),
165+
volBackups,
162166
)
163167
}
164168
err := table.Display(p)

‎internal/cmd/beta/server/backup/list/list_test.go‎

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -186,3 +186,39 @@ func TestBuildRequest(t *testing.T) {
186186
})
187187
}
188188
}
189+
190+
func TestOutputResult(t *testing.T) {
191+
type args struct {
192+
outputFormat string
193+
backups []serverbackup.Backup
194+
}
195+
tests := []struct {
196+
name string
197+
args args
198+
wantErr bool
199+
}{
200+
{
201+
name: "empty",
202+
args: args{},
203+
wantErr: false,
204+
},
205+
{
206+
name: "empty backup",
207+
args: args{
208+
backups: []serverbackup.Backup{
209+
{},
210+
},
211+
},
212+
wantErr: false,
213+
},
214+
}
215+
p := print.NewPrinter()
216+
p.Cmd = NewCmd(p)
217+
for _, tt := range tests {
218+
t.Run(tt.name, func(t *testing.T) {
219+
if err := outputResult(p, tt.args.outputFormat, tt.args.backups); (err != nil) != tt.wantErr {
220+
t.Errorf("outputResult() error = %v, wantErr %v", err, tt.wantErr)
221+
}
222+
})
223+
}
224+
}

‎internal/cmd/beta/server/backup/schedule/create/create.go‎

Lines changed: 18 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@ import (
1212
"github.com/stackitcloud/stackit-cli/internal/pkg/flags"
1313
"github.com/stackitcloud/stackit-cli/internal/pkg/globalflags"
1414
"github.com/stackitcloud/stackit-cli/internal/pkg/print"
15+
iaasClient "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client"
16+
iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils"
1517
"github.com/stackitcloud/stackit-cli/internal/pkg/services/serverbackup/client"
1618
"github.com/stackitcloud/stackit-cli/internal/pkg/utils"
1719

@@ -73,8 +75,19 @@ func NewCmd(p *print.Printer) *cobra.Command {
7375
return err
7476
}
7577

78+
serverLabel := model.ServerId
79+
// Get server name
80+
if iaasApiClient, err := iaasClient.ConfigureClient(p); err == nil {
81+
serverName, err := iaasUtils.GetServerName(ctx, iaasApiClient, model.ProjectId, model.ServerId)
82+
if err != nil {
83+
p.Debug(print.ErrorLevel, "get server name: %v", err)
84+
} else {
85+
serverLabel = serverName
86+
}
87+
}
88+
7689
if !model.AssumeYes {
77-
prompt := fmt.Sprintf("Are you sure you want to create a Backup Schedule for server %s?", model.ServerId)
90+
prompt := fmt.Sprintf("Are you sure you want to create a Backup Schedule for server %s?", serverLabel)
7891
err = p.PromptForConfirmation(prompt)
7992
if err != nil {
8093
return err
@@ -91,7 +104,7 @@ func NewCmd(p *print.Printer) *cobra.Command {
91104
return fmt.Errorf("create Server Backup Schedule: %w", err)
92105
}
93106

94-
return outputResult(p, model, resp)
107+
return outputResult(p, model.OutputFormat, serverLabel, *resp)
95108
},
96109
}
97110
configureFlags(cmd)
@@ -159,8 +172,8 @@ func buildRequest(ctx context.Context, model *inputModel, apiClient *serverbacku
159172
return req, nil
160173
}
161174

162-
func outputResult(p *print.Printer, model *inputModel, resp *serverbackup.BackupSchedule) error {
163-
switch model.OutputFormat {
175+
func outputResult(p *print.Printer, outputFormat, serverLabel string, resp serverbackup.BackupSchedule) error {
176+
switch outputFormat {
164177
case print.JSONOutputFormat:
165178
details, err := json.MarshalIndent(resp, "", " ")
166179
if err != nil {
@@ -178,7 +191,7 @@ func outputResult(p *print.Printer, model *inputModel, resp *serverbackup.Backup
178191

179192
return nil
180193
default:
181-
p.Outputf("Created server backup schedule for server %s. Backup Schedule ID: %s\n", model.ServerId, utils.PtrString(resp.Id))
194+
p.Outputf("Created server backup schedule for server %s. Backup Schedule ID: %s\n", serverLabel, utils.PtrString(resp.Id))
182195
return nil
183196
}
184197
}

0 commit comments

Comments
 (0)