ProductPresenterTest.java

package org.openlmis.core.presenter;

import com.google.inject.AbstractModule;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.openlmis.core.LMISTestRunner;
import org.openlmis.core.model.Product;
import org.openlmis.core.model.RegimeShortCode;
import org.openlmis.core.model.Regimen;
import org.openlmis.core.model.StockCard;
import org.openlmis.core.model.builder.ProductBuilder;
import org.openlmis.core.model.repository.ProductRepository;
import org.openlmis.core.model.repository.ProgramRepository;
import org.openlmis.core.model.repository.RegimenRepository;
import org.openlmis.core.model.repository.StockRepository;
import org.openlmis.core.view.viewmodel.InventoryViewModel;
import org.openlmis.core.view.viewmodel.RegimeProductViewModel;
import org.robolectric.RuntimeEnvironment;

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

import roboguice.RoboGuice;
import rx.observers.TestSubscriber;

import static org.assertj.core.util.Lists.newArrayList;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@RunWith(LMISTestRunner.class)
public class ProductPresenterTest {

    private ProductPresenter presenter;
    private ProductRepository productRepository;
    private ProgramRepository programRepository;
    private RegimenRepository regimenRepository;
    private StockRepository stockRepository;

    @Before
    public void setup() throws Exception {
        productRepository = mock(ProductRepository.class);
        programRepository = mock(ProgramRepository.class);
        stockRepository = mock(StockRepository.class);
        regimenRepository = mock(RegimenRepository.class);

        RoboGuice.overrideApplicationInjector(RuntimeEnvironment.application, new AbstractModule() {
            @Override
            protected void configure() {
                bind(ProductRepository.class).toInstance(productRepository);
                bind(ProgramRepository.class).toInstance(programRepository);
                bind(RegimenRepository.class).toInstance(regimenRepository);
                bind(StockRepository.class).toInstance(stockRepository);
            }
        });
        presenter = RoboGuice.getInjector(RuntimeEnvironment.application).getInstance(ProductPresenter.class);
    }

    @Test
    public void shouldLoadRegimeProducts() throws Exception {
        // when
        RegimeShortCode regimeShortCode = new RegimeShortCode();
        regimeShortCode.setCode("code");
        regimeShortCode.setShortCode("3TC 150mg");
        when(regimenRepository.listRegimeShortCode()).thenReturn(newArrayList(regimeShortCode));
        Product product = new Product();
        product.setPrimaryName("PrimaryName");
        when(productRepository.getByCode("code")).thenReturn(product);
        TestSubscriber<List<RegimeProductViewModel>> subscriber = new TestSubscriber<>();
        presenter.loadRegimeProducts().subscribe(subscriber);

        subscriber.awaitTerminalEvent();

        subscriber.assertNoErrors();

        assertThat(subscriber.getOnNextEvents().get(0).size(), is(1));
        assertThat(subscriber.getOnNextEvents().get(0).get(0).getShortCode(), is("3TC 150mg"));
        assertThat(subscriber.getOnNextEvents().get(0).get(0).getEntireName(), is("PrimaryName"));
    }

    @Test
    public void loadEmergencyProducts() throws Exception {
        StockCard stockCard = new StockCard();
        stockCard.setProduct(new ProductBuilder().setPrimaryName("Product name").setCode("011111").build());
        when(stockRepository.listEmergencyStockCards()).thenReturn(newArrayList(stockCard));

        TestSubscriber<List<InventoryViewModel>> subscriber = new TestSubscriber<>();
        presenter.loadEmergencyProducts().subscribe(subscriber);

        subscriber.awaitTerminalEvent();

        subscriber.assertNoErrors();

        assertThat(subscriber.getOnNextEvents().get(0).size(), is(1));
        assertThat(subscriber.getOnNextEvents().get(0).get(0).getProductName(), is("Product name"));
    }

    @Test
    public void shouldSaveRegime() throws Exception {
        when(regimenRepository.getByNameAndCategory(anyString(), any(Regimen.RegimeType.class))).thenReturn(null);

        TestSubscriber<Regimen> subscriber = new TestSubscriber<>();
        presenter.saveRegimes(getInventoryViewModels(), Regimen.RegimeType.Adults).subscribe(subscriber);

        subscriber.awaitTerminalEvent();

        subscriber.assertNoErrors();

        ArgumentCaptor<Regimen> regimenArgumentCaptor = ArgumentCaptor.forClass(Regimen.class);
        verify(regimenRepository).create(regimenArgumentCaptor.capture());

        Regimen regimen = regimenArgumentCaptor.getValue();
        assertThat(regimen.getName(), is("3TC 150mg+3TC 150mg"));
        assertThat(regimen.getType(), is(Regimen.RegimeType.Adults));
        assertThat(regimen.isCustom(), is(true));
    }

    @Test
    public void shouldNotSaveRegimeWhenHasRegimeInDB() throws Exception {
        when(regimenRepository.getByNameAndCategory(anyString(), any(Regimen.RegimeType.class))).thenReturn(new Regimen());

        TestSubscriber<Regimen> subscriber = new TestSubscriber<>();
        presenter.saveRegimes(getInventoryViewModels(), Regimen.RegimeType.Adults).subscribe(subscriber);

        subscriber.awaitTerminalEvent();

        subscriber.assertNoErrors();

        verify(regimenRepository, never()).create(any(Regimen.class));
    }

    private ArrayList<RegimeProductViewModel> getInventoryViewModels() {
        return newArrayList(new RegimeProductViewModel("3TC 150mg", "Lamivudina 150mg"), new RegimeProductViewModel("3TC 150mg", "Lamivudina 150mg"));
    }
}