import org.openqa.selenium.*;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.Parameters;
import org.testng.annotations.Test;

import static org.testng.Assert.assertEquals;

public class PolymorphismSample {

  private WebDriver driver;

  @Parameters({"browserName"})
  @BeforeTest
  public void setUp(String browserName) {
    if (browserName.equals("firefox")) {
      driver = new FirefoxDriver();
    } else if (browserName.equals("chrome")) {
      driver = new ChromeDriver();
    }
  }

  @Test
  public void loginTest() {
    driver.get("https://lingualeo.com");
    driver.findElement(By.id("login")).sendKeys("admin");
    driver.findElement(By.id("pass")).sendKeys("changeme");
    assertEquals("Home Page", driver.getTitle());
  }

}