Проблема с вашим кодом - это указанный вами xpath. Нет никаких
< div class="tt-dataset">
Метка показывается для любых предложений, которые вы получаете после любого поиска
<div class="tt-station tt-suggestion tt-selectable">
Для выбора первого предложения вы можете использовать findElements
.
WebElement textInput = driver.findElement(By.xpath("//input[@id='fromStation']"));
textInput.sendKeys("Warszawa");
List<WebElement> suggestionList = driver.findElements(By.xpath("//div[@class='tt-station tt-suggestion tt-selectable > span > i']"));
suggestion.get(0).click();
Если вы хотите щелкнуть по любому другому элементу, вы можете указать порядковый номер в соответствии с вашими потребностями.