Adding test for ota file generation

Author: rjtokenring

command/ota/massupload.go

@@ -54,6 +54,26 @@ type Result struct {
 	OtaStatus otaapi.Ota
 }
 
+func buildOtaFile(params *MassUploadParams) (string, error) {
+	var otaFile string
+	if params.DoNotApplyHeader {
+		otaFile = params.File
+	} else {
+		otaDir, err := os.MkdirTemp("", "")
+		if err != nil {
+			return "", fmt.Errorf("%s: %w", "cannot create temporary folder", err)
+		}
+		otaFile = filepath.Join(otaDir, "temp.ota")
+		defer os.RemoveAll(otaDir)
+
+		err = Generate(params.File, otaFile, params.FQBN)
+		if err != nil {
+			return "", fmt.Errorf("%s: %w", "cannot generate .ota file", err)
+		}
+	}
+	return otaFile, nil
+}
+
 // MassUpload command is used to mass upload a firmware OTA,
 // on devices of Arduino IoT Cloud.
 func MassUpload(ctx context.Context, params *MassUploadParams, cred *config.Credentials) ([]Result, error) {
@@ -78,21 +98,9 @@ func MassUpload(ctx context.Context, params *MassUploadParams, cred *config.Cred
 	}
 
 	// Generate .ota file
-	var otaFile string
-	if params.DoNotApplyHeader {
-		otaFile = params.File
-	} else {
-		otaDir, err := os.MkdirTemp("", "")
-		if err != nil {
-			return nil, fmt.Errorf("%s: %w", "cannot create temporary folder", err)
-		}
-		otaFile = filepath.Join(otaDir, "temp.ota")
-		defer os.RemoveAll(otaDir)
-
-		err = Generate(params.File, otaFile, params.FQBN)
-		if err != nil {
-			return nil, fmt.Errorf("%s: %w", "cannot generate .ota file", err)
-		}
+	otaFile, err := buildOtaFile(params)
+	if err != nil {
+		return nil, err
 	}
 
 	iotClient, err := iot.NewClient(cred)

command/ota/massupload_test.go

@@ -10,9 +10,11 @@ import (
 	otaapi "github.com/arduino/arduino-cloud-cli/internal/ota-api"
 	iotclient "github.com/arduino/iot-client-go"
 	"github.com/gofrs/uuid"
+	"github.com/stretchr/testify/assert"
 )
 
 const testFilename = "testdata/empty.bin"
+const cloudFirmwareFilename = "testdata/cloud.bin"
 
 type deviceUploaderTest struct {
 	deviceOTA func(ctx context.Context, id string, file *os.File, expireMins int) error
@@ -119,3 +121,15 @@ func TestValidateDevices(t *testing.T) {
 		t.Errorf("expected 2 invalid devices, but found %d: %v", len(i), i)
 	}
 }
+
+func TestValidateBuildOtaFile(t *testing.T) {
+
+	file, err := buildOtaFile(&MassUploadParams{
+		File:             cloudFirmwareFilename,
+		DoNotApplyHeader: false,
+		FQBN:             "arduino:samd:nano_33_iot",
+	})
+	assert.Nil(t, err)
+	assert.NotNil(t, file)
+	assert.True(t, strings.HasSuffix(file, "temp.ota"))
+}