test-rescue-agent 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 6f948ce1f72d8f7d993bc034a71551c3407dd312216f0c9b7e2037c30698a576
+  data.tar.gz: 5254ae06d79886f4809acf6dc3bcb33fa2365b84aa07ef98464f51f613843e30
+SHA512:
+  metadata.gz: ace19be132cbf1a16fe65ca102494a80c4905ca90482e1352a63ad424d80b5951e9cd2f62294baaae71da618811b7cdebdc82a865f9891eb349d32ec102088d2
+  data.tar.gz: a636c268910ea7e1526962870d338e1c539b8bed9a573ddef98d8057050c7c8c54119a9208575e59def4dec723187d56823beec809a0fe5238d9ead9f0520781

data/bin/test-rescue

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+require 'test-rescue-agent'
+TestRescueAgent::TestRunner.run

data/lib/test-rescue-agent.rb

@@ -0,0 +1,3 @@
+require 'test_rescue_agent/rspec_configuration'
+require 'test_rescue_agent/cucumber_configuration'
+require 'test_rescue_agent/test_runner'

data/lib/test_rescue_agent/cucumber_configuration.rb

@@ -0,0 +1,40 @@
+return unless ENV["SUITE_RUN_ID"] && ENV["TEST_RESCUE_TYPE"] == "cucumber"
+
+require 'test_rescue_agent/test_rescue_client'
+require 'cucumber'
+
+client = TestRescueAgent::TestRescueClient.new(
+  repository_id: ENV["REPOSITORY_ID"],
+  endpoint: ENV["TEST_RESCUE_ENDPOINT"],
+  secret: ENV["TEST_RESCUE_SECRET"],
+  api_key: ENV["TEST_RESCUE_API_KEY"]
+)
+
+at_exit do
+  client.complete_file_run(ENV["SUITE_RUN_ID"], ENV["FILE_RUN_ID"])
+end
+
+start = nil
+Before do
+  start = Time.now
+end
+
+After do |scenario|
+  failure_details = nil
+  if scenario.failed?
+    failure_details = [
+      scenario.exception.message.lines,
+      scenario.exception.backtrace
+    ].flatten.join("\n")
+  end
+  duration = Time.now - start
+  client.create_test_run(
+    ENV["SUITE_RUN_ID"],
+    ENV["FILE_RUN_ID"],
+    description: scenario.name,
+    location: scenario.location.to_s,
+    result: scenario.passed?,
+    failure_details: failure_details,
+    duration: duration
+  )
+end

data/lib/test_rescue_agent/rspec_configuration.rb

@@ -0,0 +1,62 @@
+return unless ENV["SUITE_RUN_ID"] && ENV["TEST_RESCUE_TYPE"] == "rspec"
+
+require 'test_rescue_agent/test_rescue_client'
+require 'rspec'
+
+RSpec.configure do |config|
+  def test_rescue_client
+    test_rescue_client = TestRescueAgent::TestRescueClient.new(
+      repository_id: ENV["REPOSITORY_ID"],
+      endpoint: ENV["TEST_RESCUE_ENDPOINT"],
+      secret: ENV["TEST_RESCUE_SECRET"],
+      api_key: ENV["TEST_RESCUE_API_KEY"]
+    )
+  end
+  config.color = true
+  config.tty = true
+
+  module TestRescueRSpecReporter
+    def notify_non_example_exception(exception, context_description)
+      example = RSpec::Core::Example.new(RSpec::Core::AnonymousExampleGroup, context_description, {})
+      presenter = RSpec::Core::Formatters::ExceptionPresenter.new(exception, example, indentation: 0)
+      failure_details = [presenter.colorized_message_lines, presenter.colorized_formatted_backtrace].flatten.join("\n")
+
+      test_rescue_client.create_file_run_exception(
+        ENV["SUITE_RUN_ID"],
+        ENV["FILE_RUN_ID"],
+        description: example.full_description,
+        location: example.location,
+        failure_details: failure_details,
+      )
+      super(exception, context_description)
+    end
+  end
+
+  module RSpec::Core
+    class Reporter
+      prepend TestRescueRSpecReporter
+    end
+  end
+
+  at_exit do
+    test_rescue_client.complete_file_run(ENV["SUITE_RUN_ID"], ENV["FILE_RUN_ID"])
+  end
+
+  config.after(:each) do |example|
+    failure_details = nil
+    if example.exception
+      presenter = RSpec::Core::Formatters::ExceptionPresenter.new(example.exception, example)
+      failure_details = [presenter.colorized_message_lines, presenter.colorized_formatted_backtrace].flatten.join("\n")
+    end
+    duration = Time.now - example.execution_result.started_at
+    test_rescue_client.create_test_run(
+      ENV["SUITE_RUN_ID"],
+      ENV["FILE_RUN_ID"],
+      description: example.full_description,
+      location: example.location,
+      result: !example.exception,
+      failure_details: failure_details,
+      duration: duration
+    )
+  end
+end

data/lib/test_rescue_agent/test_rescue_client.rb

@@ -0,0 +1,96 @@
+require 'test_rescue_agent/test_rescue_client/suite_run'
+require 'test_rescue_agent/test_rescue_client/file_run'
+require 'test_rescue_agent/test_rescue_client/file_run_exception'
+require 'test_rescue_agent/test_rescue_client/test_run'
+require 'httparty'
+begin
+  require 'webmock'
+rescue LoadError
+end
+
+module TestRescueAgent
+  class TestRescueClient
+    def initialize(options={})
+      @endpoint = options[:endpoint]
+      @repository_id = options[:repository_id]
+      @api_key = options[:api_key]
+      @secret = options[:secret]
+    end
+
+    def create_suite_run(sha:)
+      response = post('/suite_runs', suite_run: {
+        head_branch: 'master',
+        sha: sha,
+        title: 'Collect Pull Request Title from GitHub'
+      })
+      TestRescueClient::SuiteRun.new(self, JSON.parse(response.body))
+    end
+
+    def create_file_run(suite_run_id, attributes)
+      response = post("/suite_runs/#{suite_run_id}/file_runs", file_run: attributes)
+      TestRescueClient::FileRun.new(self, JSON.parse(response.body))
+    end
+
+    def create_file_run_exception(suite_run_id, file_run_id, attributes)
+      response = post("/suite_runs/#{suite_run_id}/file_runs/#{file_run_id}/exceptions", file_run_exception: attributes)
+      TestRescueClient::FileRunException.new(self, JSON.parse(response.body))
+    end
+
+    def claim_file_run(suite_run_id, container_id)
+      response = post("/suite_runs/#{suite_run_id}/file_runs/claim", container_id: container_id)
+      json = JSON.parse(response.body)
+      TestRescueClient::FileRun.new(self, json) if json
+    end
+
+    def complete_file_run(suite_run_id, file_run_id)
+      update_file_run(suite_run_id, file_run_id, {completed_at: Time.now.iso8601})
+    end
+
+    def update_file_run(suite_run_id, file_run_id, attributes)
+      patch("/suite_runs/#{suite_run_id}/file_runs/#{file_run_id}", file_run: attributes)
+    end
+
+    def create_test_run(suite_run_id, file_run_id, attributes)
+      response = post("/suite_runs/#{suite_run_id}/file_runs/#{file_run_id}/test_runs", test_run: attributes)
+      TestRescueClient::TestRun.new(self, JSON.parse(response.body))
+    end
+
+    private
+
+    def headers
+      {
+        'Content-Type' => 'application/json',
+        'Accept' => 'application/json',
+        'X-Api-Key' => @api_key,
+        'X-Api-Secret' => @secret
+      }
+    end
+
+    def handle_webmock
+      instance = WebMock::Config.instance
+      if instance.allow.nil?
+        instance.allow = 'www.testrescue.com'
+      elsif instance.allow.is_a?(Array)
+        instance.allow.push('www.testrescue.com') unless instance.allow.include?('www.testrescue.com')
+      else
+        WebMock::Config.instance.allow = [WebMock::Config.instance.allow, 'www.testrescue.com']
+      end
+    rescue NameError
+      # no problem. webmock not being resident means this code wasn't needed anyway
+    rescue
+      # TODO: report this to sentry, but still try to move on
+    end
+
+    def post(path, body)
+      handle_webmock
+      url = "#{@endpoint}/repositories/#{@repository_id}#{path}"
+      HTTParty.post(url, body: body.to_json, headers: headers)
+    end
+
+    def patch(path, body)
+      handle_webmock
+      url = "#{@endpoint}/repositories/#{@repository_id}#{path}"
+      HTTParty.patch(url, body: body.to_json, headers: headers)
+    end
+  end
+end

data/lib/test_rescue_agent/test_rescue_client/file_run.rb

@@ -0,0 +1,23 @@
+module TestRescueAgent
+  class TestRescueClient
+    class FileRun
+      attr_reader :id, :path, :description, :suite_run_id, :test_file_id, :created_at, :updated_at, :command
+
+      def initialize(client, attributes={})
+        @client = client
+        @id = attributes["id"]
+        @path = attributes["path"]
+        @description = attributes["description"]
+        @suite_run_id = attributes["suite_run_id"]
+        @test_file_id = attributes["test_file_id"]
+        @created_at = attributes["created_at"]
+        @updated_at = attributes["updated_at"]
+        @command = attributes["command"]
+      end
+
+      def create_test_run(attributes)
+        @client.create_test_run(suite_run_id, id, attributes)
+      end
+    end
+  end
+end

data/lib/test_rescue_agent/test_rescue_client/file_run_exception.rb

@@ -0,0 +1,17 @@
+module TestRescueAgent
+  class TestRescueClient
+    class FileRunException
+      attr_reader :id, :suite_run_id, :file_run_id, :description, :location, :failure_details
+
+      def initialize(client, attributes={})
+        @client = client
+        @id = attributes["id"]
+        @suite_run_id = attributes["suite_run_id"]
+        @file_run_id = attributes["file_run_id"]
+        @description = attributes["description"]
+        @location = attributes["location"]
+        @failure_details = attributes["failure_details"]
+      end
+    end
+  end
+end

data/lib/test_rescue_agent/test_rescue_client/suite_run.rb

@@ -0,0 +1,19 @@
+module TestRescueAgent
+  class TestRescueClient
+    class SuiteRun
+      attr_reader :id, :repository_id, :created_at, :updated_at
+
+      def initialize(client, attributes={})
+        @client = client
+        @id = attributes["id"]
+        @repository_id = attributes["repository_id"]
+        @created_at = attributes["created_at"]
+        @updated_at = attributes["updated_at"]
+      end
+
+      def create_file_run(attributes)
+        @client.create_file_run(id, attributes)
+      end
+    end
+  end
+end

data/lib/test_rescue_agent/test_rescue_client/test_run.rb

@@ -0,0 +1,20 @@
+module TestRescueAgent
+  class TestRescueClient
+    class TestRun
+      attr_reader :id, :file_run_id, :test_id, :description, :location, :result, :failure_details, :created_at, :updated_at
+
+      def initialize(client, attributes={})
+        @client = client
+        @id = attributes["id"]
+        @file_run_id = attributes["file_run_id"]
+        @test_id = attributes["test_id"]
+        @description = attributes["description"]
+        @location = attributes["location"]
+        @result = attributes["result"]
+        @failure_details = attributes["failure_details"]
+        @created_at = attributes["created_at"]
+        @updated_at = attributes["updated_at"]
+      end
+    end
+  end
+end

data/lib/test_rescue_agent/test_runner.rb

@@ -0,0 +1,18 @@
+require 'test_rescue_agent/test_rescue_client'
+
+module TestRescueAgent
+  class TestRunner
+    def self.run
+      client = TestRescueAgent::TestRescueClient.new(
+        repository_id: ENV["REPOSITORY_ID"],
+        endpoint: ENV["TEST_RESCUE_ENDPOINT"],
+        secret: ENV["TEST_RESCUE_SECRET"],
+        api_key: ENV["TEST_RESCUE_API_KEY"]
+      )
+      while file_run = client.claim_file_run(ENV["SUITE_RUN_ID"], ENV["CONTAINER_ID"])
+        puts "running \"#{file_run.command}\""
+        `#{file_run.command}`
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: test-rescue-agent
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Ryan Fogle
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-04-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: |
+  Test Rescue is a massively parallelized continuous testing platform,
+  (http://www.testrescue.com). Test Rescue provides you with large scale
+  Test automation execution performance by parallelizing each individual
+  test in its own container automatically provisioned and run in the cloud.
+  Results are provided live as your tests complete.
+  Test Rescue provides you with deep information about the performance
+  of your test suite as it runs. The Test Rescue Agent is a Gem, hosted on
+  https://github.com/fogle/test-rescue-agent/
+email: support@testrescue.com
+executables:
+- test-rescue
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/test-rescue
+- lib/test-rescue-agent.rb
+- lib/test_rescue_agent/cucumber_configuration.rb
+- lib/test_rescue_agent/rspec_configuration.rb
+- lib/test_rescue_agent/test_rescue_client.rb
+- lib/test_rescue_agent/test_rescue_client/file_run.rb
+- lib/test_rescue_agent/test_rescue_client/file_run_exception.rb
+- lib/test_rescue_agent/test_rescue_client/suite_run.rb
+- lib/test_rescue_agent/test_rescue_client/test_run.rb
+- lib/test_rescue_agent/test_runner.rb
+homepage: https://github.com/fogle/test-rescue-agent
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">"
+    - !ruby/object:Gem::Version
+      version: 1.3.1
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Test Rescue Ruby Agent
+test_files: []