SPIGOT-1464: Fixed setting Mushroom faces

1 files changed, 28 insertions, 0 deletions

diff --git a/src/test/java/org/bukkit/materials/MaterialDataTest.java b/src/test/java/org/bukkit/materials/MaterialDataTest.java
index 41dff993..3eaede3c 100644
--- a/src/test/java/org/bukkit/materials/MaterialDataTest.java
+++ b/src/test/java/org/bukkit/materials/MaterialDataTest.java
@@ -8,10 +8,12 @@ import org.bukkit.TreeSpecies;
 import org.bukkit.block.BlockFace;
 import org.bukkit.material.Door;
 import org.bukkit.material.Leaves;
+import org.bukkit.material.Mushroom;
 import org.bukkit.material.Sapling;
 import org.bukkit.material.Tree;
 import org.bukkit.material.Wood;
 import org.bukkit.material.WoodenStep;
+import org.bukkit.material.types.MushroomBlockTexture;
 import org.junit.Test;
 
 public class MaterialDataTest {
@@ -231,4 +233,30 @@ public class MaterialDataTest {
             }
         }
     }
+
+    @Test
+    public void testMushroom() {
+        Material[] mushroomTypes = new Material[] { Material.HUGE_MUSHROOM_1, Material.HUGE_MUSHROOM_2 };
+        BlockFace[] setFaces = new BlockFace[] { BlockFace.SELF, BlockFace.UP, BlockFace.NORTH,
+                BlockFace.SOUTH, BlockFace.EAST, BlockFace.WEST, BlockFace.NORTH_EAST, BlockFace.NORTH_WEST,
+                BlockFace.SOUTH_EAST, BlockFace.SOUTH_WEST };
+        MushroomBlockTexture[] textures = MushroomBlockTexture.values();
+        for (Material type : mushroomTypes) {
+            Mushroom mushroom = new Mushroom(type);
+            assertThat("Constructed with correct mushroom type", mushroom.getItemType(), equalTo(type));
+            assertThat("Constructed with default pores face", mushroom.getBlockTexture(), equalTo(MushroomBlockTexture.ALL_PORES));
+
+            for (int f = 0; f < setFaces.length; f++) {
+                mushroom = new Mushroom(type, setFaces[f]);
+                assertThat("Constructed with correct mushroom type", mushroom.getItemType(), equalTo(type));
+                assertThat("Constructed with correct texture", mushroom.getBlockTexture(), equalTo(MushroomBlockTexture.getCapByFace(setFaces[f])));
+            }
+
+            for (MushroomBlockTexture texture : textures) {
+                mushroom = new Mushroom(type, texture);
+                assertThat("Constructed with correct mushroom type", mushroom.getItemType(), equalTo(type));
+                assertThat("Constructed with correct texture", mushroom.getBlockTexture(), equalTo(texture));
+            }
+        }
+    }
 }