Skip to content

Commit c74c44a

Browse files
committed
tapdb: Update assets store tests to include new filter options
1 parent 4144de5 commit c74c44a

File tree

1 file changed

+37
-12
lines changed

1 file changed

+37
-12
lines changed

tapdb/assets_store_test.go

+37-12
Original file line numberDiff line numberDiff line change
@@ -781,16 +781,17 @@ func TestFetchAllAssets(t *testing.T) {
781781
amt: 777,
782782
scriptKey: scriptKeyWithScript,
783783
}}
784-
makeFilter := func(amt uint64, anchorHeight int32,
785-
coinSelectType tapsend.CoinSelectType) *AssetQueryFilters {
786-
784+
makeFilter := func(minAmt, maxAmt uint64, anchorHeight int32, scriptKey *asset.ScriptKey, anchorPoint *wire.OutPoint, coinSelectType tapsend.CoinSelectType) *AssetQueryFilters {
787785
constraints := tapfreighter.CommitmentConstraints{
788-
MinAmt: amt,
786+
MinAmt: minAmt,
787+
MaxAmt: maxAmt,
789788
CoinSelectType: coinSelectType,
790789
}
791790
return &AssetQueryFilters{
792791
CommitmentConstraints: constraints,
793792
MinAnchorHeight: anchorHeight,
793+
ScriptKey: scriptKey,
794+
AnchorPoint: anchorPoint,
794795
}
795796
}
796797

@@ -819,42 +820,66 @@ func TestFetchAllAssets(t *testing.T) {
819820
numAssets: 10,
820821
}, {
821822
name: "min amount",
822-
filter: makeFilter(12, 0, tapsend.ScriptTreesAllowed),
823+
filter: makeFilter(12, 0, 0, nil, nil, tapsend.ScriptTreesAllowed),
823824
numAssets: 2,
824825
}, {
825826
name: "min amount, include spent",
826-
filter: makeFilter(12, 0, tapsend.ScriptTreesAllowed),
827+
filter: makeFilter(12, 0, 0, nil, nil, tapsend.ScriptTreesAllowed),
827828
includeSpent: true,
828829
numAssets: 4,
829830
}, {
830831
name: "min amount, include leased",
831-
filter: makeFilter(12, 0, tapsend.ScriptTreesAllowed),
832+
filter: makeFilter(12, 0, 0, nil, nil, tapsend.ScriptTreesAllowed),
832833
includeLeased: true,
833834
numAssets: 5,
834835
}, {
835836
name: "min amount, include leased, include spent",
836-
filter: makeFilter(12, 0, tapsend.ScriptTreesAllowed),
837+
filter: makeFilter(12, 0, 0, nil, nil, tapsend.ScriptTreesAllowed),
837838
includeLeased: true,
838839
includeSpent: true,
839840
numAssets: 8,
840841
}, {
841842
name: "default min height, include spent",
842-
filter: makeFilter(0, 500, tapsend.ScriptTreesAllowed),
843+
filter: makeFilter(0, 0, 500, nil, nil, tapsend.ScriptTreesAllowed),
843844
includeSpent: true,
844845
numAssets: 6,
845846
}, {
846847
name: "specific height",
847-
filter: makeFilter(0, 502, tapsend.ScriptTreesAllowed),
848+
filter: makeFilter(0, 0, 502, nil, nil, tapsend.ScriptTreesAllowed),
848849
numAssets: 0,
849850
}, {
850851
name: "default min height, include spent",
851-
filter: makeFilter(0, 502, tapsend.ScriptTreesAllowed),
852+
filter: makeFilter(0, 0, 502, nil, nil, tapsend.ScriptTreesAllowed),
852853
includeSpent: true,
853854
numAssets: 1,
854855
}, {
855856
name: "script key with tapscript",
856-
filter: makeFilter(100, 0, tapsend.Bip86Only),
857+
filter: makeFilter(100, 0, 0, scriptKeyWithScript, nil, tapsend.Bip86Only),
857858
numAssets: 0,
859+
}, {
860+
name: "anchor point",
861+
filter: makeFilter(0, 0, 0, nil, &assetGen.anchorPoints[0], tapsend.ScriptTreesAllowed),
862+
numAssets: 3,
863+
}, {
864+
name: "max amount",
865+
filter: makeFilter(0, 100, 0, nil, nil, tapsend.ScriptTreesAllowed),
866+
numAssets: 4,
867+
}, {
868+
name: "max amount, include spent",
869+
filter: makeFilter(0, 100, 0, nil, nil, tapsend.ScriptTreesAllowed),
870+
includeSpent: true,
871+
numAssets: 5,
872+
}, {
873+
name: "max amount, include leased",
874+
filter: makeFilter(0, 100, 0, nil, nil, tapsend.ScriptTreesAllowed),
875+
includeLeased: true,
876+
numAssets: 6,
877+
}, {
878+
name: "max amount, include leased, include spent",
879+
filter: makeFilter(0, 100, 0, nil, nil, tapsend.ScriptTreesAllowed),
880+
includeLeased: true,
881+
includeSpent: true,
882+
numAssets: 7,
858883
}}
859884

860885
// First, we'll create a new assets store and then insert the set of

0 commit comments

Comments
 (0)