mirror of
https://github.com/aptly-dev/aptly.git
synced 2026-05-06 22:18:28 +00:00
Refactor Downloader: it accepts channel, not generates it.
This commit is contained in:
+12
-4
@@ -48,32 +48,40 @@ func (s *DownloaderSuite) TestStartupShutdown(c *C) {
|
||||
func (s *DownloaderSuite) TestDownloadOK(c *C) {
|
||||
d := NewDownloader(2)
|
||||
defer d.Shutdown()
|
||||
ch := make(chan error)
|
||||
|
||||
res := <-d.Download("http://smira.ru/", s.tempfile.Name())
|
||||
d.Download("http://smira.ru/", s.tempfile.Name(), ch)
|
||||
res := <-ch
|
||||
c.Assert(res, IsNil)
|
||||
}
|
||||
|
||||
func (s *DownloaderSuite) TestDownload404(c *C) {
|
||||
d := NewDownloader(2)
|
||||
defer d.Shutdown()
|
||||
ch := make(chan error)
|
||||
|
||||
res := <-d.Download("http://smira.ru/doesntexist", s.tempfile.Name())
|
||||
d.Download("http://smira.ru/doesntexist", s.tempfile.Name(), ch)
|
||||
res := <-ch
|
||||
c.Assert(res, ErrorMatches, "HTTP code 404.*")
|
||||
}
|
||||
|
||||
func (s *DownloaderSuite) TestDownloadConnectError(c *C) {
|
||||
d := NewDownloader(2)
|
||||
defer d.Shutdown()
|
||||
ch := make(chan error)
|
||||
|
||||
res := <-d.Download("http://nosuch.smira.ru/", s.tempfile.Name())
|
||||
d.Download("http://nosuch.smira.ru/", s.tempfile.Name(), ch)
|
||||
res := <-ch
|
||||
c.Assert(res, ErrorMatches, ".*no such host")
|
||||
}
|
||||
|
||||
func (s *DownloaderSuite) TestDownloadFileError(c *C) {
|
||||
d := NewDownloader(2)
|
||||
defer d.Shutdown()
|
||||
ch := make(chan error)
|
||||
|
||||
res := <-d.Download("http://smira.ru/", "/")
|
||||
d.Download("http://smira.ru/", "/", ch)
|
||||
res := <-ch
|
||||
c.Assert(res, ErrorMatches, ".*permission denied")
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user