51Testing软件测试论坛

标题: Java单元测试——Mock技术 [打印本页]
作者: lsekfe    时间: 2020-12-7 10:41
标题: Java单元测试——Mock技术
1.源代码
　　AccountService.java
  1. 　package com.account;

  3. 　　import com.account.Account;

  5. 　　import com.account.AccountManager;

  7. 　　public class AccountService

  9. 　　{

  11. 　　//使用的帐户管理器实现

  13. 　　    private AccountManager accountManager;

  15. 　　    //设置帐户管理器实现的设置方法

  17. 　　    public void setAccountManager( AccountManager manager )

  19. 　　    {

  21. 　　        this.accountManager = manager;

  23. 　　    }

  25. 　　    //一个设置客户经理实现从账户到账户的senderId beneficiaryId setter方法。

  27. 　　    //senderId:转出方Id

  29. 　　    //beneficiaryId：收益方Id

  31. 　　    //amount:金额

  33. 　　    public void transfer( String senderId, String beneficiaryId, long amount )

  35. 　　    {

  37. 　　        //初始化转出方与收益方，findAccountForUser为接口类方法

  39. 　　    Account sender = this.accountManager.findAccountForUser( senderId );

  41. 　　        Account beneficiary = this.accountManager.findAccountForUser( beneficiaryId );

  43. 　　        //转入和收益

  45. 　　        sender.debit( amount );

  47. 　　        beneficiary.credit( amount );

  49. 　　        //更新，updateAccount为接口类方法

  51. 　　        this.accountManager.updateAccount( sender );

  53. 　　        this.accountManager.updateAccount( beneficiary );

  55. 　　    }

  57. 　　}
复制代码
Account.java
  1. 　package com.account;

  3. 　　public class Account

  5. 　　{

  7. 　　    private String accountId;

  9. 　　    private long balance;

  11. 　　    public Account(String accountId, long initialBalance)

  13. 　　    {

  15. 　　        this.accountId = accountId;

  17. 　　        this.balance = initialBalance;

  19. 　　    }

  21. 　　    //借记

  23. 　　    public void debit( long amount )

  25. 　　    {

  27. 　　        this.balance -= amount;

  29. 　　    }

  31. 　　    //信用

  33. 　　    public void credit( long amount )

  35. 　　    {

  37. 　　        this.balance += amount;

  39. 　　    }

  41. 　　    public long getBalance()

  43. 　　    {

  45. 　　        return this.balance;

  47. 　　    }

  49. 　　}
复制代码
AccountManager.java
  1. 　package com.account;

  3. 　　import com.account.Account;

  5. 　　public interface AccountManager

  7. 　　{

  9. 　　    Account findAccountForUser(String userId );

  11. 　　    void updateAccount(Account account );   

  13. 　　}
复制代码
由于在这里AccountManager.java仅仅做了一个interface，我们主要Mock的是这个类。这几个类的类关系图如下：
[attach]131172[/attach]
　通常的调用方法如下：
  1. @Test

  3. 　　public void testTransferOK() {

  5. 　　Account sendAccount = new Account("1",200);

  7. 　　Account beneficiaryAccount = new Account("2",100);

  9. 　　AccountManager. updateAccount( senderAccount );

  11. 　　AccountManager.updateAccount( beneficiaryAccount );

  13. 　　AccountManager.findAccountForUser("1" )

  15. 　　AccountManager.findAccountForUser( "2" )

  17. 　　AccountService accountService = new AccountService();

  19. 　　accountService.setAccountManager(AccountManager);

  21. 　　accountService.transfer("1","2",50); //转钱

  23. 　　Assertions.assertEquals(150,sendAccount.getBalance());

  25. 　　Assertions.assertEquals(150,beneficiaryAccount.getBalance());

  27. 　　}
复制代码
2.最通用的Mock技术　　StubAccountManager.java
  1. package com.account;

  3. 　　import java.util.HashMap;

  5. 　　public class StubAccountManager implements AccountManager{

  7. 　　private HashMap<String,Account> accounts = new HashMap<String,Account>();

  9. 　　public void addAcount(String userId,Account account){

  11. 　　this.accounts.put(userId,account);

  13. 　　}

  15. 　　

  17. 　　public Account findAccountForUser(String userId){

  19. 　　return this.accounts.get(userId);

  21. 　　}

  23. 　　

  25. 　　public void updateAccount(Account account){

  27. 　　//do nothing

  29. 　　}

  31. 　　}
复制代码
Account.java
  1. import static org.junit.Assert.assertEquals;

  3. 　　import org.junit.jupiter.api.Test;

  5. 　　public class TestAccountService {

  7. 　　@Test

  9. 　　public void testTransferOK() {

  11. 　　StubAccountManager stubAccountManager = new StubAccountManager(); //定义MockAccountManager类

  13. 　　Account sendAccount = new Account("1",200); //定义收钱方和出钱方两个Account

  15. 　　Account beneficiaryAccount = new Account("2",100);

  17. 　　stubAccountManager.addAcount("1", sendAccount); //初始化收钱方和出钱方HashMap

  19. 　　stubAccountManager.addAcount("2", beneficiaryAccount);

  21. 　　AccountService accountService = new AccountService(); //初始化AccountService类

  23. 　　accountService.setAccountManager(stubAccountManager); //初始化AccountManager

  25. 　　accountService.transfer("1","2",50); //转钱

  27. 　　Assertions.assertEquals(150,sendAccount.getBalance()); //判断转换后收付方金额是否正确

  29. 　　Assertions.assertEquals(150,beneficiaryAccount.getBalance());

  31. 　　}}
复制代码
3．EasyMock技术　　EasyMock需要以下两个jar包：easymock-2.4.jar和easymockclassextension-2.4.jar
　　TestAccountServiceEasyMock.java
  1. package com.account;

  3. 　　import static org.easymock.EasyMock.createMock;

  5. 　　import static org.easymock.EasyMock.replay;

  7. 　　import static org.easymock.EasyMock.expect;

  9. 　　import static org.easymock.EasyMock.verify;

  11. 　　import org.junit.jupiter.api.BeforeEach;

  13. 　　import org.junit.jupiter.api.AfterEach;

  15. 　　import org.junit.jupiter.api.Assertions;

  17. 　　import org.junit.jupiter.api.DisplayName;

  19. 　　import org.junit.jupiter.api.Test;

  21. 　　import com.account.Account;

  23. 　　import com.account.AccountManager;

  25. 　　import com.account.AccountService;

  27. 　　public class TestAccountServiceEasyMock {

  29. 　　private AccountManager mockAccountManager;

  31. 　　    @BeforeEach

  33. 　　    public void setUp()

  35. 　　    {

  37. 　　        //初始化easyMock

  39. 　　    mockAccountManager = createMock("mockAccountManager", AccountManager.class );

  41. 　　    }

  43. 　　    @Test

  45. 　　    @DisplayName("测试转账")

  47. 　　    public void testTransferOk()

  49. 　　    {

  51. 　　        Account senderAccount = new Account( "1", 200 );

  53. 　　        Account beneficiaryAccount = new Account( "2", 100 );

  55. 　　        

  57. 　　        //开始定义期望

  59. 　　        mockAccountManager.updateAccount( senderAccount );

  61. 　　        mockAccountManager.updateAccount( beneficiaryAccount );

  63. 　　        //EasyMock的expect和replay方法

  65. 　　        expect( mockAccountManager.findAccountForUser( "1" ) ).andReturn( senderAccount ); //期望返回senderAccount

  67. 　　        expect( mockAccountManager.findAccountForUser( "2" ) ).andReturn( beneficiaryAccount ); //期望返beneficiaryAccount

  69. 　　        replay( mockAccountManager );//切换到replay状态 Record-> replay，在replay状态才可以进行验证

  71. 　　        AccountService accountService = new AccountService();

  73. 　　        accountService.setAccountManager( mockAccountManager );

  75. 　　        accountService.transfer( "1", "2", 50 );

  77. 　　        Assertions.assertEquals( 150, senderAccount.getBalance() );

  79. 　　        Assertions.assertEquals( 150, beneficiaryAccount.getBalance() );

  81. 　　    }

  83. 　　    @AfterEach

  85. 　　    public void tearDown()

  87. 　　    {

  89. 　　        verify( mockAccountManager );

  91. 　　    }

  93. 　　}
复制代码
4. JMock技术　　JMock依赖下面11个jar包。另外JMock不完全兼容JUnit5
[attach]131173[/attach]
　TestAccountServiceJMock.java
  1. package com.account;

  3. 　　import org.jmock.integration.junit4.JMock;

  5. 　　import org.jmock.integration.junit4.JUnit4Mockery;

  7. 　　import org.jmock.Expectations;

  9. 　　import org.jmock.Mockery;

  11. 　　import org.junit.jupiter.api.Assertions;

  13. 　　import org.junit.jupiter.api.BeforeEach;

  15. 　　import org.junit.jupiter.api.DisplayName;

  17. 　　import org.junit.jupiter.api.Test;

  19. 　　import org.junit.runner.RunWith;

  21. 　　import com.account.Account;

  23. 　　import com.account.AccountManager;

  25. 　　import com.account.AccountService;

  27. 　　@RunWith(JMock.class)

  29. 　　public class TestAccountServiceJMock {

  31. 　　/**

  33. 　　     * The mockery context that we use to create our mocks.

  35. 　　     */

  37. 　　    private Mockery context = new JUnit4Mockery();

  39. 　　    /**

  41. 　　     * The mock instance of the AccountManager to use.

  43. 　　     */

  45. 　　    private AccountManager mockAccountManager;

  47. 　　@BeforeEach

  49. 　　public void setUp(){

  51. 　　 mockAccountManager = context.mock( AccountManager.class );

  53. 　　}

  55. 　　@Test

  57. 　　@DisplayName("测试转账")

  59. 　　    public void testTransferOk()

  61. 　　    {

  63. 　　        final Account senderAccount = new Account( "1", 200 );

  65. 　　        final Account beneficiaryAccount = new Account( "2", 100 );

  67. 　　        context.checking( new Expectations()

  69. 　　        {

  71. 　　            {

  73. 　　                oneOf( mockAccountManager ).findAccountForUser( "1" );

  75. 　　                will( returnValue( senderAccount ) );

  77. 　　                oneOf( mockAccountManager ).findAccountForUser( "2" );

  79. 　　                will( returnValue( beneficiaryAccount ) );

  81. 　　                oneOf( mockAccountManager ).updateAccount( senderAccount );

  83. 　　                oneOf( mockAccountManager ).updateAccount( beneficiaryAccount );

  85. 　　            }

  87. 　　        } );

  89. 　　        AccountService accountService = new AccountService();

  91. 　　        accountService.setAccountManager( mockAccountManager );

  93. 　　        accountService.transfer( "1", "2", 50 );

  95. 　　        Assertions.assertEquals( 150, senderAccount.getBalance() );

  97. 　　        Assertions.assertEquals( 150, beneficiaryAccount.getBalance() );

  99. 　　    }  

  101. 　　}
复制代码
4.1 One，one of　　JMock2.4版以前：one；
　　JMock2.51版以后：one of。
  1. oneOf (anObject).doSomething(); will(returnValue(10));

  3. 　　oneOf (anObject).doSomething(); will(returnValue(20));

  5. 　　oneOf (anObject).doSomething(); will(returnValue(30));
复制代码
　第一次调用时会返回10，第二次会返回20，第三次会返回30。　　4.2 atLeast(n).of
  1. 　atLeast(1).of (anObject).doSomething();  

  3. 　　will(onConsecutiveCalls( returnValue(10),  returnValue(20),  returnValue(30)));
复制代码
　这里atLeast (1)表明doSomething方法将至少被调用一次，但不超过3次。且调用的返回值分别是10、20、30.
[attach]131174[/attach]
5. mockito技术　　需要mockito-all-1.9.5.jar包。
  1. 　package com.account;

  3. 　　import static org.mockito.Mockito.*;

  5. 　　import org.junit.jupiter.api.Assertions;

  7. 　　import org.junit.jupiter.api.BeforeEach;

  9. 　　import org.junit.jupiter.api.DisplayName;

  11. 　　import org.junit.jupiter.api.Test;

  13. 　　import org.mockito.Mockito;

  15. 　　import com.account.Account;

  17. 　　import com.account.AccountManager;

  19. 　　import com.account.AccountService;

  21. 　　public class TestAccountServiceMockito {

  23. 　　private AccountManager mockAccountManager;

  25. 　　private Account senderAccount;

  27. 　　private Account beneficiaryAccount;

  29. 　　@BeforeEach

  31. 　　public void setUp(){

  33. 　　mockAccountManager = Mockito.mock(AccountManager.class);

  35. 　　

  37. 　　senderAccount = new Account( "1", 200 );

  39. 　　        beneficiaryAccount = new Account( "2", 100 );

  41. 　　        

  43. 　　        mockAccountManager.updateAccount( senderAccount );

  45. 　　        mockAccountManager.updateAccount( beneficiaryAccount );

  47. 　　        

  49. 　　        when(mockAccountManager.findAccountForUser("1")).thenReturn( senderAccount );

  51. 　　        when(mockAccountManager.findAccountForUser("2")).thenReturn( beneficiaryAccount );

  53. 　　}

  55. 　　

  57. 　　@Test

  59. 　　@DisplayName("测试转账")

  61. 　　public void test() {

  63. 　　        AccountService accountService = new AccountService();

  65. 　　        accountService.setAccountManager( mockAccountManager );

  67. 　　        accountService.transfer( "1", "2", 50 );

  69. 　　        Assertions.assertEquals( 150, senderAccount.getBalance() );

  71. 　　        Assertions.assertEquals( 150, beneficiaryAccount.getBalance() );

  73. 　　}

  75. 　　}
复制代码





















欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/) Powered by Discuz! X3.2