From d350e9edba66e8c9c2589c294f6c36c8e7084214 Mon Sep 17 00:00:00 2001 From: Andrey Smirnov Date: Tue, 4 Feb 2014 13:06:55 +0400 Subject: [PATCH] Fix broken code due to addition of Progress. --- debian/remote.go | 2 +- utils/fake.go | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/debian/remote.go b/debian/remote.go index 24572d27..ec13bb01 100644 --- a/debian/remote.go +++ b/debian/remote.go @@ -236,7 +236,7 @@ func (repo *RemoteRepo) Download(d utils.Downloader, packageCollection *PackageC // Save package meta information to DB err := list.ForEach(func(p *Package) error { - d.GetProgress().Add(1) + d.GetProgress().AddBar(1) return packageCollection.Update(p) }) diff --git a/utils/fake.go b/utils/fake.go index ef199c31..db44d7ae 100644 --- a/utils/fake.go +++ b/utils/fake.go @@ -17,6 +17,7 @@ type expectedRequest struct { // to stub out results type FakeDownloader struct { expected []expectedRequest + progress *Progress } // Check interface @@ -28,6 +29,8 @@ var ( func NewFakeDownloader() *FakeDownloader { result := &FakeDownloader{} result.expected = make([]expectedRequest, 0) + result.progress = NewProgress() + result.progress.Start() return result } @@ -108,6 +111,7 @@ func (f *FakeDownloader) Download(url string, filename string, result chan<- err // Shutdown does nothing func (f *FakeDownloader) Shutdown() { + f.progress.Shutdown() } // Pause does nothing @@ -120,5 +124,5 @@ func (f *FakeDownloader) Resume() { // GetProgress does nothing func (f *FakeDownloader) GetProgress() *Progress { - return nil + return f.progress }