Function TestGetDownloadRetriesBodyReadError

cmd/get_test.go:209–256

(t *testing.T)

}

func TestGetDownloadRetriesBodyReadError(t *testing.T) {
	delays := stubRetrySleep(t)
	tmpDir := t.TempDir()
	dst := filepath.Join(tmpDir, "downloaded.txt")
	content := "complete file content"

	calls := 0
	mock := &mockFilesClient{
		downloadFn: func(arg *files.DownloadArg) (*files.FileMetadata, io.ReadCloser, error) {
			calls++
			meta := &files.FileMetadata{
				Metadata: files.Metadata{PathDisplay: "/test.txt"},
				Size:     uint64(len(content)),
			}
			if calls == 1 {
				return meta, &failingReadCloser{data: []byte("partial")}, nil
			}
			return meta, io.NopCloser(strings.NewReader(content)), nil
		},
	}

	err := downloadFile(mock, "/test.txt", dst)
	if err != nil {
		t.Errorf("expected nil error, got %v", err)
	}
	if calls != 2 {
		t.Errorf("expected 2 calls, got %d", calls)
	}
	if len(*delays) != 1 {
		t.Fatalf("expected 1 sleep, got %d", len(*delays))
	}

	got, err := os.ReadFile(dst)
	if err != nil {
		t.Fatalf("failed to read downloaded file: %v", err)
	}
	if string(got) != content {
		t.Errorf("got %q, want %q", string(got), content)
	}

	entries, err := os.ReadDir(tmpDir)
	if err != nil {
		t.Fatalf("failed to read temp dir: %v", err)
	}
	if len(entries) != 1 || entries[0].Name() != filepath.Base(dst) {
		t.Fatalf("expected only final destination file, got %v", entries)
	}
}

func TestGetDownloadPreservesDestinationSymlink(t *testing.T) {
	tmpDir := t.TempDir()