Mocking Java enum to add a value to test fail case

by

Here is a complete example.

The code is almost like your original (just simplified better test validation):

public enum MyEnum {A, B}

public class Bar {

    public int foo(MyEnum value) {
        switch (value) {
            case A: return 1;
            case B: return 2;
        }
        throw new IllegalArgumentException("Do not know how to handle " + value);
    }
}

And here is the unit test with full code coverage, the test works with Powermock (1.4.10), Mockito (1.8.5) and JUnit (4.8.2):

@RunWith(PowerMockRunner.class)
public class BarTest {

    private Bar bar;

    @Before
    public void createBar() {
        bar = new Bar();
    }

    @Test(expected = IllegalArgumentException.class)
    @PrepareForTest(MyEnum.class)
    public void unknownValueShouldThrowException() throws Exception {
        MyEnum C = mock(MyEnum.class);
        when(C.ordinal()).thenReturn(2);

        PowerMockito.mockStatic(MyEnum.class);
        PowerMockito.when(MyEnum.values()).thenReturn(new MyEnum[]{MyEnum.A, MyEnum.B, C});

        bar.foo(C);
    }

    @Test
    public void AShouldReturn1() {
        assertEquals(1, bar.foo(MyEnum.A));
    }

    @Test
    public void BShouldReturn2() {
        assertEquals(2, bar.foo(MyEnum.B));
    }
}

Result:

Tests run: 3, Failures: 0, Errors: 0, Skipped: 0, Time elapsed: 0.628 sec

More Related Contents:

  1. Mocking static methods with Mockito
  2. How to mock void methods with Mockito
  3. Difference between @Mock and @InjectMocks
  4. Using Mockito to test abstract classes
  5. What’s the best mock framework for Java? [closed]
  6. Final method mocking
  7. How to tell a Mockito mock object to return something different the next time it is called?
  8. How to mock new Date() in java using Mockito
  9. Mockito – @Spy vs @Mock
  10. mock or stub for chained call
  11. What are mock objects in Java?
  12. Why are my mocked methods not calld when executing a unit test?
  13. Forming Mockito “grammars”
  14. Mocking behaviour resets after each test with PowerMock
  15. Why are my mocked methods not called when executing a unit test?
  16. How to assign class level data in a TestNG unit test class
  17. How do I use Comparator to define a custom sort order?
  18. Java Enum definition
  19. Using Mockito with multiple calls to the same method with the same arguments
  20. Java enum and additional class files
  21. A for-loop to iterate over an enum in Java
  22. Ways to save enums in database
  23. Why shouldn’t Java enum literals be able to have generic type parameters?
  24. Conveniently map between enum and int / String
  25. How to convert enum value to int?
  26. Mockito verify order / sequence of method calls
  27. Enumerations: Why? When?
  28. Jersey – How to mock service
  29. Switch over type in java
  30. What is the reason behind Enum.hashCode()?

Leave a Comment