@@ -16,9 +16,11 @@ import (
16
16
"github.com/hyperledger/fabric/common/flogging"
17
17
"github.com/hyperledger/fabric/common/util"
18
18
"github.com/hyperledger/fabric/msp"
19
- "github.com/hyperledger/fabric/peer/chaincode/api"
19
+ ccapi "github.com/hyperledger/fabric/peer/chaincode/api"
20
20
"github.com/hyperledger/fabric/peer/chaincode/mock"
21
21
"github.com/hyperledger/fabric/peer/common"
22
+ "github.com/hyperledger/fabric/peer/common/api"
23
+ cmock "github.com/hyperledger/fabric/peer/common/mock"
22
24
cb "github.com/hyperledger/fabric/protos/common"
23
25
pb "github.com/hyperledger/fabric/protos/peer"
24
26
"github.com/hyperledger/fabric/protos/utils"
@@ -65,12 +67,14 @@ func TestInvokeCmd(t *testing.T) {
65
67
getBroadcastClient := common .GetBroadcastClientFnc
66
68
getDefaultSigner := common .GetDefaultSignerFnc
67
69
getDeliverClient := common .GetDeliverClientFnc
70
+ getPeerDeliverClient := common .GetPeerDeliverClientFnc
68
71
defer func () {
69
72
common .GetEndorserClientFnc = getEndorserClient
70
73
common .GetOrdererEndpointOfChainFnc = getOrdererEndpointOfChain
71
74
common .GetBroadcastClientFnc = getBroadcastClient
72
75
common .GetDefaultSignerFnc = getDefaultSigner
73
76
common .GetDeliverClientFnc = getDeliverClient
77
+ common .GetPeerDeliverClientFnc = getPeerDeliverClient
74
78
}()
75
79
common .GetEndorserClientFnc = func (string , string ) (pb.EndorserClient , error ) {
76
80
return mockCF .EndorserClients [0 ], nil
@@ -95,18 +99,26 @@ func TestInvokeCmd(t *testing.T) {
95
99
96
100
// Error case 3: getDeliverClient returns error
97
101
t .Logf ("Start error case 3: getDeliverClient returns error" )
98
- common .GetDeliverClientFnc = func (string , string ) (api. DeliverClient , error ) {
102
+ common .GetDeliverClientFnc = func (string , string ) (pb. Deliver_DeliverClient , error ) {
99
103
return nil , errors .New ("error" )
100
104
}
101
105
err = cmd .Execute ()
102
106
assert .Error (t , err )
103
107
104
- // Error case 4: getDefaultSignerFnc returns error
105
- t .Logf ("Start error case 4: getDefaultSignerFnc returns error" )
108
+ // Error case 4 : getPeerDeliverClient returns error
109
+ t .Logf ("Start error case 4: getPeerDeliverClient returns error" )
110
+ common .GetPeerDeliverClientFnc = func (string , string ) (api.PeerDeliverClient , error ) {
111
+ return nil , errors .New ("error" )
112
+ }
113
+ err = cmd .Execute ()
114
+ assert .Error (t , err )
115
+
116
+ // Error case 5: getDefaultSignerFnc returns error
117
+ t .Logf ("Start error case 5: getDefaultSignerFnc returns error" )
106
118
common .GetEndorserClientFnc = func (string , string ) (pb.EndorserClient , error ) {
107
119
return mockCF .EndorserClients [0 ], nil
108
120
}
109
- common .GetDeliverClientFnc = func (string , string ) (api.DeliverClient , error ) {
121
+ common .GetPeerDeliverClientFnc = func (string , string ) (api.PeerDeliverClient , error ) {
110
122
return mockCF .DeliverClients [0 ], nil
111
123
}
112
124
common .GetDefaultSignerFnc = func () (msp.SigningIdentity , error ) {
@@ -116,8 +128,8 @@ func TestInvokeCmd(t *testing.T) {
116
128
assert .Error (t , err )
117
129
common .GetDefaultSignerFnc = common .GetDefaultSigner
118
130
119
- // Error case 5 : getOrdererEndpointOfChainFnc returns error
120
- t .Logf ("Start error case 5 : getOrdererEndpointOfChainFnc returns error" )
131
+ // Error case 6 : getOrdererEndpointOfChainFnc returns error
132
+ t .Logf ("Start error case 6 : getOrdererEndpointOfChainFnc returns error" )
121
133
common .GetEndorserClientFnc = func (string , string ) (pb.EndorserClient , error ) {
122
134
return mockCF .EndorserClients [0 ], nil
123
135
}
@@ -127,8 +139,8 @@ func TestInvokeCmd(t *testing.T) {
127
139
err = cmd .Execute ()
128
140
assert .Error (t , err )
129
141
130
- // Error case 6 : getBroadcastClient returns error
131
- t .Logf ("Start error case 6 : getBroadcastClient returns error" )
142
+ // Error case 7 : getBroadcastClient returns error
143
+ t .Logf ("Start error case 7 : getBroadcastClient returns error" )
132
144
common .GetOrdererEndpointOfChainFnc = func (chainID string , signer msp.SigningIdentity , endorserClient pb.EndorserClient ) ([]string , error ) {
133
145
return []string {"localhost:9999" }, nil
134
146
}
@@ -204,7 +216,7 @@ func getMockChaincodeCmdFactory() (*ChaincodeCmdFactory, error) {
204
216
mockEndorserClients := []pb.EndorserClient {common .GetMockEndorserClient (mockResponse , nil ), common .GetMockEndorserClient (mockResponse , nil )}
205
217
mockBroadcastClient := common .GetMockBroadcastClient (nil )
206
218
mockDC := getMockDeliverClient ()
207
- mockDeliverClients := []api.DeliverClient {mockDC , mockDC }
219
+ mockDeliverClients := []api.PeerDeliverClient {mockDC , mockDC }
208
220
mockCF := & ChaincodeCmdFactory {
209
221
EndorserClients : mockEndorserClients ,
210
222
Signer : signer ,
@@ -225,7 +237,7 @@ func getMockChaincodeCmdFactoryWithErr() (*ChaincodeCmdFactory, error) {
225
237
errMsg := "invoke error"
226
238
mockEndorserClients := []pb.EndorserClient {common .GetMockEndorserClient (nil , errors .New (errMsg ))}
227
239
mockBroadcastClient := common .GetMockBroadcastClient (nil )
228
- mockDeliverClients := []api.DeliverClient {getMockDeliverClient ()}
240
+ mockDeliverClients := []api.PeerDeliverClient {getMockDeliverClient ()}
229
241
mockCF := & ChaincodeCmdFactory {
230
242
EndorserClients : mockEndorserClients ,
231
243
Signer : signer ,
@@ -261,7 +273,7 @@ func getMockChaincodeCmdFactoryEndorsementFailure(ccRespStatus int32, ccRespPayl
261
273
262
274
mockEndorserClients := []pb.EndorserClient {common .GetMockEndorserClient (mockRespFailure , nil )}
263
275
mockBroadcastClient := common .GetMockBroadcastClient (nil )
264
- mockDeliverClients := []api.DeliverClient {getMockDeliverClient ()}
276
+ mockDeliverClients := []api.PeerDeliverClient {getMockDeliverClient ()}
265
277
mockCF := & ChaincodeCmdFactory {
266
278
EndorserClients : mockEndorserClients ,
267
279
Signer : signer ,
@@ -280,13 +292,13 @@ func createCIS() *pb.ChaincodeInvocationSpec {
280
292
}
281
293
282
294
// creates a mock deliver client with a response that contains txid0
283
- func getMockDeliverClient () * mock. DeliverClient {
295
+ func getMockDeliverClient () * cmock. PeerDeliverClient {
284
296
return getMockDeliverClientResponseWithTxID ("txid0" )
285
297
}
286
298
287
- func getMockDeliverClientResponseWithTxID (txID string ) * mock. DeliverClient {
288
- mockDC := & mock. DeliverClient {}
289
- mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (api .Deliver , error ) {
299
+ func getMockDeliverClientResponseWithTxID (txID string ) * cmock. PeerDeliverClient {
300
+ mockDC := & cmock. PeerDeliverClient {}
301
+ mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (ccapi .Deliver , error ) {
290
302
return getMockDeliverConnectionResponseWithTxID (txID ), nil
291
303
}
292
304
// mockDC.DeliverReturns(nil, fmt.Errorf("not implemented!!"))
@@ -305,9 +317,9 @@ func getMockDeliverConnectionResponseWithTxID(txID string) *mock.Deliver {
305
317
return mockDF
306
318
}
307
319
308
- func getMockDeliverClientRespondsWithFilteredBlocks (fb []* pb.FilteredBlock ) * mock. DeliverClient {
309
- mockDC := & mock. DeliverClient {}
310
- mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (api .Deliver , error ) {
320
+ func getMockDeliverClientRespondsWithFilteredBlocks (fb []* pb.FilteredBlock ) * cmock. PeerDeliverClient {
321
+ mockDC := & cmock. PeerDeliverClient {}
322
+ mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (ccapi .Deliver , error ) {
311
323
mockDF := & mock.Deliver {}
312
324
for i , f := range fb {
313
325
resp := & pb.DeliverResponse {
@@ -322,9 +334,9 @@ func getMockDeliverClientRespondsWithFilteredBlocks(fb []*pb.FilteredBlock) *moc
322
334
return mockDC
323
335
}
324
336
325
- func getMockDeliverClientRegisterAfterDelay (delayChan chan struct {}) * mock. DeliverClient {
326
- mockDC := & mock. DeliverClient {}
327
- mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (api .Deliver , error ) {
337
+ func getMockDeliverClientRegisterAfterDelay (delayChan chan struct {}) * cmock. PeerDeliverClient {
338
+ mockDC := & cmock. PeerDeliverClient {}
339
+ mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (ccapi .Deliver , error ) {
328
340
mockDF := & mock.Deliver {}
329
341
mockDF .SendStub = func (* cb.Envelope ) error {
330
342
<- delayChan
@@ -335,9 +347,9 @@ func getMockDeliverClientRegisterAfterDelay(delayChan chan struct{}) *mock.Deliv
335
347
return mockDC
336
348
}
337
349
338
- func getMockDeliverClientRespondAfterDelay (delayChan chan struct {}) * mock. DeliverClient {
339
- mockDC := & mock. DeliverClient {}
340
- mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (api .Deliver , error ) {
350
+ func getMockDeliverClientRespondAfterDelay (delayChan chan struct {}) * cmock. PeerDeliverClient {
351
+ mockDC := & cmock. PeerDeliverClient {}
352
+ mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (ccapi .Deliver , error ) {
341
353
mockDF := & mock.Deliver {}
342
354
mockDF .RecvStub = func () (* pb.DeliverResponse , error ) {
343
355
<- delayChan
@@ -353,9 +365,9 @@ func getMockDeliverClientRespondAfterDelay(delayChan chan struct{}) *mock.Delive
353
365
return mockDC
354
366
}
355
367
356
- func getMockDeliverClientWithErr (errMsg string ) * mock. DeliverClient {
357
- mockDC := & mock. DeliverClient {}
358
- mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (api .Deliver , error ) {
368
+ func getMockDeliverClientWithErr (errMsg string ) * cmock. PeerDeliverClient {
369
+ mockDC := & cmock. PeerDeliverClient {}
370
+ mockDC .DeliverFilteredStub = func (ctx context.Context , opts ... grpc.CallOption ) (ccapi .Deliver , error ) {
359
371
return nil , fmt .Errorf (errMsg )
360
372
}
361
373
return mockDC
0 commit comments