From c67976551ac4cebc636bef35ea73ff1ffca65bb2 Mon Sep 17 00:00:00 2001
From: evan-forbes <evan.samuel.forbes@gmail.com>
Date: Wed, 14 Dec 2022 18:09:46 -0600
Subject: [PATCH] chore: add unit test for bug

---
 pkg/shares/shares_test.go | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/pkg/shares/shares_test.go b/pkg/shares/shares_test.go
index 0b4c3656f3..79b6ba8145 100644
--- a/pkg/shares/shares_test.go
+++ b/pkg/shares/shares_test.go
@@ -1,6 +1,7 @@
 package shares
 
 import (
+	"bytes"
 	"context"
 	"testing"
 	"time"
@@ -10,6 +11,7 @@ import (
 	"github.com/celestiaorg/rsmt2d"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	tmrand "github.com/tendermint/tendermint/libs/rand"
 	coretypes "github.com/tendermint/tendermint/types"
 )
 
@@ -62,6 +64,32 @@ func TestMerge(t *testing.T) {
 	}
 }
 
+// TestPadFirstIndexedBlob ensures that we are adding padding to the first share
+// instead of calculating the value.
+func TestPadFirstIndexedBlob(t *testing.T) {
+	tx := tmrand.Bytes(300)
+	blob := tmrand.Bytes(300)
+	index := 100
+	indexedTx, err := coretypes.MarshalIndexWrapper(100, tx)
+	require.NoError(t, err)
+
+	bd := coretypes.Data{
+		Txs: []coretypes.Tx{indexedTx},
+		Blobs: []coretypes.Blob{
+			{NamespaceID: []byte{1, 2, 3, 4, 5, 6, 7, 8}, Data: blob, ShareVersion: appconsts.ShareVersionZero},
+		},
+		SquareSize: 64,
+	}
+
+	shares, err := Split(bd, true)
+	require.NoError(t, err)
+
+	resShare, err := shares[index].RawData()
+	require.NoError(t, err)
+
+	require.True(t, bytes.Contains(resShare, blob))
+}
+
 func TestFuzz_Merge(t *testing.T) {
 	t.Skip()
 	// run random shares through processCompactShares for a minute