使用 EasyMock 模拟方法

Question

I try to create unit test fro the following method but i can't find a solution to mock the call inside each method, Could you please help me with creating Junit Test for those methods using EasyMock :

private static final WebServiceCache<JWebService> SERVICE = new WebServiceCache<>();
public int getCount() {
    int res = -1;
    try {
        String count = SERVICE.invokeSecurelly(new WS<String>() {
            @Override
            public String execute() throws Exception {
                return getWS().getList();
            }
        });
        res = Integer.parseInt(count);
    } catch (Exception e) {
        LOGGER.error("Count Exception" + e);
    }
    return res;
}

public int getKeyNumber() {
    int res = -1;
    try {
        String keyId = SERVICE.invokeSecurelly(new WS<String>() {
            @Override
            public String execute() throws Exception {
                return getWS().getID();
            }
        });
        
        res = Integer.parseInt(keyId);
    } catch (Exception e) {
        LOGGER.error("getKeyNumBer returns an error" + e);
    }
    return res;
}

Thanks in advance

Answer 1

To do something clean, you need some refactoring to make it testable. Here is what I would end up with... guessing part of the missing core in your example.

public class MyClass {
    private static final Logger LOGGER = Logger.getLogger(MyClass.class.getName());
    private final WebServiceCache<JWebService> service;

    public MyClass(WebServiceCache<JWebService> service) {
        this.service = service;
    }
    
    private int getValue(Supplier<String> invoked) {
        try {
            String count = service.invokeSecurelly(invoked::get);
            return Integer.parseInt(count);
        } catch (Exception e) {
            LOGGER.error("Count Exception", e);
        }
        return -1;
    }

    public int getCount() {
        return getValue(() -> getWS().getList());
    }

    public int getKeyNumber() {
        return getValue(() -> getWS().getID());
    }

    private Stuff getWS() { // guessing where getWS() is
        return null;
    }
}

From there, if we assume it's getWS() that you want to mock, it would look like this.

import org.easymock.EasyMockSupport;
import org.junit.Before;
import org.junit.Test;

import static org.assertj.core.api.Assertions.assertThat;
import static org.easymock.EasyMock.expect;

public class MyClassTest extends EasyMockSupport {

    private WebServiceCache<JWebService> cache = new WebServiceCache<>();
    private MyClass tested = partialMockBuilder(MyClass.class)
            .addMockedMethod("getWS")
            .withConstructor(cache)
            .mock();
    private Stuff stuff = mock(Stuff.class);

    @Before
    public void before() {
        expect(tested.getWS()).andStubReturn(stuff);
    }

    @Test
    public void getCount() {
        expect(stuff.getList()).andStubReturn("8");
        replayAll();

        assertThat(tested.getCount()).isEqualTo(8);
    }

    @Test
    public void getKeyNumber() {
        expect(stuff.getID()).andStubReturn("8");
        replayAll();

        assertThat(tested.getKeyNumber()).isEqualTo(8);
    }
}