add test cases

add docs to PutBlock
celestiaorg · Mar 10, 2021 · 5b9ebda · 5b9ebda
1 parent 06f639e
commit 5b9ebda
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/types/block.go b/types/block.go
@@ -264,7 +264,8 @@ func mustPush(rowTree *nmt.NamespacedMerkleTree, id namespace.ID, data []byte) {
 	}
 }
 
-// PutBlock add
+// PutBlock posts and pins erasured block data to IPFS using the provided
+// ipld.NodeAdder. Note: the erasured data is currently recomputed
 func (b *Block) PutBlock(ctx context.Context, api format.NodeAdder) error {
 	if api == nil {
 		return errors.New("no ipfs node adder provided")

diff --git a/types/block_test.go b/types/block_test.go
@@ -1331,7 +1331,10 @@ func TestPutBlock(t *testing.T) {
 		expectErr bool
 		errString string
 	}{
-		{"basic", generateRandomData(16), false, ""},
+		{"no leaves", generateRandomData(0), false, ""},
+		{"single leaf", generateRandomData(1), false, ""},
+		{"16 leaves", generateRandomData(16), false, ""},
+		{"max square size", generateRandomData(MaxSquareSize), false, ""},
 	}
 	ctx := context.Background()
 	for _, tc := range testCases {