internal/image/download_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

package image

import "testing"

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

	d := NewDownloader(&mockSearcher{name: nanoBananaSource}, &DownloadOptions{
		FileNamePattern: "{word}_{source}",
	})
	result := &SearchResult{
		URL:    "data:image/png;base64,AAAA",
		Source: nanoBananaSource,
	}

	if got := d.generateFileName("ябълка", result, 0); got != "ябълка_nanobanana.png" {
		t.Fatalf("generateFileName() = %q, want %q", got, "ябълка_nanobanana.png")
	}
}