service层

IdcServiceTest.java

import com.XXX.sysdev.virtualsys.BaseTest;
import com.XXX.sysdev.virtualsys.dao.IdcDao;
import com.XXX.sysdev.virtualsys.model.Idc;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import java.util.ArrayList;
import java.util.List;

import static org.mockito.Matchers.anyInt;
import static org.mockito.Matchers.anyLong;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.when;
import static org.testng.Assert.assertEquals;


@Test
public class IdcServiceTest extends BaseTest {

    @Mock(name="idcDao")
    private IdcDao idcDao;

    @InjectMocks
    private IdcService idcService;

    private static  List<Idc> idcs ;


    @BeforeMethod(groups = "idc_service")
    public void testInit(){
        idcs = new ArrayList<Idc>();
        Idc idc1 = new Idc();
        idc1.setIdcId(123L);
        idc1.setIdcName("test01");
        idc1.setIp("127.0.0.1");
        idc1.setPort(8083);
        idcs.add(idc1);

        Idc idc2 = new Idc();
        idc2.setIdcId(124L);
        idc2.setIdcName("test02");
        idc2.setIp("127.0.0.2");
        idc2.setPort(8083);
        idcs.add(idc2);

    }

    /**
     * 测试获取所有的机房
     * */
    @Test(groups = "idc_service")
    public void testGetAll(){
        when(idcDao.getList()).thenReturn(idcs);
        List<Idc> results = idcService.getAll();
        assertEquals(2,results.size());
        assertEquals(results.get(0).getIdcName(),"test01");
    }

    @Test(groups = "idc_service")
    public void testGetByIdcId(){
        when(idcDao.getByIdcId(anyLong())).thenReturn(idcs.get(0));
        Idc idc = idcService.getByIdcId(123L);
        assertEquals("test01",idc.getIdcName());
        assertEquals("127.0.0.1",idc.getIp());

    }

    @Test(groups = "idc_service")
    public void testFindIdc(){
        when(idcDao.find(anyString(),anyInt())).thenReturn(idcs.get(0));
        Idc idc = idcService.findIdc("127.0.0.1",8083);
        assertEquals("test01",idc.getIdcName());

    }
}