Cucumber-Jvm WebDriver implementation switching

Hooks.java

public class Hooks {

    @Autowired
    private WebDriverFactory webDriverFactory;

    @Before("@javascript")
	public void setJavascriptWebDriver() {
        webDriverFactory.selectJavascriptDriver();
        webDriverFactory.getCurrentWebDriver().manage().timeouts().implicitlyWait(1000, TimeUnit.MILLISECONDS);
    }

    @Before("~@javascript")
    public void setNoJavascriptWebDriver() {
        webDriverFactory.selectDefaultDriver();
    }

}

webdriver-context.xml

<?xml version="1.0" encoding="UTF-8"?>
<beans ...>

    <bean name="default-webdriver" class="org.openqa.selenium.htmlunit.HtmlUnitDriver" lazy-init="true" />
    <bean name="javascript-webdriver" class="org.openqa.selenium.firefox.FirefoxDriver" destroy-method="close" lazy-init="true" />

</beans>

WebDriverFactory.java

@Component
public class WebDriverFactory implements ApplicationContextAware {

    private ApplicationContext applicationContext;

    private WebDriver currentWebDriver;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) {
        this.applicationContext = applicationContext;
    }

    public void selectJavascriptDriver() {
        currentWebDriver = getJsWebDriver();
    }

    public void selectDefaultDriver() {
        currentWebDriver = getDefaultWebDriver();
    }

    public WebDriver getCurrentWebDriver() {
        return currentWebDriver;
    }

    private WebDriver getJsWebDriver() {
        return applicationContext.getBean("javascript-webdriver", WebDriver.class);
    }

    private WebDriver getDefaultWebDriver() {
        return applicationContext.getBean("default-webdriver", WebDriver.class);
    }
}