ResultControllerTestCase.java

package unit.ui.controllers.results_controller;

import kernel.views.VariableProviderContainer;
import kernel.views.variables.VoltageProvider;
import org.jmock.Expectations;
import org.jmock.Mockery;
import org.junit.Before;
import unit.ui.MockVoltageProvider;
import unit.ui.controllers.ControllersTestCase;

public abstract class ResultControllerTestCase extends ControllersTestCase {

    protected static final String queryForResultsTab = "#results-tab";
    protected static final String queryForPressureTab = "#pressure-tab";
    protected static final String queryForVoltageTab = "#voltage-tab";
    protected static final String queryForVoltageChart = "#voltage-chart";

    protected final MockVoltageProvider mockVoltageProvider =
            (MockVoltageProvider) applicationContext.getBean(
                    VoltageProvider.class
            );

    @Before
    public void navigateToResultsTab(){
        clickOn(queryForResultsTab);
    }

    @Before
    public void setMockVoltageProvider(){
        applicationContext.getBean(Mockery.class).checking(
                new ExpectationsForTest()
        );
    }



    private class ExpectationsForTest extends Expectations {
        public ExpectationsForTest(){
            allowing(applicationContext.getBean(VariableProviderContainer.class)
            ).getVoltageProvider();
            will(returnValue(mockVoltageProvider));
        }
    }
}