Прежде чем углубляться в детали Mockito Framework, давайте посмотрим на приложение в действии. В этом примере мы создали макет Stock Service, чтобы получить фиктивную цену некоторых акций, а также протестировали модуль java под названием Portfolio.
Процесс обсуждается ниже в пошаговом порядке.
Шаг 1 — Создайте класс JAVA для представления акции
Файл: Stock.java
public class Stock { private String stockId; private String name; private int quantity; public Stock(String stockId, String name, int quantity){ this.stockId = stockId; this.name = name; this.quantity = quantity; } public String getStockId() { return stockId; } public void setStockId(String stockId) { this.stockId = stockId; } public int getQuantity() { return quantity; } public String getTicker() { return name; } }
Шаг 2 — Создание интерфейса StockService для получения цены акции
Файл: StockService.java
public interface StockService { public double getPrice(Stock stock); }
Шаг 3 — Создание портфолио класса для представления портфолио любого клиента
Файл: Portfolio.java
import java.util.List; public class Portfolio { private StockService stockService; private List<Stock> stocks; public StockService getStockService() { return stockService; } public void setStockService(StockService stockService) { this.stockService = stockService; } public List<Stock> getStocks() { return stocks; } public void setStocks(List<Stock> stocks) { this.stocks = stocks; } public double getMarketValue(){ double marketValue = 0.0; for(Stock stock:stocks){ marketValue += stockService.getPrice(stock) * stock.getQuantity(); } return marketValue; } }
Шаг 4 — Проверьте класс портфолио
Давайте проверим класс Portfolio, введя в него макет StockService. Mock будет создан Mockito.
Файл: PortfolioTester.java
package com.tutorialspoint.mock; import java.util.ArrayList; import java.util.List; import static org.mockito.Mockito.*; public class PortfolioTester { Portfolio portfolio; StockService stockService; public static void main(String[] args){ PortfolioTester tester = new PortfolioTester(); tester.setUp(); System.out.println(tester.testMarketValue()?"pass":"fail"); } public void setUp(){ //Create a portfolio object which is to be tested portfolio = new Portfolio(); //Create the mock object of stock service stockService = mock(StockService.class); //set the stockService to the portfolio portfolio.setStockService(stockService); } public boolean testMarketValue(){ //Creates a list of stocks to be added to the portfolio List<Stock> stocks = new ArrayList<Stock>(); Stock googleStock = new Stock("1","Google", 10); Stock microsoftStock = new Stock("2","Microsoft",100); stocks.add(googleStock); stocks.add(microsoftStock); //add stocks to the portfolio portfolio.setStocks(stocks); //mock the behavior of stock service to return the value of various stocks when(stockService.getPrice(googleStock)).thenReturn(50.00); when(stockService.getPrice(microsoftStock)).thenReturn(1000.00); double marketValue = portfolio.getMarketValue(); return marketValue == 100500.0; } }
Шаг 5 — Проверьте результат
Скомпилируйте классы, используя компилятор javac следующим образом:
C:\Mockito_WORKSPACE>javac Stock.java StockService.java Portfolio.java PortfolioTester.java
Теперь запустите PortfolioTester, чтобы увидеть результат —
C:\Mockito_WORKSPACE>java PortfolioTester
Проверьте вывод