photoprism/pkg/fs/fileinfo_test.go

package fs

import (
	"io/ioutil"
	"os"
	"testing"
	"time"

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

func TestNewFileInfo(t *testing.T) {
	info, err := os.Stat("testdata/test.jpg")

	if err != nil {
		t.Fatal(err)
	}

	result := NewFileInfo(info, "Photos/")

	assert.Equal(t, "test.jpg", result.Name)
	assert.Equal(t, "/Photos/test.jpg", result.Abs)
	assert.Equal(t, int64(10990), result.Size)
	assert.IsType(t, time.Time{}, result.Date)
	assert.Equal(t, false, result.Dir)
}

func TestNewFileInfos(t *testing.T) {
	infos, err := ioutil.ReadDir("testdata")

	if err != nil {
		t.Fatal(err)
	}

	result := NewFileInfos(infos, "/")

	if len(result) < 1 {
		t.Fatal("empty result")
	}

	expected := map[string]FileInfo{
		"test.jpg":      {Abs: "/test.jpg", Size: 10990, Dir: false},
		"CATYELLOW.jpg": {Abs: "/CATYELLOW.jpg", Size: 70790, Dir: false},
		"directory":     {Abs: "/directory", Size: 4096, Dir: true},
		"linked":        {Abs: "/linked", Size: 4096, Dir: true},
	}

	for _, file := range result {
		assert.NotEmpty(t, file.Name)
		assert.NotEmpty(t, file.Abs)
		assert.NotEmpty(t, file.Size)
		assert.False(t, file.Date.IsZero())

		if info, ok := expected[file.Name]; ok {
			assert.Equal(t, info.Abs, file.Abs, "%s expected for %s", info.Abs, file.Name)
			if info.Dir {
				assert.GreaterOrEqual(t, info.Size, file.Size, "size < %d expected for %s", info.Size, file.Name)
			} else {
				assert.Equal(t, info.Size, file.Size, "%d expected for %s", info.Size, file.Name)
			}
			assert.Equal(t, info.Dir, file.Dir, "%t expected for %s", info.Dir, file.Name)
		}
	}
}

func TestFileInfos_Less(t *testing.T) {
	infos := FileInfos{
		{Name: "test.jpg", Abs: "/test.jpg", Size: 10990, Dir: false},
		{Name: "CATYELLOW.jpg", Abs: "/CATYELLOW.jpg", Size: 70790, Dir: false},
		{Name: "directory", Abs: "/directory", Size: 256, Dir: true},
		{Name: "linked", Abs: "/linked", Size: 256, Dir: true},
	}

	assert.Equal(t, false, infos.Less(0, 0))
	assert.Equal(t, false, infos.Less(0, 1))
	assert.Equal(t, true, infos.Less(1, 0))
	assert.Equal(t, true, infos.Less(2, 3))
	assert.Equal(t, false, infos.Less(3, 2))
}

func TestFileInfos_Swap(t *testing.T) {
	infos := FileInfos{
		{Name: "test.jpg", Abs: "/test.jpg", Size: 10990, Dir: false},
		{Name: "CATYELLOW.jpg", Abs: "/CATYELLOW.jpg", Size: 70790, Dir: false},
		{Name: "directory", Abs: "/directory", Size: 256, Dir: true},
		{Name: "linked", Abs: "/linked", Size: 256, Dir: true},
	}

	assert.Equal(t, "test.jpg", infos[0].Name)
	assert.Equal(t, "CATYELLOW.jpg", infos[1].Name)
	infos.Swap(0, 1)
	assert.Equal(t, "CATYELLOW.jpg", infos[0].Name)
	assert.Equal(t, "test.jpg", infos[1].Name)
}

func TestFileInfos_Len(t *testing.T) {
	infos := FileInfos{
		{Name: "test.jpg", Abs: "/test.jpg", Size: 10990, Dir: false},
		{Name: "CATYELLOW.jpg", Abs: "/CATYELLOW.jpg", Size: 70790, Dir: false},
		{Name: "directory", Abs: "/directory", Size: 256, Dir: true},
		{Name: "linked", Abs: "/linked", Size: 256, Dir: true},
	}

	assert.Equal(t, 4, infos.Len())
}

func TestFileInfos_Abs(t *testing.T) {
	infos := FileInfos{
		{Name: "test.jpg", Abs: "/test.jpg", Size: 10990, Dir: false},
		{Name: "CATYELLOW.jpg", Abs: "/CATYELLOW.jpg", Size: 70790, Dir: false},
		{Name: "directory", Abs: "/directory", Size: 256, Dir: true},
		{Name: "linked", Abs: "/linked", Size: 256, Dir: true},
	}

	assert.Equal(t, []string{"/test.jpg", "/CATYELLOW.jpg", "/directory", "/linked"}, infos.Abs())
}