TestSubjectPowermockAnnotationTest.java

package samples.junit4.annotationbased;

import org.easymock.TestSubject;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.easymock.annotation.Mock;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import org.powermock.reflect.Whitebox;
import samples.injectmocks.InjectDemo;
import samples.injectmocks.InjectDependencyHolder;
import samples.injectmocks.InjectDependencyHolderQualifier;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.powermock.api.easymock.PowerMock.expectPrivate;
import static org.powermock.api.easymock.PowerMock.replay;

/**
 * Asserts that {@link @TestSubject} with PowerMock and mock witch created via @org.powermock.api.easymock.annotation.Mock.
 */
@RunWith(PowerMockRunner.class)
@PrepareForTest(InjectDemo.class)
public class TestSubjectPowermockAnnotationTest {

    @SuppressWarnings("unused")
    @Mock
    private InjectDemo injectDemoEasymock;

    @TestSubject
    private final InjectDependencyHolder dependencyHolder = new InjectDependencyHolder();

    @SuppressWarnings("unused")
    @Mock(fieldName = "injectDemoQualifier")
    private InjectDemo injectDemoQualifierEasymock;

    @TestSubject
    private final InjectDependencyHolderQualifier dependencyHolderQualifier = new InjectDependencyHolderQualifier();


    @Test
    public void injectMocksWorksWithPowermockMock() {
        assertNotNull("dependencyHolder is null", dependencyHolder.getInjectDemo());
    }

    @Test
    public void testInjectDemoSay() throws Exception {

        InjectDemo tested = dependencyHolder.getInjectDemo();

        String expected = "Hello altered World";
        expectPrivate(tested,"say","hello").andReturn("Hello altered World");
        replay(tested);

        String actual = Whitebox.invokeMethod(tested,"say", "hello");

        assertEquals("Expected and actual did not match", expected, actual);
    }

    @Test
    public void injectMocksWorksWithPowermockMockWithQualifier() {
        assertNotNull("dependencyHolder is null", dependencyHolderQualifier.getInjectDemoQualifier());
    }

    @Test
    public void testInjectDemoQualifier() throws Exception {

        InjectDemo tested = dependencyHolderQualifier.getInjectDemoQualifier();

        String expected = "Hello altered World";
        expectPrivate(tested,"say","hello").andReturn("Hello altered World");
        replay(tested);

        String actual = Whitebox.invokeMethod(tested,"say", "hello");

        assertEquals("Expected and actual did not match", expected, actual);

    }

}