io.github.carousell.aws.UploadTestPackage Maven / Gradle / Ivy

Go to download
package io.github.carousell.aws;

import com.amazonaws.services.devicefarm.model.AWSDeviceFarmException;
import com.amazonaws.services.devicefarm.model.UploadType;
import com.amazonaws.services.dynamodbv2.model.AttributeValue;
import com.amazonaws.services.dynamodbv2.model.UpdateItemRequest;
import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;
import org.apache.maven.project.MavenProject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/** Mojo for uploading test packages to AWS Device Farm */
@Mojo(name = "uploadTestPackage")
public class UploadTestPackage extends AbstractMojo {

  private static final Logger LOG = LoggerFactory.getLogger(UploadTestPackage.class);

  @Parameter(defaultValue = "${project}", readonly = true, required = true)
  private MavenProject project;

  @Parameter(required = true)
  private String awsDeviceFarmAccessKey;

  @Parameter(required = true)
  private String awsDeviceFarmSecretKey;

  @Parameter(required = true)
  private String awsDynamoAccessKey;

  @Parameter(required = true)
  private String awsDynamoSecretKey;

  @Parameter(defaultValue = "us-east-1")
  private String awsRegion;

  @Parameter(required = true)
  private String projectArn;

  @Parameter(required = true)
  private String testPackage;

  @Parameter(required = true)
  private String testPackageName;

  @Parameter(defaultValue = "APPIUM_JAVA_JUNIT_TEST_PACKAGE")
  private String uploadType;

  @Parameter(required = true)
  private String scenarioIds;

  public void execute() throws MojoExecutionException, MojoFailureException {
    try {
      AWSDeviceFarmService awsDeviceFarmService =
          new AWSDeviceFarmService(awsDeviceFarmAccessKey, awsDeviceFarmSecretKey, awsRegion);

      // upload test package
      LOG.info("Uploading test package from {}", testPackage);
      String testArn =
          awsDeviceFarmService
              .upload(new File(testPackage), projectArn, UploadType.valueOf(uploadType), true)
              .getArn();

      LOG.info("Setting testArn {} as Maven property", testArn);
      project.getProperties().put("testArn", testArn);

      LOG.info("Publishing testArn {} to DynamoDB", testArn);
      updateDynamoValue(testPackageName, testArn, scenarioIds);

    } catch (AWSDeviceFarmException | InterruptedException | IOException exception) {
      LOG.error("Error while triggering AWS test", exception);
    }
  }

  private void updateDynamoValue(String key, String arn, String scenarioIds) {
    AWSDynamoService awsDynamoService =
        new AWSDynamoService(awsDynamoAccessKey, awsDynamoSecretKey, awsRegion);
    HashMap keyMap = new HashMap();
    keyMap.put("key", new AttributeValue().withS(key));

    Map expressionAttributeValues = new HashMap();
    expressionAttributeValues.put(":arn", new AttributeValue().withS(arn));
    expressionAttributeValues.put(":scenarioIds", new AttributeValue().withS(scenarioIds));

    UpdateItemRequest updateItemRequest =
        new UpdateItemRequest()
            .withTableName("testPackages")
            .withKey(keyMap)
            .withUpdateExpression("set testPackageArn = :arn, scenarioIds = :scenarioIds")
            .withExpressionAttributeValues(expressionAttributeValues);

    awsDynamoService.getAwsDynamo().updateItem(updateItemRequest);
  }
}