Skip to content

Commit b1e029b

Browse files
committed
[FAB-10090] Increase UT coverage ledger/pvtdatapolicy
This CR adds unit test to increase the UT coverage for the package ledger/pvtdatapolicy Change-Id: I2a4a5494298fc62f14f51527830ee6fb72d9107c Signed-off-by: manish <manish.sethi@gmail.com>
1 parent 8760a8c commit b1e029b

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

core/ledger/pvtdatapolicy/btlpolicy_test.go

+25
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ SPDX-License-Identifier: Apache-2.0
77
package pvtdatapolicy
88

99
import (
10+
"math"
1011
"testing"
1112

1213
"github.com/hyperledger/fabric/core/common/privdata"
@@ -37,3 +38,27 @@ func TestBTLPolicy(t *testing.T) {
3738
_, ok := err.(privdata.NoSuchCollectionError)
3839
assert.True(t, ok)
3940
}
41+
42+
func TestExpiringBlock(t *testing.T) {
43+
mockCollectionStore := testutil.NewMockCollectionStore()
44+
mockCollectionStore.SetBTL("ns1", "coll1", 100)
45+
mockCollectionStore.SetBTL("ns1", "coll2", 200)
46+
mockCollectionStore.SetBTL("ns1", "coll3", 0)
47+
btlPolicy := ConstructBTLPolicy(mockCollectionStore)
48+
49+
expiringBlk, err := btlPolicy.GetExpiringBlock("ns1", "coll1", 50)
50+
assert.NoError(t, err)
51+
assert.Equal(t, uint64(151), expiringBlk)
52+
53+
expiringBlk, err = btlPolicy.GetExpiringBlock("ns1", "coll2", 50)
54+
assert.NoError(t, err)
55+
assert.Equal(t, uint64(251), expiringBlk)
56+
57+
expiringBlk, err = btlPolicy.GetExpiringBlock("ns1", "coll3", 50)
58+
assert.NoError(t, err)
59+
assert.Equal(t, uint64(math.MaxUint64), expiringBlk)
60+
61+
expiringBlk, err = btlPolicy.GetExpiringBlock("ns1", "coll4", 50)
62+
_, ok := err.(privdata.NoSuchCollectionError)
63+
assert.True(t, ok)
64+
}

0 commit comments

Comments
 (0)