Refactor tests to use general method

SolutionGuidance · Aug 28, 2018 · 923c986 · 923c986
1 parent bf89be1
commit 923c986
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/...gration-tests/src/test/java/gov/medicaid/features/enrollment/ui/OrganizationInfoPage.java b/...gration-tests/src/test/java/gov/medicaid/features/enrollment/ui/OrganizationInfoPage.java
@@ -2,8 +2,6 @@
 
 import cucumber.api.PendingException;
 
-import static org.assertj.core.api.Assertions.assertThat;
-
 public class OrganizationInfoPage extends EnrollmentPage {
     private static final String ORGANIZATION_FEIN_ERROR_MESSAGE =
             "Organization FEIN length must be 9 characters.";
@@ -105,7 +103,6 @@ public void setContactEmail(String contactEmail) {
     }
 
     public void checkForFeinError() throws Exception {
-        assertThat($(".errorInfo > ._15_fein").getText())
-                .contains(ORGANIZATION_FEIN_ERROR_MESSAGE);
+        checkForFormError("_15_fein", ORGANIZATION_FEIN_ERROR_MESSAGE);
     }
 }
diff --git a/...integration-tests/src/test/java/gov/medicaid/features/enrollment/ui/PersonalInfoPage.java b/...integration-tests/src/test/java/gov/medicaid/features/enrollment/ui/PersonalInfoPage.java
@@ -4,8 +4,6 @@
 
 import java.time.LocalDate;
 
-import static org.assertj.core.api.Assertions.assertThat;
-
 /**
  * PageObject to interact with the "Personal Info" step of an individual
  * provider enrollment. This page is reached by logging in, creating an
@@ -52,7 +50,6 @@ public void clickNext() {
     }
 
     public void checkForTooYoungError() throws Exception {
-        assertThat($(".errorInfo > ._02_dob").getText())
-                .contains(PROVIDER_TOO_YOUNG_ERROR_MESSAGE);
+        checkForFormError("_02_dob", PROVIDER_TOO_YOUNG_ERROR_MESSAGE);
     }
 }