Skip to content

Commit

Permalink
Fix linter errors
Browse files Browse the repository at this point in the history
  • Loading branch information
tzdybal committed Jan 28, 2021
1 parent 9eb31c8 commit 6171e35
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 12 deletions.
2 changes: 1 addition & 1 deletion store/rootmulti/proof.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,6 @@ func DefaultProofRuntime() (prt *merkle.ProofRuntime) {

func SMTProofRuntime() (prt *merkle.ProofRuntime) {
prt = merkle.NewProofRuntime()
prt.RegisterOpDecoder(smt.SMTProofType, smt.ProofDecoder)
prt.RegisterOpDecoder(smt.ProofType, smt.ProofDecoder)
return prt
}
18 changes: 9 additions & 9 deletions store/smt/proof.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,28 +20,28 @@ const (
)

const (
SMTProofType = "smt"
ProofType = "smt"
)

type SMTProof struct {
type ProofOp struct {
Root []byte
Key []byte
Hasher HasherType
Proof smt.SparseMerkleProof
}

var _ merkle.ProofOperator = &SMTProof{}
var _ merkle.ProofOperator = &ProofOp{}

func NewSMTProof(root, key []byte, hasher HasherType, proof smt.SparseMerkleProof) *SMTProof {
return &SMTProof{
func NewProofOp(root, key []byte, hasher HasherType, proof smt.SparseMerkleProof) *ProofOp {
return &ProofOp{
Root: root,
Key: key,
Hasher: hasher,
Proof: proof,
}
}

func (p *SMTProof) Run(args [][]byte) ([][]byte, error) {
func (p *ProofOp) Run(args [][]byte) ([][]byte, error) {
switch len(args) {
case 0: // non-membership proof
if !smt.VerifyProof(p.Proof, p.Root, p.Key, []byte{}, getHasher(p.Hasher)) {
Expand All @@ -57,11 +57,11 @@ func (p *SMTProof) Run(args [][]byte) ([][]byte, error) {
return [][]byte{p.Root}, nil
}

func (p *SMTProof) GetKey() []byte {
func (p *ProofOp) GetKey() []byte {
return p.Key
}

func (p *SMTProof) ProofOp() tmmerkle.ProofOp {
func (p *ProofOp) ProofOp() tmmerkle.ProofOp {
var data bytes.Buffer
enc := gob.NewEncoder(&data)
enc.Encode(p)
Expand All @@ -74,7 +74,7 @@ func (p *SMTProof) ProofOp() tmmerkle.ProofOp {

func ProofDecoder(pop tmmerkle.ProofOp) (merkle.ProofOperator, error) {
dec := gob.NewDecoder(bytes.NewBuffer(pop.Data))
var proof SMTProof
var proof ProofOp
err := dec.Decode(&proof)
if err != nil {
return nil, err
Expand Down
4 changes: 2 additions & 2 deletions store/smt/proof_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ func TestProofOpInterface(t *testing.T) {
proof, err := tree.Prove(key)
require.True(t, smt.VerifyProof(proof, root, key, value, hasher))

storeProofOp := smtstore.NewSMTProof(root, key, smtstore.SHA256, proof)
storeProofOp := smtstore.NewProofOp(root, key, smtstore.SHA256, proof)
require.NotNil(t, storeProofOp)
// inclusion proof
r, err := storeProofOp.Run([][]byte{value})
Expand All @@ -50,7 +50,7 @@ func TestProofOpInterface(t *testing.T) {
// encode
tmProofOp := storeProofOp.ProofOp()
assert.NotNil(t, tmProofOp)
assert.Equal(t, smtstore.SMTProofType, tmProofOp.Type)
assert.Equal(t, smtstore.ProofType, tmProofOp.Type)
assert.Equal(t, key, tmProofOp.Key, key)
assert.NotEmpty(t, tmProofOp.Data)

Expand Down

0 comments on commit 6171e35

Please sign in to comment.