ServiceATest_Test4J.java

package org.test4j.testng.jmockit.demo1;

import java.math.BigDecimal;
import java.util.Collections;
import java.util.List;

import mockit.Mock;

import org.test4j.module.core.utility.MessageHelper;
import org.test4j.module.jmockit.demo1.Database;
import org.test4j.module.jmockit.demo1.EntityX;
import org.test4j.module.jmockit.demo1.ServiceA;
import org.test4j.module.jmockit.demo1.ServiceB;
import org.test4j.testng.Test4J;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

@SuppressWarnings("rawtypes")
@Test(groups = "test4j")
public class ServiceATest_Test4J extends Test4J {

    public static class MockDatabase extends MockUp<Database> {
        @Mock(invocations = 1)
        public static List<?> find(String ql, Object arg1) {
            want.string(ql).notNull();
            want.object(arg1).notNull();
            return Collections.EMPTY_LIST;
        }

        @Mock(maxInvocations = 1)
        public static void save(Object o) {
            want.object(o).notNull();
        }
    }

    @BeforeMethod
    public void setUp() {
        new MockDatabase();
    }

    @Test
    public void doBusinessOperationXyz() throws Exception {
        final BigDecimal total = new BigDecimal("125.40");

        MessageHelper.info("doBusinessOperationXyz-test4j");
        new MockUp(ServiceB.class) {
            @Mock(invocations = 1)
            public BigDecimal computeTotal(List<?> items) {
                want.collection(items).notNull();
                return total;
            }
        };

        EntityX data = new EntityX(5, "abc", "5453-1");
        new ServiceA().doBusinessOperationXyz(data);
        want.number(data.getTotal()).isEqualTo(total);
    }

    @Test(expectedExceptions = Exception.class)
    public void doBusinessOperationXyzWithInvalidItemStatus() throws Exception {
        new MockUp(ServiceB.class) {
            @Mock
            public BigDecimal computeTotal(List<?> items) throws Exception {
                throw new Exception();
            }
        };

        EntityX data = new EntityX(5, "abc", "5453-1");
        new ServiceA().doBusinessOperationXyz(data);
    }
}