PowerMock笔记-Answer接口

最新推荐文章于 2024-11-09 14:28:12 发布

原创最新推荐文章于 2024-11-09 14:28:12 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

PowerMock 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一个使用PowerMockito进行单元测试的例子，通过模拟EmployeeDao类的行为来验证EmployeeService类的功能。测试中覆盖了正常情况及异常情况，确保业务逻辑正确。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import lombok.Data;

/**
 * @author chenjianfei
 */
@Data
public class Employee {
    private String name;
    private Integer age;
}

public class EmployeeDao {
    public String findByName(String name) {
        throw new UnsupportedOperationException();
    }
}

public class EmployeeService {

    public String find(String name) {
        EmployeeDao employeeDao = new EmployeeDao();
        return employeeDao.findByName(name);
    }
}

import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Answers;
import org.mockito.ArgumentMatchers;
import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import static org.junit.Assert.*;

@RunWith(PowerMockRunner.class)
@PrepareForTest(EmployeeService.class)
public class EmployeeServiceTest {

    @Test
    public void findByNameWithAnswer() {
        try {
            EmployeeDao employeeDao = PowerMockito.mock(EmployeeDao.class);
            PowerMockito.whenNew(EmployeeDao.class).withAnyArguments().thenReturn(employeeDao);
            PowerMockito.when(employeeDao.findByName(Mockito.anyString())).then(invocation -> {
                String args = (String) invocation.getArguments()[0];
                switch (args) {
                    case "Jacky":
                        return "I am Jacky";
                    case "Andy":
                        return "I am Andy";
                    default:
                        throw new RuntimeException("Not Support:" + args);
                }
            });
            EmployeeService employeeService = new EmployeeService();
            Assert.assertEquals("I am Jacky",employeeService.find("Jacky"));
            Assert.assertEquals("I am Andy",employeeService.find("Andy"));

            try{
                String anyValue = employeeService.find("anyValue");
                Assert.fail();
            }catch (Exception e){
                Assert.assertTrue(e instanceof RuntimeException);
            }
        } catch (Exception e) {
            Assert.fail();
        }
    }
}