TestStubForReturnMethod.java

最新推荐文章于 2021-11-02 21:24:52 发布

原创最新推荐文章于 2021-11-02 21:24:52 发布 · 292 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#code #单元测试

本文介绍了一个使用Mockito进行单元测试的例子，演示了如何为返回类型的方法设置预期结果，并展示了四种不同的测试方法实现。

import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Spy;

import static org.mockito.Mockito.*;
import org.testng.annotations.Test;

public class TestStubForReturnMethod extends BaseTeser{
        @Mock
        private Weapon weapon1;

        @Spy
        private Weapon weapon2;

        @Mock
        private Weapon weapon;

        @InjectMocks
        private Soldier soldier;

        //@Test
        public void test1(){
            when(weapon1.getWeaponName()).thenReturn("Gun");
            p(weapon1.getWeaponName());
        }

        //@Test
        public void test2(){
            when(weapon2.getWeaponName()).thenReturn("Gun");
            p("===");
            p(weapon2.getWeaponName());
        }

        //@Test
        public void test3(){
            doReturn("Gun").when(weapon2).getWeaponName();
            p("===");
            p(weapon2.getWeaponName());
        }

        @Test
        public void test4(){
            doReturn("Gun").when(weapon).getWeaponName();
            p("===");
            soldier.showWeapon();
        }
}