b0db33f919
* captplanet standalone farmer setup * Bandwidth Allocation * utils.Close method changed to utils.LogClose * Get build temporarily working * Get/Put for PSClient should take payer bandwidth allocations rather than the NewPSClient function * Update example client to reflect changes in client API * Update ecclient to use latest PSClient, Make NewPSClient return error also * Updated pieceranger tests to check for errors; sign method should take byte array * Handle defers in store.go better * Fix defer functions in psdb.go * fun times * Protobuf bandwidthallocation data is now a byte array * Remove psservice package and merge it into pstore server * Write wrapper for database calls * Change all expiration names in protobuf to be more informative; add defer in retrieve; remove old comment * Make PSDB tests implementation independent rather than method independent * get rid of payer, renter in ecclient * add context monitoring in store and retrieve
106 lines
3.6 KiB
Go
106 lines
3.6 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: storj.io/storj/pkg/piecestore/rpc/client (interfaces: PSClient)
|
|
|
|
// Package ecclient is a generated GoMock package.
|
|
package ecclient
|
|
|
|
import (
|
|
context "context"
|
|
io "io"
|
|
reflect "reflect"
|
|
time "time"
|
|
|
|
gomock "github.com/golang/mock/gomock"
|
|
client "storj.io/storj/pkg/piecestore/rpc/client"
|
|
ranger "storj.io/storj/pkg/ranger"
|
|
piecestore "storj.io/storj/protos/piecestore"
|
|
)
|
|
|
|
// MockPSClient is a mock of PSClient interface
|
|
type MockPSClient struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockPSClientMockRecorder
|
|
}
|
|
|
|
// MockPSClientMockRecorder is the mock recorder for MockPSClient
|
|
type MockPSClientMockRecorder struct {
|
|
mock *MockPSClient
|
|
}
|
|
|
|
// NewMockPSClient creates a new mock instance
|
|
func NewMockPSClient(ctrl *gomock.Controller) *MockPSClient {
|
|
mock := &MockPSClient{ctrl: ctrl}
|
|
mock.recorder = &MockPSClientMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use
|
|
func (m *MockPSClient) EXPECT() *MockPSClientMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CloseConn mocks base method
|
|
func (m *MockPSClient) CloseConn() error {
|
|
ret := m.ctrl.Call(m, "CloseConn")
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// CloseConn indicates an expected call of CloseConn
|
|
func (mr *MockPSClientMockRecorder) CloseConn() *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseConn", reflect.TypeOf((*MockPSClient)(nil).CloseConn))
|
|
}
|
|
|
|
// Delete mocks base method
|
|
func (m *MockPSClient) Delete(arg0 context.Context, arg1 client.PieceID) error {
|
|
ret := m.ctrl.Call(m, "Delete", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Delete indicates an expected call of Delete
|
|
func (mr *MockPSClientMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockPSClient)(nil).Delete), arg0, arg1)
|
|
}
|
|
|
|
// Get mocks base method
|
|
func (m *MockPSClient) Get(arg0 context.Context, arg1 client.PieceID, arg2 int64, arg3 *piecestore.PayerBandwidthAllocation) (ranger.RangeCloser, error) {
|
|
ret := m.ctrl.Call(m, "Get", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(ranger.RangeCloser)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Get indicates an expected call of Get
|
|
func (mr *MockPSClientMockRecorder) Get(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockPSClient)(nil).Get), arg0, arg1, arg2, arg3)
|
|
}
|
|
|
|
// Meta mocks base method
|
|
func (m *MockPSClient) Meta(arg0 context.Context, arg1 client.PieceID) (*piecestore.PieceSummary, error) {
|
|
ret := m.ctrl.Call(m, "Meta", arg0, arg1)
|
|
ret0, _ := ret[0].(*piecestore.PieceSummary)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Meta indicates an expected call of Meta
|
|
func (mr *MockPSClientMockRecorder) Meta(arg0, arg1 interface{}) *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Meta", reflect.TypeOf((*MockPSClient)(nil).Meta), arg0, arg1)
|
|
}
|
|
|
|
// Put mocks base method
|
|
func (m *MockPSClient) Put(arg0 context.Context, arg1 client.PieceID, arg2 io.Reader, arg3 time.Time, arg4 *piecestore.PayerBandwidthAllocation) error {
|
|
ret := m.ctrl.Call(m, "Put", arg0, arg1, arg2, arg3, arg4)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Put indicates an expected call of Put
|
|
func (mr *MockPSClientMockRecorder) Put(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call {
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Put", reflect.TypeOf((*MockPSClient)(nil).Put), arg0, arg1, arg2, arg3, arg4)
|
|
}
|