JUnit 5 in Java 8 using Fizz Buzz example

src/main/java/com/techprimers/testing/FizzBuzz.java

@@ -0,0 +1,14 @@
+package com.techprimers.testing;
+
+public class FizzBuzz {
+
+    public String play(int number) {
+
+        if (number == 0) throw new IllegalArgumentException("Number must not be 0");
+        if (number % 3 == 0) return "Fizz";
+        if (number % 5 == 0) return "Buzz";
+
+
+        return String.valueOf(number);
+    }
+}

src/test/java/com/techprimers/testing/FizzBuzzTest.java

@@ -0,0 +1,48 @@
+package com.techprimers.testing;
+
+import org.junit.jupiter.api.*;
+
+class FizzBuzzTest {
+
+    public FizzBuzz fB;
+
+    @BeforeEach
+    public void setUp() {
+        fB = new FizzBuzz();
+    }
+
+    @DisplayName("Play FizzBuzz with number = 1")
+    @Test
+    public void testNumber() {
+        String fizzBuzz = fB.play(1);
+        Assertions.assertEquals(fizzBuzz, "1");
+    }
+
+    @DisplayName("Play FizzBuzz with number = 3")
+    @Test
+    public void testFizz() {
+        String fizzBuzz = fB.play(3);
+        Assertions.assertEquals(fizzBuzz, "Fizz");
+    }
+
+    @DisplayName("Play FizzBuzz with number = 5")
+    @Test
+    public void testBuzz() {
+        String fizzBuzz = fB.play(5);
+        Assertions.assertEquals(fizzBuzz, "Buzz");
+    }
+
+    @DisplayName("Don't Play FizzBuzz with number = 0")
+    @Test
+    public void testZero() {
+
+        Assertions.assertThrows(IllegalArgumentException.class,
+                () -> fB.play(0));
+    }
+
+    @AfterEach
+    public void tearDown() {
+        fB = null;
+    }
+
+}