clean up

jprov/server/attestation.go

@@ -99,6 +99,10 @@ func attest(w *http.ResponseWriter, r *http.Request, cmd *cobra.Command, q *queu
 	}
 
 	upload, err := addMsgAttest(address, attest.Cid, q)
+	if err != nil {
+		http.Error(*w, err.Error(), http.StatusBadRequest)
+		return
+	}
 
 	upload.Callback.Wait()
 

jprov/server/file_server_test.go

@@ -39,6 +39,7 @@ func TestWriteResponse(t *testing.T) {
 			}
 
 			err := server.WriteResponse(rec, upload, c.fid, c.cid)
+			assert.NoError(t, err)
 
 			resp := types.UploadResponse{
 				CID: c.cid,

jprov/server/proofs.go

@@ -41,7 +41,7 @@ func GetMerkleTree(ctx client.Context, filename string) (*merkletree.MerkleTree,
 		return &merkletree.MerkleTree{}, fmt.Errorf("unable to find merkle tree for: %s", filename)
 	}
 
-	return merkletree.ImportMerkleTree(rawTree, sha3.New512())	
+	return merkletree.ImportMerkleTree(rawTree, sha3.New512())
 }
 
 func GenerateMerkleProof(tree merkletree.MerkleTree, index int, item []byte) (valid bool, proof *merkletree.Proof, err error) {
@@ -57,7 +57,7 @@ func GenerateMerkleProof(tree merkletree.MerkleTree, index int, item []byte) (va
 	}
 
 	valid, err = merkletree.VerifyProofUsing(h.Sum(nil), false, proof, [][]byte{tree.Root()}, sha3.New512())
-	return 
+	return
 }
 
 func CreateMerkleForProof(clientCtx client.Context, filename string, index int, ctx *utils.Context) (string, string, error) {
@@ -223,7 +223,7 @@ func requestAttestation(clientCtx client.Context, cid string, hashList string, i
 
 	pwg.Wait()
 
-	if count < 3 {//NOTE: this value can change in chain params
+	if count < 3 { // NOTE: this value can change in chain params
 		fmt.Println("failed to get enough attestations...")
 		return fmt.Errorf("failed to get attestations")
 	}

jprov/server/proofs_test.go

@@ -1,36 +1,36 @@
 package server_test
 
 import (
-	"testing"
-	"io"
-	"fmt"
 	"crypto/sha256"
+	"fmt"
+	"io"
+	"testing"
 
 	"github.com/JackalLabs/jackal-provider/jprov/server"
-//	"github.com/JackalLabs/jackal-provider/jprov/types"
-//	"github.com/JackalLabs/jackal-provider/jprov/testutils"
+	//	"github.com/JackalLabs/jackal-provider/jprov/types"
+	//	"github.com/JackalLabs/jackal-provider/jprov/testutils"
 	"github.com/stretchr/testify/require"
 	merkletree "github.com/wealdtech/go-merkletree"
 	"github.com/wealdtech/go-merkletree/sha3"
 )
 
 func TestGenerateMerkleProof(t *testing.T) {
 	cases := map[string]struct {
-		index int
-		item []byte
+		index    int
+		item     []byte
 		expValid bool
-		expErr bool
+		expErr   bool
 	}{
 		"valid proof": {
-			index: 0,
-			item: []byte("hello"),
+			index:    0,
+			item:     []byte("hello"),
 			expValid: true,
-			expErr: false,
+			expErr:   false,
 		},
 	}
 
 	for name, c := range cases {
-		t.Run(name, func(t *testing.T){
+		t.Run(name, func(t *testing.T) {
 			data := [][]byte{[]byte("hello"), []byte("world")}
 			for i, item := range data {
 				h := sha256.New()