rollback the ToFile func implement, and add testcase

This commit is contained in:
smallfish
2014-08-18 21:29:45 +08:00
parent de87529387
commit c16b7be9ac
2 changed files with 26 additions and 2 deletions

View File

@@ -421,11 +421,15 @@ func (b *BeegoHttpRequest) ToFile(filename string) error {
}
defer f.Close()
data, err := b.Bytes()
resp, err := b.getResponse()
if err != nil {
return err
}
_, err = f.Write(data)
if resp.Body == nil {
return nil
}
defer resp.Body.Close()
_, err = io.Copy(f, resp.Body)
return err
}