fastlane regression testing

require "json"
require "net/http"
require "uri"

QATECH_API = "https://api.qa.tech/v1"

def qatech_request(method, path, body = nil)
  uri = URI("#{QATECH_API}#{path}")
  request = (method == :post ? Net::HTTP::Post : Net::HTTP::Get).new(uri)
  request["Authorization"] = "Bearer #{ENV.fetch('QATECH_API_TOKEN')}"
  request["Content-Type"] = "application/json"
  if body
    # Org-scoped API keys need the project on every request.
    body = body.merge(projectShortId: ENV["QATECH_PROJECT_ID"]) if ENV["QATECH_PROJECT_ID"]
    request.body = body.to_json
  end
  response = Net::HTTP.start(uri.host, uri.port, use_ssl: true) { |http| http.request(request) }
  UI.user_error!("QA.tech #{path} failed (#{response.code}): #{response.body}") unless response.is_a?(Net::HTTPSuccess)
  JSON.parse(response.body)
end

# Uploads a build and returns its applicationBuildShortId.
def qatech_upload_build(file:, platform:)
  app_id = ENV.fetch("QATECH_APP_ID")
  upload = qatech_request(:post, "/applications/#{app_id}/builds/upload-url", { fileName: File.basename(file) })

  uri = URI(upload["uploadUrl"])
  File.open(file, "rb") do |io|
    put = Net::HTTP::Put.new(uri)
    put["Content-Type"] = "application/octet-stream"
    put["Content-Length"] = File.size(file).to_s
    put.body_stream = io
    response = Net::HTTP.start(uri.host, uri.port, use_ssl: true) { |http| http.request(put) }
    UI.user_error!("Build upload failed (#{response.code})") unless response.is_a?(Net::HTTPSuccess)
  end

  build = qatech_request(:post, "/applications/#{app_id}/builds", { platform: platform, buildToken: upload["buildToken"] })
  UI.success("Uploaded #{build['applicationBuildShortId']}")
  build["applicationBuildShortId"]
end

# Starts the test plan on the build and fails the lane unless the run passes.
def qatech_run_test_plan(build_short_id:, wait: true)
  run = qatech_request(:post, "/run", {
    testPlanShortId: ENV.fetch("QATECH_TEST_PLAN_ID"),
    applications: [{
      applicationShortId: ENV.fetch("QATECH_APP_ID"),
      environment: { applicationBuildShortId: build_short_id }
    }]
  })
  run_id = run["run"]["shortId"]
  UI.message("Started QA.tech run #{run_id}: #{run['run']['url']}")
  return run_id unless wait

  loop do
    status = qatech_request(:get, "/run/#{run_id}")
    if %w[COMPLETED ERROR CANCELLED].include?(status["status"])
      UI.user_error!("QA.tech run #{run_id} finished with #{status['result']}") unless status["result"] == "PASSED"
      UI.success("QA.tech run #{run_id} passed")
      return run_id
    end
    sleep(30)
  end
end

platform :android do
  lane :qatech do
    gradle(task: "assemble", build_type: "Debug")
    build_id = qatech_upload_build(
      file: lane_context[SharedValues::GRADLE_APK_OUTPUT_PATH],
      platform: "android"
    )
    qatech_run_test_plan(build_short_id: build_id)
  end
end

platform :ios do
  lane :qatech do
    root = File.expand_path("..", Dir.pwd) # lanes run from fastlane/
    derived_data = File.join(root, "build")

    # Simulator build: device and App Store .ipa files cannot run on simulators.
    sh("xcodebuild", "-workspace", File.join(root, "MyApp.xcworkspace"),
       "-scheme", "MyApp", "-configuration", "Debug",
       "-sdk", "iphonesimulator", "-destination", "generic/platform=iOS Simulator",
       "-derivedDataPath", derived_data, "build")

    products = File.join(derived_data, "Build/Products/Debug-iphonesimulator")
    sh("cd '#{products}' && zip -qr MyApp.zip MyApp.app")

    build_id = qatech_upload_build(file: File.join(products, "MyApp.zip"), platform: "ios")
    qatech_run_test_plan(build_short_id: build_id)
  end
end