implement functions in java; adjust tests

Author: AdamPDotty

java/src/org/openqa/selenium/support/locators/RelativeLocator.java

@@ -147,6 +147,50 @@ public RelativeBy toRightOf(By locator) {
       return simpleDirection("right", locator);
     }
 
+    public RelativeBy straightAbove(WebElement element) {
+      Require.nonNull("Element to search straight above of", element);
+      return simpleDirection("straightAbove", element);
+    }
+
+    public RelativeBy straightAbove(By locator) {
+      Require.nonNull("Locator", locator);
+      assertLocatorCanBeSerialized(locator);
+      return simpleDirection("straightAbove", locator);
+    }
+
+    public RelativeBy straightBelow(WebElement element) {
+      Require.nonNull("Element to search straight below of", element);
+      return simpleDirection("straightBelow", element);
+    }
+
+    public RelativeBy straightBelow(By locator) {
+      Require.nonNull("Locator", locator);
+      assertLocatorCanBeSerialized(locator);
+      return simpleDirection("straightBelow", locator);
+    }
+
+    public RelativeBy straightLeftOf(WebElement element) {
+      Require.nonNull("Element to search straight to the left of", element);
+      return simpleDirection("straightLeft", element);
+    }
+
+    public RelativeBy straightLeftOf(By locator) {
+      Require.nonNull("Locator", locator);
+      assertLocatorCanBeSerialized(locator);
+      return simpleDirection("straightLeft", locator);
+    }
+
+    public RelativeBy straightRightOf(WebElement element) {
+      Require.nonNull("Element to search straight to the right of", element);
+      return simpleDirection("straightRight", element);
+    }
+
+    public RelativeBy straightRightOf(By locator) {
+      Require.nonNull("Locator", locator);
+      assertLocatorCanBeSerialized(locator);
+      return simpleDirection("straightRight", locator);
+    }
+
     public RelativeBy near(WebElement element) {
       Require.nonNull("Element to search near", element);
       return near(element, CLOSE_IN_PIXELS);

java/test/org/openqa/selenium/support/locators/RelativeLocatorTest.java

@@ -184,7 +184,7 @@ void shouldBeAbleToFindElementsStraightRightOfAnother() {
 
     WebElement left = driver.findElement(By.id("left"));
 
-    List<WebElement> elements = driver.findElements(with(tagName("td")).straightAbove(left));
+    List<WebElement> elements = driver.findElements(with(tagName("td")).straightRight(left));
     List<String> ids =
         elements.stream().map(e -> e.getAttribute("id")).collect(Collectors.toList());