@@ -8,6 +8,7 @@ package config
8
8
9
9
import (
10
10
"fmt"
11
+ "runtime"
11
12
"testing"
12
13
13
14
"github.com/golang/mock/gomock"
@@ -105,16 +106,21 @@ func TestDiskLimaConfigApplier_Apply(t *testing.T) {
105
106
require .NoError (t , err )
106
107
cmd .EXPECT ().Output ().Return ([]byte ("13.0.0" ), nil )
107
108
creator .EXPECT ().Create ("sw_vers" , "-productVersion" ).Return (cmd )
108
- deps .EXPECT ().Arch ()
109
+ deps .EXPECT ().Arch (). Return ( runtime . GOARCH )
109
110
},
110
111
postRunCheck : func (t * testing.T , fs afero.Fs ) {
111
- sociFileName := fmt .Sprintf (sociFileNameFormat , sociVersion , "" )
112
+ sociFileName := fmt .Sprintf (sociFileNameFormat , sociVersion , runtime . GOARCH )
112
113
sociDownloadURL := fmt .Sprintf (sociDownloadURLFormat , sociVersion , sociFileName )
114
+ sociShaSum := sociAMD64Sha256Sum
115
+ if runtime .GOARCH == "arm64" {
116
+ sociShaSum = sociARM64Sha256Sum
117
+ }
113
118
sociServiceDownloadURL := fmt .Sprintf (sociServiceDownloadURLFormat , sociVersion )
114
119
sociInstallationScript := fmt .Sprintf (sociInstallationScriptFormat ,
115
120
sociInstallationProvisioningScriptHeader ,
116
- sociDownloadURL ,
117
121
sociFileName ,
122
+ sociDownloadURL ,
123
+ sociShaSum ,
118
124
sociServiceDownloadURL )
119
125
120
126
buf , err := afero .ReadFile (fs , "/override.yaml" )
@@ -257,16 +263,21 @@ func TestDiskLimaConfigApplier_Apply(t *testing.T) {
257
263
require .NoError (t , err )
258
264
cmd .EXPECT ().Output ().Return ([]byte ("13.0.0" ), nil )
259
265
creator .EXPECT ().Create ("sw_vers" , "-productVersion" ).Return (cmd )
260
- deps .EXPECT ().Arch ()
266
+ deps .EXPECT ().Arch (). Return ( runtime . GOARCH )
261
267
},
262
268
postRunCheck : func (t * testing.T , fs afero.Fs ) {
263
- sociFileName := fmt .Sprintf (sociFileNameFormat , sociVersion , "" )
269
+ sociFileName := fmt .Sprintf (sociFileNameFormat , sociVersion , runtime . GOARCH )
264
270
sociDownloadURL := fmt .Sprintf (sociDownloadURLFormat , sociVersion , sociFileName )
271
+ sociShaSum := sociAMD64Sha256Sum
272
+ if runtime .GOARCH == "arm64" {
273
+ sociShaSum = sociARM64Sha256Sum
274
+ }
265
275
sociServiceDownloadURL := fmt .Sprintf (sociServiceDownloadURLFormat , sociVersion )
266
276
sociInstallationScript := fmt .Sprintf (sociInstallationScriptFormat ,
267
277
sociInstallationProvisioningScriptHeader ,
268
- sociDownloadURL ,
269
278
sociFileName ,
279
+ sociDownloadURL ,
280
+ sociShaSum ,
270
281
sociServiceDownloadURL )
271
282
272
283
buf , err := afero .ReadFile (fs , "/override.yaml" )
@@ -316,16 +327,21 @@ func TestDiskLimaConfigApplier_Apply(t *testing.T) {
316
327
require .NoError (t , err )
317
328
cmd .EXPECT ().Output ().Return ([]byte ("13.0.0" ), nil )
318
329
creator .EXPECT ().Create ("sw_vers" , "-productVersion" ).Return (cmd )
319
- deps .EXPECT ().Arch ()
330
+ deps .EXPECT ().Arch (). Return ( runtime . GOARCH )
320
331
},
321
332
postRunCheck : func (t * testing.T , fs afero.Fs ) {
322
- sociFileName := fmt .Sprintf (sociFileNameFormat , sociVersion , "" )
333
+ sociFileName := fmt .Sprintf (sociFileNameFormat , sociVersion , runtime . GOARCH )
323
334
sociDownloadURL := fmt .Sprintf (sociDownloadURLFormat , sociVersion , sociFileName )
335
+ sociShaSum := sociAMD64Sha256Sum
336
+ if runtime .GOARCH == "arm64" {
337
+ sociShaSum = sociARM64Sha256Sum
338
+ }
324
339
sociServiceDownloadURL := fmt .Sprintf (sociServiceDownloadURLFormat , sociVersion )
325
340
sociInstallationScript := fmt .Sprintf (sociInstallationScriptFormat ,
326
341
sociInstallationProvisioningScriptHeader ,
327
- sociDownloadURL ,
328
342
sociFileName ,
343
+ sociDownloadURL ,
344
+ sociShaSum ,
329
345
sociServiceDownloadURL )
330
346
331
347
buf , err := afero .ReadFile (fs , "/override.yaml" )
@@ -392,7 +408,7 @@ func TestDiskLimaConfigApplier_Apply(t *testing.T) {
392
408
require .NoError (t , err )
393
409
cmd .EXPECT ().Output ().Return ([]byte ("13.0.0" ), nil )
394
410
creator .EXPECT ().Create ("sw_vers" , "-productVersion" ).Return (cmd )
395
- deps .EXPECT ().Arch ().Return ("arm64" )
411
+ deps .EXPECT ().Arch ().Return (runtime . GOARCH )
396
412
},
397
413
postRunCheck : func (t * testing.T , fs afero.Fs ) {
398
414
buf , err := afero .ReadFile (fs , "/override.yaml" )
0 commit comments