c3d3f41d30
Removes most instances of pb.SignedMessage (there's more to take out but they shouldn't hurt anyone as is). There used to be places in psserver where a PieceID was hmac'd with the SatelliteID, which was gotten from a SignedMessage. This PR makes it so some functions access the SatelliteID from the Payer Bandwidth Allocation instead. This requires passing a SatelliteID into psserver functions where they weren't before, so the following proto messages have been changed: * PieceId - satellite_id field added This is so the psserver.Piece function has access to the SatelliteID when it needs to get the namespaced pieceID. This proto message should probably be renamed to PieceRequest, or a new PieceRequest message should be created so this isn't misnamed. * PieceDelete - satellite_id field added This is so the psserver.Delete function has access to the SatelliteID when receiving a request to Delete.
113 lines
3.7 KiB
Go
113 lines
3.7 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: storj.io/storj/pkg/piecestore/psclient (interfaces: Client)
|
|
|
|
// Package ecclient is a generated GoMock package.
|
|
package ecclient
|
|
|
|
import (
|
|
context "context"
|
|
gomock "github.com/golang/mock/gomock"
|
|
io "io"
|
|
reflect "reflect"
|
|
pb "storj.io/storj/pkg/pb"
|
|
psclient "storj.io/storj/pkg/piecestore/psclient"
|
|
ranger "storj.io/storj/pkg/ranger"
|
|
storj "storj.io/storj/pkg/storj"
|
|
time "time"
|
|
)
|
|
|
|
// MockPSClient is a mock of Client 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
|
|
}
|
|
|
|
// Close mocks base method
|
|
func (m *MockPSClient) Close() error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Close")
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Close indicates an expected call of Close
|
|
func (mr *MockPSClientMockRecorder) Close() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockPSClient)(nil).Close))
|
|
}
|
|
|
|
// Delete mocks base method
|
|
func (m *MockPSClient) Delete(arg0 context.Context, arg1 psclient.PieceID, arg2 storj.NodeID) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Delete", arg0, arg1, arg2)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Delete indicates an expected call of Delete
|
|
func (mr *MockPSClientMockRecorder) Delete(arg0, arg1, arg2 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockPSClient)(nil).Delete), arg0, arg1, arg2)
|
|
}
|
|
|
|
// Get mocks base method
|
|
func (m *MockPSClient) Get(arg0 context.Context, arg1 psclient.PieceID, arg2 int64, arg3 *pb.PayerBandwidthAllocation) (ranger.Ranger, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Get", arg0, arg1, arg2, arg3)
|
|
ret0, _ := ret[0].(ranger.Ranger)
|
|
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 {
|
|
mr.mock.ctrl.T.Helper()
|
|
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 psclient.PieceID) (*pb.PieceSummary, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Meta", arg0, arg1)
|
|
ret0, _ := ret[0].(*pb.PieceSummary)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Meta indicates an expected call of Meta
|
|
func (mr *MockPSClientMockRecorder) Meta(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
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 psclient.PieceID, arg2 io.Reader, arg3 time.Time, arg4 *pb.PayerBandwidthAllocation) error {
|
|
m.ctrl.T.Helper()
|
|
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 {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Put", reflect.TypeOf((*MockPSClient)(nil).Put), arg0, arg1, arg2, arg3, arg4)
|
|
}
|