sap-jenkins-library/cmd/azureBlobUpload_test.go

//go:build unit
// +build unit

package cmd

import (
	"context"
	"errors"
	"fmt"
	"io/fs"
	"log"
	"net/http"
	"os"
	"testing"

	"github.com/Azure/azure-sdk-for-go/sdk/storage/azblob"
	"github.com/stretchr/testify/assert"
)

type mockAzureContainerAPI func(blobName string) (*azblob.BlockBlobClient, error)

func (m mockAzureContainerAPI) NewBlockBlobClient(blobName string) (*azblob.BlockBlobClient, error) {
	return m(blobName)
}

func mockAzureContainerClient(t *testing.T, fail bool) azureContainerAPI {
	return mockAzureContainerAPI(func(blobName string) (*azblob.BlockBlobClient, error) {
		t.Helper()
		if fail {
			return nil, fmt.Errorf("error containerClient")
		}
		return &azblob.BlockBlobClient{}, nil
	})
}

func uploadFuncMock(ctx context.Context, api *azblob.BlockBlobClient, file *os.File, o azblob.UploadOption) (*http.Response, error) {
	return &http.Response{}, nil
}

func TestRunAzureBlobUpload(t *testing.T) {
	t.Parallel()

	t.Run("positive tests", func(t *testing.T) {
		t.Parallel()

		t.Run("happy path", func(t *testing.T) {
			t.Parallel()
			// create temporary file
			f, err := os.CreateTemp("", "tmpfile-")
			if err != nil {
				log.Fatal(err)
			}
			defer f.Close()
			defer os.Remove(f.Name())
			data := []byte("test test test")
			if _, err := f.Write(data); err != nil {
				log.Fatal(err)
			}
			// initialization
			config := azureBlobUploadOptions{
				FilePath: f.Name(),
			}
			container := mockAzureContainerClient(t, false)
			// test
			err = executeUpload(&config, container, uploadFuncMock)
			// assert
			assert.NoError(t, err)
		})
	})

	t.Run("negative tests", func(t *testing.T) {
		t.Parallel()

		t.Run("error path", func(t *testing.T) {
			t.Parallel()
			// initialization
			config := azureBlobUploadOptions{
				FilePath: "nonExistingFilepath",
			}
			container := mockAzureContainerClient(t, false)
			// test
			err := executeUpload(&config, container, uploadFuncMock)
			// assert
			assert.IsType(t, &fs.PathError{}, errors.Unwrap(err))
		})

		t.Run("error containerClient", func(t *testing.T) {
			t.Parallel()
			// create temporary file
			f, err := os.CreateTemp("", "tmpfile-")
			if err != nil {
				log.Fatal(err)
			}
			defer f.Close()
			defer os.Remove(f.Name())
			data := []byte("test test test")
			if _, err := f.Write(data); err != nil {
				log.Fatal(err)
			}
			// initialization
			config := azureBlobUploadOptions{
				FilePath: f.Name(),
			}
			container := mockAzureContainerClient(t, true)
			// test
			err = executeUpload(&config, container, uploadFuncMock)
			// assert
			assert.EqualError(t, err, "Could not instantiate Azure blockBlobClient from Azure Container Client: error containerClient")
		})

		t.Run("error credentials", func(t *testing.T) {
			t.Parallel()
			// initialization
			config := azureBlobUploadOptions{
				JSONCredentialsAzure: `{
					"account_name": "name",
					"container_name": "container"
				  }`,
				FilePath: "nonExistingFilepath",
			}
			// test
			_, err := setup(&config)
			// assert
			assert.EqualError(t, err, "Azure credentials are not valid: Key: 'azureCredentials.SASToken' Error:Field validation for 'SASToken' failed on the 'required' tag")
		})

		t.Run("error JSONStruct", func(t *testing.T) {
			t.Parallel()
			// initialization
			config := azureBlobUploadOptions{
				JSONCredentialsAzure: `faulty json`,
			}
			// test
			_, err := setup(&config)
			// assert
			assert.EqualError(t, err, "Could not read JSONCredentialsAzure: invalid character 'u' in literal false (expecting 'l')")
		})
	})
}