SeleniumHelper.java 1.2 KB

123456789101112131415161718192021222324252627282930313233
  1. package jnpf.selenium;
  2. import jnpf.selenium.consts.SeleniumConsts;
  3. import jnpf.selenium.driver.SeleniumBrowser;
  4. import jnpf.selenium.properties.SeleniumProperties;
  5. import lombok.Getter;
  6. import lombok.Setter;
  7. import org.springframework.beans.BeansException;
  8. import org.springframework.context.ApplicationContext;
  9. import org.springframework.context.ApplicationContextAware;
  10. import org.springframework.context.annotation.Configuration;
  11. @Configuration(proxyBeanMethods = false)
  12. public class SeleniumHelper implements ApplicationContextAware {
  13. private static ApplicationContext applicationContext;
  14. @Getter
  15. @Setter
  16. private static SeleniumProperties seleniumProperties;
  17. public static <T extends SeleniumBrowser<?, ?, ?>> T getBrowser() {
  18. SeleniumBrowser<?, ?, ?> seleniumBrowser = (SeleniumBrowser<?, ?, ?>) applicationContext.getBean(SeleniumConsts.BROWSER_BEAN_PREFIX + seleniumProperties.getBrowserType());
  19. return (T) seleniumBrowser;
  20. }
  21. @Override
  22. public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
  23. SeleniumHelper.applicationContext = applicationContext;
  24. SeleniumHelper.seleniumProperties =applicationContext.getBean(SeleniumProperties.class);
  25. }
  26. }