mock_client_test.go

// Copyright 2022 The Moov Authors
// Use of this source code is governed by an Apache License
// license that can be found in the LICENSE file.

package go_ftp_test

import (
	"errors"
	"io"
	"io/fs"
	"strings"
	"testing"

	ftp "github.com/moov-io/go-ftp"

	"github.com/stretchr/testify/require"
)

func TestMockClient(t *testing.T) {
	client := ftp.NewMockClient(t)
	require.NoError(t, client.Ping())
	defer require.NoError(t, client.Close())

	_, err := client.Open("/missing.txt")
	require.Error(t, err)

	err = client.Delete("/missing.txt")
	require.Error(t, err)

	body := io.NopCloser(strings.NewReader("contents"))
	err = client.UploadFile("/exists.txt", body)
	require.NoError(t, err)

	client.Err = errors.New("bad error")
	err = client.UploadFile("/exists.txt", body)
	require.Error(t, err)

	// reset mock client err
	client.Err = nil

	paths, err := client.ListFiles("/")
	require.NoError(t, err)
	require.Len(t, paths, 1)
	require.Equal(t, "/exists.txt", paths[0])
}

func TestMockClient_ListAndOpenFiles(t *testing.T) {
	client := ftp.NewMockClient(t)
	require.NoError(t, client.Ping())
	defer require.NoError(t, client.Close())

	require.NoError(t, client.UploadFile("/path/f1.txt", io.NopCloser(strings.NewReader("foo"))))
	require.NoError(t, client.UploadFile("/path/f2.txt", io.NopCloser(strings.NewReader("foo"))))

	foundFiles, err := client.ListFiles("/path/")
	require.NoError(t, err)
	require.Len(t, foundFiles, 2)

	for _, file := range foundFiles {
		found, err := client.Open(file)
		require.NoError(t, err)
		require.NotNil(t, found)

		contents, err := io.ReadAll(found.Contents)
		require.NoError(t, err)
		require.Equal(t, "foo", string(contents))
		require.NoError(t, found.Close())

		// Consume the file with Reader
		found, err = client.Reader(file)
		require.NoError(t, err)
		require.NotNil(t, found)

		contents, err = io.ReadAll(found.Contents)
		require.NoError(t, err)
		require.Equal(t, "foo", string(contents))
		require.NoError(t, found.Close())
	}

	// Walk the directory and list files
	var walkedFiles []string
	err = client.Walk("/path/", func(path string, d fs.DirEntry, err error) error {
		if err != nil {
			return err
		}
		walkedFiles = append(walkedFiles, path)
		return nil
	})
	require.NoError(t, err)
	require.Contains(t, walkedFiles, "f1.txt", "f2.txt")
}