@@ -162,7 +162,7 @@ func (s *AttestationsTestSuite) TestVerifyMembership() {
162162 newTimestamp := uint64 (2 * time .Second .Nanoseconds ())
163163 stateAttestation := s .createStateAttestation (newHeight , newTimestamp )
164164 signers := []int {0 , 1 , 2 }
165- updateProof := s .createAttestationProof (stateAttestation , signers )
165+ updateProof := s .createAttestationProof (stateAttestation , signers , attestations . AttestationTypeState )
166166
167167 err := s .lightClientModule .VerifyClientMessage (ctx , clientID , updateProof )
168168 s .Require ().NoError (err )
@@ -177,7 +177,7 @@ func (s *AttestationsTestSuite) TestVerifyMembership() {
177177 // Commitment is stored as raw value (not hashed)
178178 hashedPath := crypto .Keccak256 (tc .attestedPath )
179179 packetAttestation := s .createPacketAttestation (newHeight , []attestations.PacketCompact {{Path : hashedPath , Commitment : tc .value }})
180- membershipProof := s .createAttestationProof (packetAttestation , signers )
180+ membershipProof := s .createAttestationProof (packetAttestation , signers , attestations . AttestationTypePacket )
181181 membershipProofBz := s .marshalProof (membershipProof )
182182
183183 proofHeight := clienttypes .NewHeight (0 , newHeight )
@@ -210,7 +210,7 @@ func (s *AttestationsTestSuite) TestVerifyMembershipMalformedProof() {
210210 s .Require ().ErrorContains (err , "failed to unmarshal proof" )
211211
212212 malformedData := []byte ("invalid-packet-attestation-data" )
213- proof := s .createAttestationProof (malformedData , []int {0 , 1 , 2 })
213+ proof := s .createAttestationProof (malformedData , []int {0 , 1 , 2 }, attestations . AttestationTypePacket )
214214 proofBz := s .marshalProof (proof )
215215
216216 err = s .lightClientModule .VerifyMembership (ctx , clientID , proofHeight , 0 , 0 , proofBz , path , value )
@@ -237,7 +237,7 @@ func (s *AttestationsTestSuite) TestVerifyMembershipVariableLengthPath() {
237237 hashedPath := crypto .Keccak256 (shortPath )
238238 packetAttestation := s .createPacketAttestation (newHeight , []attestations.PacketCompact {{Path : hashedPath , Commitment : value32 }})
239239 signers := []int {0 , 1 , 2 }
240- proof := s .createAttestationProof (packetAttestation , signers )
240+ proof := s .createAttestationProof (packetAttestation , signers , attestations . AttestationTypePacket )
241241 proofBz := s .marshalProof (proof )
242242
243243 proofHeight := clienttypes .NewHeight (0 , newHeight )
@@ -261,7 +261,7 @@ func (s *AttestationsTestSuite) TestVerifyMembershipInvalidKeyPathLength() {
261261 hashedPath := crypto .Keccak256 (bytes .Repeat ([]byte {0x01 }, 32 ))
262262 packetAttestation := s .createPacketAttestation (newHeight , []attestations.PacketCompact {{Path : hashedPath , Commitment : value32 }})
263263 signers := []int {0 , 1 , 2 }
264- proof := s .createAttestationProof (packetAttestation , signers )
264+ proof := s .createAttestationProof (packetAttestation , signers , attestations . AttestationTypePacket )
265265 proofBz := s .marshalProof (proof )
266266 proofHeight := clienttypes .NewHeight (0 , newHeight )
267267
@@ -291,7 +291,7 @@ func (s *AttestationsTestSuite) TestVerifyNonMembershipInvalidKeyPathLength() {
291291 hashedPath := crypto .Keccak256 (bytes .Repeat ([]byte {0x01 }, 32 ))
292292 packetAttestation := s .createPacketAttestation (newHeight , []attestations.PacketCompact {{Path : hashedPath , Commitment : make ([]byte , 32 )}})
293293 signers := []int {0 , 1 , 2 }
294- proof := s .createAttestationProof (packetAttestation , signers )
294+ proof := s .createAttestationProof (packetAttestation , signers , attestations . AttestationTypePacket )
295295 proofBz := s .marshalProof (proof )
296296 proofHeight := clienttypes .NewHeight (0 , newHeight )
297297
@@ -399,7 +399,7 @@ func (s *AttestationsTestSuite) TestVerifyNonMembership() {
399399 newTimestamp := uint64 (2 * time .Second .Nanoseconds ())
400400 stateAttestation := s .createStateAttestation (newHeight , newTimestamp )
401401 signers := []int {0 , 1 , 2 }
402- updateProof := s .createAttestationProof (stateAttestation , signers )
402+ updateProof := s .createAttestationProof (stateAttestation , signers , attestations . AttestationTypeState )
403403
404404 err := s .lightClientModule .VerifyClientMessage (ctx , clientID , updateProof )
405405 s .Require ().NoError (err )
@@ -417,7 +417,7 @@ func (s *AttestationsTestSuite) TestVerifyNonMembership() {
417417 }
418418 }
419419 packetAttestation := s .createPacketAttestation (newHeight , hashedPackets )
420- nonMembershipProof := s .createAttestationProof (packetAttestation , signers )
420+ nonMembershipProof := s .createAttestationProof (packetAttestation , signers , attestations . AttestationTypePacket )
421421 nonMembershipProofBz := s .marshalProof (nonMembershipProof )
422422
423423 proofHeight := clienttypes .NewHeight (0 , newHeight )
0 commit comments