Polish

samples/javaconfig/users/src/integration-test/java/sample/pages/BasePage.java

@@ -0,0 +1,74 @@
+/*
+ * Copyright 2014-2017 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package sample.pages;
+
+import org.openqa.selenium.By;
+import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.WebElement;
+import org.openqa.selenium.support.PageFactory;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+/**
+ * @author Pool Dolorier
+ */
+public abstract class BasePage {
+
+	private WebDriver driver;
+
+	public BasePage(WebDriver driver) {
+		this.driver = driver;
+	}
+
+	public WebDriver getDriver() {
+		return this.driver;
+	}
+
+	public static void get(WebDriver driver, String get) {
+		String baseUrl = "http://localhost:" + System.getProperty("tomcat.port");
+		driver.get(baseUrl + get);
+	}
+
+	public static void getFrom(WebDriver driver, String resourceUrl) {
+		driver.get(resourceUrl);
+	}
+
+	public HomePage home(WebDriver driver, String resourceUrl) {
+		getFrom(driver, resourceUrl);
+		return PageFactory.initElements(driver, HomePage.class);
+	}
+
+	public LinkPage linkPage(WebDriver driver, String resourceUrl) {
+		getFrom(driver, resourceUrl);
+		return PageFactory.initElements(driver, LinkPage.class);
+	}
+
+	public String getSwitchElementId(String user) {
+		return "switchAccount" + user;
+	}
+
+	public WebElement getElementById(String id) {
+		return this.driver.findElement(By.id(id));
+	}
+
+	public String getContentAttributeByElementId(String id, String attribute) {
+		WebElement element = getElementById(id);
+		assertThat(element.getAttribute(attribute)).isNotEmpty();
+		return element.getAttribute(attribute);
+	}
+
+}