ente/cli/pkg/disk_test.go

package pkg

import (
	"path/filepath"
	"strings"
	"testing"
)

func TestGenerateUniqueFileName(t *testing.T) {
	existingFilenames := make(map[string]bool)
	testFilename := "FullSizeRender.jpg" // what Apple calls shared files

	existingFilenames[strings.ToLower(testFilename)] = true

	a := &albumDiskInfo{
		FileNames: &existingFilenames,
	}

	// this is taken from downloadEntry()
	extension := filepath.Ext(testFilename)
	baseFileName := strings.TrimSuffix(filepath.Clean(filepath.Base(testFilename)), extension)

	for i := 0; i < 100; i++ {
		newFilename := a.GenerateUniqueFileName(baseFileName, extension)
		if strings.Contains(newFilename, "_1_2") {
			t.Fatalf("Filename contained _1_2")
		} else {
			// add generated name to existing files
			existingFilenames[strings.ToLower(newFilename)] = true
		}
	}
}