Java tutorial
/*************************************************************************** Copyright (c) 2016, EPAM SYSTEMS INC 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 com.epam.dlab.automation.jenkins; import com.epam.dlab.automation.exceptions.JenkinsException; import com.epam.dlab.automation.helper.ConfigPropertyValue; import com.epam.dlab.automation.helper.NamingHelper; import com.epam.dlab.automation.http.HttpStatusCode; import com.jayway.restassured.RestAssured; import com.jayway.restassured.authentication.FormAuthConfig; import com.jayway.restassured.http.ContentType; import com.jayway.restassured.response.Response; import com.jayway.restassured.specification.RequestSpecification; import org.apache.commons.codec.binary.Base64; import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.Logger; import org.testng.Assert; import java.time.Duration; import java.util.regex.Matcher; import java.util.regex.Pattern; import static com.jayway.restassured.RestAssured.given; public class JenkinsService { private static final Logger LOGGER = LogManager.getLogger(JenkinsService.class); private final String awsAccessKeyId; private final String awsSecretAccessKey; private String ssnURL; private String serviceBaseName; private FormAuthConfig config = new FormAuthConfig(JenkinsConfigProperties.JENKINS_JOB_NAME_SEARCH, "username", "password"); public JenkinsService() { if (!ConfigPropertyValue.isUseJenkins()) { ssnURL = ConfigPropertyValue.getSsnUrl(); serviceBaseName = ConfigPropertyValue.getServiceBaseName(); } awsAccessKeyId = convertToParam(ConfigPropertyValue.getAwsAccessKeyId()); awsSecretAccessKey = convertToParam(ConfigPropertyValue.getAwsSecretAccessKey()); } private String convertToParam(String s) { s = s.replaceAll("/", "%2F"); return s; } public String getSsnURL() { return ssnURL; } public String getServiceBaseName() { return serviceBaseName; } private String getQueueStatus() { return getWhen(ContentType.XML).get(JenkinsUrls.API).getBody().xmlPath() .getString(JenkinsResponseElements.IN_QUEUE_ELEMENT); } private void waitForJenkinsStartup(Duration duration) throws InterruptedException { String actualStatus; long timeout = duration.toMillis(); long expiredTime = System.currentTimeMillis() + timeout; while ((actualStatus = getQueueStatus()).endsWith(JenkinsConfigProperties.SUCCESS_STATUS)) { Thread.sleep(JenkinsConfigProperties.JENKINS_REQUEST_TIMEOUT); if (timeout != 0 && expiredTime < System.currentTimeMillis()) { actualStatus = getQueueStatus(); break; } } if (actualStatus.endsWith(JenkinsConfigProperties.SUCCESS_STATUS)) { LOGGER.info("ERROR: Timeout has been expired for Jenkins"); LOGGER.info(" timeout is {}"); } } public String runJenkinsJob(String jenkinsJobURL) throws InterruptedException { if (!ConfigPropertyValue.isUseJenkins()) { return ConfigPropertyValue.getJenkinsBuildNumber(); } baseUriInitialize(jenkinsJobURL); String dateAsString = NamingHelper.generateRandomValue(); Response responsePostJob = getWhen(ContentType.URLENC) .body(String.format(JenkinsConfigProperties.jenkinsJobStartBody, awsAccessKeyId, awsSecretAccessKey, dateAsString, ConfigPropertyValue.getClusterOsUser(), ConfigPropertyValue.getClusterOsFamily(), awsAccessKeyId, awsSecretAccessKey, dateAsString, ConfigPropertyValue.getClusterOsUser(), ConfigPropertyValue.getClusterOsFamily())) .post(jenkinsJobURL + "build"); Assert.assertEquals(responsePostJob.statusCode(), HttpStatusCode.OK); waitForJenkinsStartup(ConfigPropertyValue.getTimeoutJenkinsAutotest()); setBuildNumber(); checkBuildResult(); setJenkinsURLServiceBaseName(); return ConfigPropertyValue.getJenkinsBuildNumber(); } public String getJenkinsJob() throws InterruptedException { if (!ConfigPropertyValue.isUseJenkins()) { return ConfigPropertyValue.getJenkinsBuildNumber(); } baseUriInitialize(ConfigPropertyValue.getJenkinsJobURL()); setBuildNumber(); checkBuildResult(); setJenkinsURLServiceBaseName(); return ConfigPropertyValue.getJenkinsBuildNumber(); } private static void baseUriInitialize(String value) { RestAssured.baseURI = value; } private void setBuildNumber() { if (ConfigPropertyValue.getJenkinsBuildNumber() != null) { LOGGER.info("Jenkins build number is {}", ConfigPropertyValue.getJenkinsBuildNumber()); return; } String buildName = getWhen(ContentType.URLENC).get(JenkinsUrls.LAST_BUILD).getBody().htmlPath() .getString(JenkinsResponseElements.HTML_TITLE); Pattern pattern = Pattern.compile("\\s#\\d+(?!\\d+)\\s"); Matcher matcher = pattern.matcher(buildName); if (matcher.find()) { ConfigPropertyValue.setJenkinsBuildNumber(matcher.group().substring(2).trim()); } else { throw new JenkinsException("Jenkins job was failed. There is no buildNumber"); } LOGGER.info("Jenkins build number is {}", ConfigPropertyValue.getJenkinsBuildNumber()); } private void checkBuildResult() throws InterruptedException { String buildResult; long timeout = ConfigPropertyValue.getTimeoutJenkinsAutotest().toMillis(); long expiredTime = System.currentTimeMillis() + timeout; do { buildResult = getWhen(ContentType.JSON) .get(ConfigPropertyValue.getJenkinsBuildNumber() + JenkinsUrls.JSON_PRETTY).getBody().jsonPath() .getString(JenkinsResponseElements.RESULT); if (buildResult == null) { if (timeout != 0 && expiredTime < System.currentTimeMillis()) { throw new JenkinsException("Timeout has been expired for Jenkins build. Timeout is " + ConfigPropertyValue.getTimeoutJenkinsAutotest()); } Thread.sleep(JenkinsConfigProperties.JENKINS_REQUEST_TIMEOUT); } } while (buildResult == null); if (!buildResult.equals("SUCCESS")) { throw new JenkinsException("Jenkins job was failed. Build result is not success"); } } private void setJenkinsURLServiceBaseName() { String jenkinsHoleURL = getWhen(ContentType.TEXT) .get(ConfigPropertyValue.getJenkinsBuildNumber() + JenkinsUrls.LOG_TEXT).getBody().prettyPrint(); Pattern pattern = Pattern.compile("Jenkins URL:(.+)"); Matcher matcher = pattern.matcher(jenkinsHoleURL); if (matcher.find()) { ssnURL = matcher.group(1).replaceAll("/jenkins", ""); } pattern = Pattern.compile("Service base name:(.+)"); matcher = pattern.matcher(jenkinsHoleURL); if (matcher.find()) { serviceBaseName = matcher.group(1); } else { throw new JenkinsException("SSN URL in Jenkins job not found"); } } private RequestSpecification getWhen(ContentType contentType) { return given() .header(JenkinsConfigProperties.AUTHORIZATION, String.format( JenkinsConfigProperties.AUTHORIZATION_KEY, base64CredentialDecode(ConfigPropertyValue.get(ConfigPropertyValue.JENKINS_USERNAME), ConfigPropertyValue.get(ConfigPropertyValue.JENKINS_PASS)))) .auth() .form(ConfigPropertyValue.getJenkinsUsername(), ConfigPropertyValue.getJenkinsPassword(), config) .contentType(contentType).when(); } private static String base64CredentialDecode(String user, String password) { byte[] bytesEncoded = Base64.encodeBase64(String.format("%s:%s", user, password).getBytes()); return new String(bytesEncoded); } }