client_test.go

package getter

import (
	"context"
	"os"
	"path/filepath"
	"testing"

	testing_helper "github.com/hashicorp/go-getter/v2/helper/testing"
)

func TestSmb_ClientGet(t *testing.T) {
	smbTestsPreCheck(t)

	tests := []struct {
		name   string
		rawURL string
		mode   Mode
		file   string
		fail   bool
	}{
		{
			"smb scheme subdir with registered authentication in private share",
			"smb://user:password@samba/private/subdir",
			ModeDir,
			"file.txt",
			false,
		},
		{
			"smb scheme file with registered authentication with file in private share",
			"smb://user:password@samba/private/subdir/file.txt",
			ModeFile,
			"file.txt",
			false,
		},
		{
			"smb scheme file without authentication in public share",
			"smb://samba/public/subdir/file.txt",
			ModeFile,
			"file.txt",
			false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			dst := testing_helper.TempDir(t)
			defer os.RemoveAll(dst)

			if tt.mode == ModeFile {
				dst = filepath.Join(dst, tt.file)
			}

			req := &Request{
				Dst:     dst,
				Src:     tt.rawURL,
				GetMode: tt.mode,
			}

			result, err := DefaultClient.Get(context.Background(), req)

			fail := err != nil
			if tt.fail != fail {
				if fail {
					t.Fatalf("err: unexpected error %s", err.Error())
				}
				t.Fatalf("err: expecting to fail but it did not")
			}

			if !tt.fail {
				if result == nil {
					t.Fatalf("err: get result should not be nil")
				}
				if result.Dst != dst {
					t.Fatalf("err: expected destination: %s \n actual destination: %s", dst, result.Dst)
				}
				if tt.mode == ModeDir {
					dst = filepath.Join(dst, tt.file)
				}
				// Verify if the file was successfully downloaded
				// and exists at the destination folder
				testing_helper.AssertContents(t, dst, "Hello\n")
			}
		})
	}
}