HandleDeviceAlreadyCreatedExceptionTest.java

package unit.ui.controllers.device_setup_controller.handle_go_button_clicked;

import exceptions.DeviceAlreadyCreatedException;
import javafx.scene.text.Text;
import org.jetbrains.annotations.Contract;
import org.junit.Test;

import java.util.Optional;

import static org.junit.Assert.fail;

public final class HandleDeviceAlreadyCreatedExceptionTest extends
        HandleGoButtonClickedTestCase {
    private static final String queryForMessage =
            "#device-created-exception-message";

    @Contract(" -> !null")
    @Override
    protected ExpectationsForMockFactory getMockingExpectations(){
        return new ExceptionExpectations();
    }

    @Test
    public void clickGoButton(){
        clickOn(queryForGoButton);

        Optional<Text> message = lookup(queryForMessage).tryQuery();

        if (!message.isPresent()){
            fail(
                    String.format(
                            "Did not find the message. I used the query %s",
                            queryForMessage
                    )
            );
        }
    }

    private class ExceptionExpectations extends ExpectationsForMockFactory {
        public ExceptionExpectations(){
            super();
            thrownExpectation();
        }

        private void thrownExpectation(){
            try {
                oneOf(factory).makePowerSupply();
                will(throwException(
                        new DeviceAlreadyCreatedException(
                                "The device already exists"
                        )
                ));
                oneOf(pressureGaugeFactory).makePressureGauge();
            } catch (Exception error){
                error.printStackTrace();
            }
        }
    }
}