Fix tests

License: MIT
Signed-off-by: Hector Sanjuan <code@hector.link>
This commit is contained in:
Hector Sanjuan 2018-03-13 16:30:47 +01:00
parent c231fe8060
commit a746255719

View File

@ -1,6 +1,7 @@
package disk
import (
"context"
"errors"
"testing"
@ -43,13 +44,13 @@ func badRPCClient(t *testing.T) *rpc.Client {
// return nil
// }
func (mock *badRPCService) IPFSRepoSize(in struct{}, out *uint64) error {
func (mock *badRPCService) IPFSRepoSize(ctx context.Context, in struct{}, out *uint64) error {
*out = 2
mock.nthCall++
return errors.New("fake error")
}
func (mock *badRPCService) IPFSFreeSpace(in struct{}, out *uint64) error {
func (mock *badRPCService) IPFSFreeSpace(ctx context.Context, in struct{}, out *uint64) error {
*out = 2
mock.nthCall++
return errors.New("fake error")