@@ -68,7 +68,7 @@ contract BicPackTest is Test {
6868 erc20.mint (owner, 1000 * 10 ** 18 );
6969 erc721.mint (owner, 1 );
7070 erc721.mint (owner, 2 );
71- erc1155.mint (owner, 1 , 100 );
71+ erc1155.mint (owner, 1 , 150 );
7272 erc1155.mint (owner, 2 , 50 );
7373
7474 // Approve pack contract to spend tokens
@@ -157,6 +157,22 @@ contract BicPackTest is Test {
157157 assertEq (packId, 0 );
158158 assertEq (packTotalSupply, 4 ); // 16 total reward units / 4 per open = 4 packs
159159 assertEq (pack.balanceOf (recipient, packId), 4 );
160+
161+ ITokenBundle.Token[] memory newContents = new ITokenBundle.Token [](1 );
162+ newContents[0 ] = ITokenBundle.Token ({
163+ assetContract: address (erc1155),
164+ tokenType: ITokenBundle.TokenType.ERC1155 ,
165+ tokenId: 1 ,
166+ totalAmount: 80
167+ });
168+ uint256 [] memory newNumOfRewardUnits = new uint256 [](1 );
169+ newNumOfRewardUnits[0 ] = 4 ;
170+ vm.prank (owner);
171+ (uint256 newPackTotalSupply , uint256 newSupplyAdded ) = pack.addPackContents (packId, newContents,newNumOfRewardUnits, recipient);
172+ assertEq (pack.canUpdatePack (packId), true ); // Can still update pack after adding contents
173+ assertEq (newPackTotalSupply, 5 );
174+ assertEq (pack.balanceOf (recipient, packId), 5 );
175+
160176 }
161177
162178 function testCreatePackRevertsWhenNotOwner () public {
0 commit comments