@@ -46,6 +46,10 @@ func TestInit(t *testing.T) {
4646func TestInvoke (t * testing.T ) {
4747 e := new (EndorserOneValidSignature )
4848 stub := shim .NewMockStub ("endorseronevalidsignature" , e )
49+ successResponse := & pb.Response {Status : 200 , Payload : []byte ("payload" )}
50+ failResponse := & pb.Response {Status : 500 , Message : "error" }
51+ successRes , _ := putils .GetBytesResponse (successResponse )
52+ failRes , _ := putils .GetBytesResponse (failResponse )
4953
5054 // Initialize ESCC supplying the identity of the signer
5155 args := [][]byte {[]byte ("DEFAULT" ), []byte ("PEER" )}
@@ -72,27 +76,47 @@ func TestInvoke(t *testing.T) {
7276 t .Fatalf ("escc invoke should have failed with invalid number of args: %v" , args )
7377 }
7478
79+ // Failed path: Not enough parameters
80+ args = [][]byte {[]byte ("test" ), []byte ("test" ), []byte ("test" ), []byte ("test" )}
81+ if res := stub .MockInvoke ("1" , args ); res .Status == shim .OK {
82+ t .Fatalf ("escc invoke should have failed with invalid number of args: %v" , args )
83+ }
84+
7585 // Failed path: header is null
76- args = [][]byte {[]byte ("test" ), nil , []byte ("test" ), []byte ("test" )}
86+ args = [][]byte {[]byte ("test" ), nil , []byte ("test" ), successRes , []byte ("test" )}
7787 if res := stub .MockInvoke ("1" , args ); res .Status == shim .OK {
7888 fmt .Println ("Invoke" , args , "failed" , string (res .Message ))
7989 t .Fatalf ("escc invoke should have failed with a null header. args: %v" , args )
8090 }
8191
8292 // Failed path: payload is null
83- args = [][]byte {[]byte ("test" ), []byte ("test" ), nil , []byte ("test" )}
93+ args = [][]byte {[]byte ("test" ), []byte ("test" ), nil , successRes , []byte ("test" )}
8494 if res := stub .MockInvoke ("1" , args ); res .Status == shim .OK {
8595 fmt .Println ("Invoke" , args , "failed" , string (res .Message ))
8696 t .Fatalf ("escc invoke should have failed with a null payload. args: %v" , args )
8797 }
8898
99+ // Failed path: response is null
100+ args = [][]byte {[]byte ("test" ), []byte ("test" ), []byte ("test" ), nil , []byte ("test" )}
101+ if res := stub .MockInvoke ("1" , args ); res .Status == shim .OK {
102+ fmt .Println ("Invoke" , args , "failed" , string (res .Message ))
103+ t .Fatalf ("escc invoke should have failed with a null response. args: %v" , args )
104+ }
105+
89106 // Failed path: action struct is null
90- args = [][]byte {[]byte ("test" ), []byte ("test" ), []byte ("test" ), nil }
107+ args = [][]byte {[]byte ("test" ), []byte ("test" ), []byte ("test" ), successRes , nil }
91108 if res := stub .MockInvoke ("1" , args ); res .Status == shim .OK {
92109 fmt .Println ("Invoke" , args , "failed" , string (res .Message ))
93110 t .Fatalf ("escc invoke should have failed with a null action struct. args: %v" , args )
94111 }
95112
113+ // Failed path: status code >=500
114+ args = [][]byte {[]byte ("test" ), []byte ("test" ), []byte ("test" ), failRes , []byte ("test" )}
115+ if res := stub .MockInvoke ("1" , args ); res .Status == shim .OK {
116+ fmt .Println ("Invoke" , args , "failed" , string (res .Message ))
117+ t .Fatalf ("escc invoke should have failed with a null response. args: %v" , args )
118+ }
119+
96120 // Successful path - create a proposal
97121 cs := & pb.ChaincodeSpec {
98122 ChaincodeID : & pb.ChaincodeID {Name : "foo" },
@@ -127,15 +151,15 @@ func TestInvoke(t *testing.T) {
127151 // success test 1: invocation with mandatory args only
128152 simRes := []byte ("simulation_result" )
129153
130- args = [][]byte {[]byte ("" ), proposal .Header , proposal .Payload , simRes }
154+ args = [][]byte {[]byte ("" ), proposal .Header , proposal .Payload , successRes , simRes }
131155 res := stub .MockInvoke ("1" , args )
132156 if res .Status != shim .OK {
133157 t .Fail ()
134158 t .Fatalf ("escc invoke failed with: %s" , res .Message )
135159 return
136160 }
137161
138- err = validateProposalResponse (res .Payload , proposal , nil , simRes , nil )
162+ err = validateProposalResponse (res .Payload , proposal , nil , successResponse , simRes , nil )
139163 if err != nil {
140164 t .Fail ()
141165 t .Fatalf ("%s" , err )
@@ -145,15 +169,15 @@ func TestInvoke(t *testing.T) {
145169 // success test 2: invocation with mandatory args + events
146170 events := []byte ("events" )
147171
148- args = [][]byte {[]byte ("" ), proposal .Header , proposal .Payload , simRes , events }
172+ args = [][]byte {[]byte ("" ), proposal .Header , proposal .Payload , successRes , simRes , events }
149173 res = stub .MockInvoke ("1" , args )
150174 if res .Status != shim .OK {
151175 t .Fail ()
152176 t .Fatalf ("escc invoke failed with: %s" , res .Message )
153177 return
154178 }
155179
156- err = validateProposalResponse (res .Payload , proposal , nil , simRes , events )
180+ err = validateProposalResponse (res .Payload , proposal , nil , successResponse , simRes , events )
157181 if err != nil {
158182 t .Fail ()
159183 t .Fatalf ("%s" , err )
@@ -163,23 +187,23 @@ func TestInvoke(t *testing.T) {
163187 // success test 3: invocation with mandatory args + events and visibility
164188 visibility := []byte ("visibility" )
165189
166- args = [][]byte {[]byte ("" ), proposal .Header , proposal .Payload , simRes , events , visibility }
190+ args = [][]byte {[]byte ("" ), proposal .Header , proposal .Payload , successRes , simRes , events , visibility }
167191 res = stub .MockInvoke ("1" , args )
168192 if res .Status != shim .OK {
169193 t .Fail ()
170194 t .Fatalf ("escc invoke failed with: %s" , res .Message )
171195 return
172196 }
173197
174- err = validateProposalResponse (res .Payload , proposal , visibility , simRes , events )
198+ err = validateProposalResponse (res .Payload , proposal , visibility , successResponse , simRes , events )
175199 if err != nil {
176200 t .Fail ()
177201 t .Fatalf ("%s" , err )
178202 return
179203 }
180204}
181205
182- func validateProposalResponse (prBytes []byte , proposal * pb.Proposal , visibility []byte , simRes []byte , events []byte ) error {
206+ func validateProposalResponse (prBytes []byte , proposal * pb.Proposal , visibility []byte , response * pb. Response , simRes []byte , events []byte ) error {
183207 if visibility == nil {
184208 // TODO: set visibility to the default visibility mode once modes are defined
185209 }
@@ -224,6 +248,17 @@ func validateProposalResponse(prBytes []byte, proposal *pb.Proposal, visibility
224248 return fmt .Errorf ("could not unmarshal the chaincode action structure: err %s" , err )
225249 }
226250
251+ // validate that the response match
252+ if cact .Response .Status != response .Status {
253+ return fmt .Errorf ("response status do not match" )
254+ }
255+ if cact .Response .Message != response .Message {
256+ return fmt .Errorf ("response message do not match" )
257+ }
258+ if bytes .Compare (cact .Response .Payload , response .Payload ) != 0 {
259+ return fmt .Errorf ("response payload do not match" )
260+ }
261+
227262 // validate that the results match
228263 if bytes .Compare (cact .Results , simRes ) != 0 {
229264 return fmt .Errorf ("results do not match" )
0 commit comments