How to mock an enum singleton class using Mockito/Powermock?

by

If you want to stub out what INSTANCE returns, you can do it but it’s kind of nasty (using reflection & bytecode manipulation). I created & tested a simple project with three classes using the PowerMock 1.4.12 / Mockito 1.9.0. All classes were in the same package.

SingletonObject.java

public enum SingletonObject {
    INSTANCE;
    private int num;

    protected void setNum(int num) {
        this.num = num;
    }

    public int getNum() {
        return num;
    }
}

SingletonConsumer.java

public class SingletonConsumer {
    public String consumeSingletonObject() {
        return String.valueOf(SingletonObject.INSTANCE.getNum());
    }
}

SingletonConsumerTest.java

import static org.junit.Assert.*;
import static org.powermock.api.mockito.PowerMockito.*;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import org.powermock.reflect.Whitebox;

@RunWith(PowerMockRunner.class)
@PrepareForTest({SingletonObject.class})
public class SingletonConsumerTest {
    @Test
    public void testConsumeSingletonObject() throws Exception {
        SingletonObject mockInstance = mock(SingletonObject.class);
        Whitebox.setInternalState(SingletonObject.class, "INSTANCE", mockInstance);

        when(mockInstance.getNum()).thenReturn(42);

        assertEquals("42", new SingletonConsumer().consumeSingletonObject());
    }
}

The call to the Whitebox.setInternalState replaces INSTANCE with a mock object that you can manipulate within your test.

More Related Contents:

  1. Implementing Singleton with an Enum (in Java)
  2. PowerMockito mock single static method and return object
  3. How to mock private method for testing using PowerMock?
  4. Mock a constructor with parameter
  5. Using PowerMockito.whenNew() is not getting mocked and original method is called
  6. How thread-safe is enum in java?
  7. Mocking Logger and LoggerFactory with PowerMock and Mockito
  8. PowerMock, mock a static method, THEN call real methods on all other statics
  9. Powermock – java.lang.IllegalStateException: Failed to transform class
  10. How do I use Comparator to define a custom sort order?
  11. Java Enum definition
  12. What’s the advantage of a Java enum versus a class with public static final fields?
  13. Using Mockito with multiple calls to the same method with the same arguments
  14. How to tell a Mockito mock object to return something different the next time it is called?
  15. Java enum and additional class files
  16. A for-loop to iterate over an enum in Java
  17. Ways to save enums in database
  18. Mockito: Mock private field initialization
  19. Why shouldn’t Java enum literals be able to have generic type parameters?
  20. Conveniently map between enum and int / String
  21. How to supply Enum value to an annotation from a Constant in Java
  22. Mockito test a void method throws an exception
  23. Why are my mocked methods not calld when executing a unit test?
  24. Switch over type in java
  25. What is the reason behind Enum.hashCode()?
  26. Is it possible to write a generic enum converter for JPA?
  27. How to create a singleton class
  28. Can not compile enums in Eclipse
  29. How to enable enum inheritance
  30. Integration tests for AspectJ

Leave a Comment