1212import net .minecraft .world .entity .AgeableMob ;
1313import net .minecraft .world .entity .EntitySpawnReason ;
1414import net .minecraft .world .entity .EntityType ;
15+ import net .minecraft .world .entity .EntityTypes ;
1516import net .minecraft .world .entity .animal .sheep .Sheep ;
1617import net .minecraft .world .entity .player .Player ;
1718import net .minecraft .world .item .DyeColor ;
2425import net .minecraft .world .level .block .entity .DispenserBlockEntity ;
2526import net .minecraft .world .level .block .state .BlockState ;
2627import net .minecraft .world .phys .BlockHitResult ;
27- import org . cyclops . energeticsheep . entity . EntityEnergeticSheepConfigCommon ;
28+ import net . minecraft . world . phys . Vec3 ;
2829import org .cyclops .cyclopscore .gametest .GameTest ;
2930import org .cyclops .energeticsheep .Reference ;
3031import org .cyclops .energeticsheep .RegistryEntries ;
3132import org .cyclops .energeticsheep .entity .EntityAIEatGrassFast ;
3233import org .cyclops .energeticsheep .entity .EntityEnergeticSheepCommon ;
34+ import org .cyclops .energeticsheep .entity .EntityEnergeticSheepConfigCommon ;
3335
3436/**
3537 * @author rubensworks
@@ -72,7 +74,7 @@ public void testShearEnergetic(GameTestHelper helper) {
7274
7375 // Give energetic shears to player
7476 Player player = helper .makeMockPlayer (GameType .SURVIVAL );
75- ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS );
77+ ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS . getHolder () );
7678 player .setItemInHand (InteractionHand .MAIN_HAND , itemStack );
7779
7880 helper .succeedWhen (() -> {
@@ -133,14 +135,14 @@ public void testShearEnergeticLeavesWithPower(GameTestHelper helper) {
133135
134136 // Give energetic shears with power to player
135137 Player player = helper .makeMockPlayer (GameType .SURVIVAL );
136- ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS );
138+ ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS . getHolder () );
137139 player .setItemInHand (InteractionHand .MAIN_HAND , itemStack );
138140 RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().setEnergyStored (itemStack , RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().getMaxEnergyStored (itemStack ), player , player .getUsedItemHand ());
139141
140142 helper .succeedWhen (() -> {
141143 // Break leaves with energetic shears
142144 BlockState blockState = helper .getBlockState (POS );
143- InteractionResult interactionResult = itemStack .useOn (new UseOnContext (player , InteractionHand .MAIN_HAND , new BlockHitResult (POS . getCenter ( ), Direction .NORTH , helper .absolutePos (POS ), false )));
145+ InteractionResult interactionResult = itemStack .useOn (new UseOnContext (player , InteractionHand .MAIN_HAND , new BlockHitResult (Vec3 . atCenterOf ( POS ), Direction .NORTH , helper .absolutePos (POS ), false )));
144146 helper .assertFalse (interactionResult == InteractionResult .FAIL , Component .literal ("Interaction must succeed" ));
145147 helper .assertTrue (itemStack .getItem ().mineBlock (itemStack , helper .getLevel (), blockState , helper .absolutePos (POS ), player ), Component .literal ("Item can not mine block" ));
146148 helper .assertTrue (player .hasCorrectToolForDrops (blockState ), Component .literal ("Player must have correct tool" ));
@@ -158,12 +160,12 @@ public void testShearEnergeticLeavesNoPower(GameTestHelper helper) {
158160
159161 // Give energetic shears without power to player
160162 Player player = helper .makeMockPlayer (GameType .SURVIVAL );
161- ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS );
163+ ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS . getHolder () );
162164 player .setItemInHand (InteractionHand .MAIN_HAND , itemStack );
163165
164166 helper .succeedWhen (() -> {
165167 // Attempt to break leaves with energetic shears that has no power
166- InteractionResult interactionResult = itemStack .useOn (new UseOnContext (player , InteractionHand .MAIN_HAND , new BlockHitResult (POS . getCenter ( ), Direction .NORTH , helper .absolutePos (POS ), false )));
168+ InteractionResult interactionResult = itemStack .useOn (new UseOnContext (player , InteractionHand .MAIN_HAND , new BlockHitResult (Vec3 . atCenterOf ( POS ), Direction .NORTH , helper .absolutePos (POS ), false )));
167169 helper .assertFalse (interactionResult == InteractionResult .FAIL , Component .literal ("Interaction must fail" ));
168170
169171 helper .assertItemEntityNotPresent (Items .ACACIA_LEAVES );
@@ -178,14 +180,14 @@ public void testShearEnergeticWithPowerRegularSheep(GameTestHelper helper) {
178180
179181 // Give energetic shears to player
180182 Player player = helper .makeMockPlayer (GameType .SURVIVAL );
181- ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS );
183+ ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS . getHolder () );
182184 player .setItemInHand (InteractionHand .MAIN_HAND , itemStack );
183185 RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().setEnergyStored (itemStack , RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().getMaxEnergyStored (itemStack ), player , player .getUsedItemHand ());
184186
185187 helper .succeedWhen (() -> {
186188 // Right click with energetic shears
187189 InteractionResult result = player .interactOn (entity , InteractionHand .MAIN_HAND , entity .position ());
188- helper .assertItemEntityPresent (Items .WHITE_WOOL );
190+ helper .assertItemEntityPresent (Items .WOOL . white () );
189191 helper .assertTrue (RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().getEnergyStored (player .getMainHandItem ()) < RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().getMaxEnergyStored (player .getMainHandItem ()), Component .literal ("No energy was consumed from shears" ));
190192 helper .assertTrue (result .equals (InteractionResult .SUCCESS ), Component .literal ("Interaction failed" ));
191193 });
@@ -198,13 +200,13 @@ public void testShearEnergeticNoPowerRegularSheep(GameTestHelper helper) {
198200
199201 // Give energetic shears to player
200202 Player player = helper .makeMockPlayer (GameType .SURVIVAL );
201- ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS );
203+ ItemStack itemStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS . getHolder () );
202204 player .setItemInHand (InteractionHand .MAIN_HAND , itemStack );
203205
204206 helper .succeedWhen (() -> {
205207 // Right click with energetic shears
206208 InteractionResult result = player .interactOn (entity , InteractionHand .MAIN_HAND , entity .position ());
207- helper .assertItemEntityNotPresent (Items .WHITE_WOOL );
209+ helper .assertItemEntityNotPresent (Items .WOOL . white () );
208210 helper .assertTrue (RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().getEnergyStored (player .getMainHandItem ()) == 0 , Component .literal ("Shears do not have enough energy" ));
209211 helper .assertTrue (result .equals (InteractionResult .PASS ), Component .literal ("Interaction did not pass" ));
210212 });
@@ -393,7 +395,7 @@ public void testDispenseEnergeticShears(GameTestHelper helper) {
393395
394396 // Put energetic shears (empty, no charge) into the dispenser
395397 DispenserBlockEntity dispenserBlockEntity = (DispenserBlockEntity ) helper .getLevel ().getBlockEntity (helper .absolutePos (dispenserPos ));
396- ItemStack shearsStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS );
398+ ItemStack shearsStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS . getHolder () );
397399 dispenserBlockEntity .setItem (0 , shearsStack );
398400
399401 helper .succeedWhen (() -> {
@@ -418,14 +420,14 @@ public void testDispenseEnergeticShearsOnRegularSheep(GameTestHelper helper) {
418420 helper .setBlock (dispenserPos , Blocks .DISPENSER .defaultBlockState ().setValue (DispenserBlock .FACING , Direction .EAST ));
419421
420422 // Spawn regular white sheep directly in front of the dispenser (at POS)
421- Sheep entity = helper .spawn (EntityType .SHEEP , POS );
423+ Sheep entity = helper .spawn (EntityTypes .SHEEP , POS );
422424 entity .finalizeSpawn (helper .getLevel (), helper .getLevel ().getCurrentDifficultyAt (helper .absolutePos (POS )), EntitySpawnReason .NATURAL , null );
423425 entity .setColor (DyeColor .WHITE );
424426
425427 // Put energetic shears (fully charged) into the dispenser
426428 DispenserBlockEntity dispenserBlockEntity = (DispenserBlockEntity ) helper .getLevel ().getBlockEntity (helper .absolutePos (dispenserPos ));
427429 Player tempPlayer = helper .makeMockPlayer (GameType .SURVIVAL );
428- ItemStack shearsStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS );
430+ ItemStack shearsStack = new ItemStack (RegistryEntries .ITEM_ENERGETIC_SHEARS . getHolder () );
429431 tempPlayer .setItemInHand (InteractionHand .MAIN_HAND , shearsStack );
430432 RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().setEnergyStored (shearsStack , RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().getMaxEnergyStored (shearsStack ), tempPlayer , InteractionHand .MAIN_HAND );
431433 dispenserBlockEntity .setItem (0 , tempPlayer .getMainHandItem ());
@@ -439,7 +441,7 @@ public void testDispenseEnergeticShearsOnRegularSheep(GameTestHelper helper) {
439441 dispenserBlockEntity .setItem (0 , result );
440442
441443 // Regular wool should have been dropped
442- helper .assertItemEntityPresent (Items .WHITE_WOOL );
444+ helper .assertItemEntityPresent (Items .WOOL . white () );
443445 // Some energy should have been consumed from the shears
444446 helper .assertTrue (RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().getEnergyStored (result ) < RegistryEntries .ITEM_ENERGETIC_SHEARS .value ().getMaxEnergyStored (result ), Component .literal ("Energy should have been consumed from shears after shearing regular sheep" ));
445447 });
@@ -453,7 +455,7 @@ protected static EntityEnergeticSheepCommon SPAWN(GameTestHelper helper) {
453455 }
454456
455457 protected static Sheep SPAWN_REGULAR (GameTestHelper helper ) {
456- Sheep entity = helper .spawn (EntityType .SHEEP , POS .above ());
458+ Sheep entity = helper .spawn (EntityTypes .SHEEP , POS .above ());
457459 entity .finalizeSpawn (helper .getLevel (), helper .getLevel ().getCurrentDifficultyAt (POS ), EntitySpawnReason .NATURAL , null );
458460 return entity ;
459461 }
0 commit comments