knapsack_pro 0.0.1 → 0.1.0

data/lib/knapsack_pro/client/connection.rb

@@ -0,0 +1,94 @@
+module KnapsackPro
+  module Client
+    class Connection
+      TIMEOUT = 15
+
+      def initialize(action)
+        @action = action
+      end
+
+      def call
+        send(action.http_method)
+      end
+
+      def success?
+        !!response
+      end
+
+      def errors?
+        !!(response && (response['errors'] || response['error']))
+      end
+
+      private
+
+      attr_reader :action, :response
+
+      def logger
+        KnapsackPro.logger
+      end
+
+      def endpoint
+        KnapsackPro::Config::Env.endpoint
+      end
+
+      def endpoint_url
+        endpoint + action.endpoint_path
+      end
+
+      def request_hash
+        action
+        .request_hash
+        .merge({
+          test_suite_token: test_suite_token
+        })
+      end
+
+      def request_body
+        request_hash.to_json
+      end
+
+      def test_suite_token
+        KnapsackPro::Config::Env.test_suite_token
+      end
+
+      def json_headers
+        {
+          'Content-Type' => 'application/json',
+          'Accept' => 'application/json'
+        }
+      end
+
+      def parse_response(body)
+        return '' if body == '' || body.nil?
+        JSON.parse(body)
+      rescue JSON::ParserError
+        nil
+      end
+
+      def post
+        uri = URI.parse(endpoint_url)
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = (uri.scheme == 'https')
+        http.open_timeout = TIMEOUT
+        http.read_timeout = TIMEOUT
+
+        http_response = http.post(uri.path, request_body, json_headers)
+        @response = parse_response(http_response.body)
+
+        request_uuid = http_response.header['X-Request-Id']
+
+        logger.info("API request UUID: #{request_uuid}")
+        logger.info('API response:')
+        if errors?
+          logger.error(response)
+        else
+          logger.info(response)
+        end
+
+        response
+      rescue Errno::ECONNREFUSED, EOFError, Net::OpenTimeout, Net::ReadTimeout => e
+        logger.warn(e.inspect)
+      end
+    end
+  end
+end

data/lib/knapsack_pro/config/ci/base.rb

@@ -0,0 +1,22 @@
+module KnapsackPro
+  module Config
+    module CI
+      class Base
+        def node_total
+        end
+
+        def node_index
+        end
+
+        def commit_hash
+        end
+
+        def branch
+        end
+
+        def project_dir
+        end
+      end
+    end
+  end
+end

data/lib/knapsack_pro/config/ci/buildkite.rb

@@ -0,0 +1,27 @@
+module KnapsackPro
+  module Config
+    module CI
+      class Buildkite < Base
+        def node_total
+          ENV['BUILDKITE_PARALLEL_JOB_COUNT']
+        end
+
+        def node_index
+          ENV['BUILDKITE_PARALLEL_JOB']
+        end
+
+        def commit_hash
+          ENV['BUILDKITE_COMMIT']
+        end
+
+        def branch
+          ENV['BUILDKITE_BRANCH']
+        end
+
+        def project_dir
+          ENV['BUILDKITE_BUILD_CHECKOUT_PATH']
+        end
+      end
+    end
+  end
+end

data/lib/knapsack_pro/config/ci/circle.rb

@@ -0,0 +1,29 @@
+module KnapsackPro
+  module Config
+    module CI
+      class Circle < Base
+        def node_total
+          ENV['CIRCLE_NODE_TOTAL']
+        end
+
+        def node_index
+          ENV['CIRCLE_NODE_INDEX']
+        end
+
+        def commit_hash
+          ENV['CIRCLE_SHA1']
+        end
+
+        def branch
+          ENV['CIRCLE_BRANCH']
+        end
+
+        def project_dir
+          project_repo_name = ENV['CIRCLE_PROJECT_REPONAME']
+          "/home/ubuntu/#{project_repo_name}" if project_repo_name
+        end
+      end
+    end
+  end
+end
+

data/lib/knapsack_pro/config/ci/semaphore.rb

@@ -0,0 +1,28 @@
+module KnapsackPro
+  module Config
+    module CI
+      class Semaphore < Base
+        def node_total
+          ENV['SEMAPHORE_THREAD_COUNT']
+        end
+
+        def node_index
+          index = ENV['SEMAPHORE_CURRENT_THREAD']
+          index.to_i - 1 if index
+        end
+
+        def commit_hash
+          ENV['REVISION']
+        end
+
+        def branch
+          ENV['BRANCH_NAME']
+        end
+
+        def project_dir
+          ENV['SEMAPHORE_PROJECT_DIR']
+        end
+      end
+    end
+  end
+end

data/lib/knapsack_pro/config/env.rb

@@ -0,0 +1,111 @@
+module KnapsackPro
+  module Config
+    class Env
+      class << self
+        def ci_node_total
+          (ENV['KNAPSACK_PRO_CI_NODE_TOTAL'] ||
+            ci_env_for(:node_total) ||
+            1).to_i
+        end
+
+        def ci_node_index
+          (ENV['KNAPSACK_PRO_CI_NODE_INDEX'] ||
+            ci_env_for(:node_index) ||
+            0).to_i
+        end
+
+        def commit_hash
+          ENV['KNAPSACK_PRO_COMMIT_HASH'] ||
+            ci_env_for(:commit_hash)
+        end
+
+        def branch
+          ENV['KNAPSACK_PRO_BRANCH'] ||
+            ci_env_for(:branch)
+        end
+
+        def project_dir
+          ENV['KNAPSACK_PRO_PROJECT_DIR'] ||
+            ci_env_for(:project_dir)
+        end
+
+        def test_file_pattern
+          ENV['KNAPSACK_PRO_TEST_FILE_PATTERN']
+        end
+
+        def repository_adapter
+          ENV['KNAPSACK_PRO_REPOSITORY_ADAPTER']
+        end
+
+        def recording_enabled
+          ENV['KNAPSACK_PRO_RECORDING_ENABLED']
+        end
+
+        def recording_enabled?
+          recording_enabled == 'true'
+        end
+
+        def endpoint
+          env_name = 'KNAPSACK_PRO_ENDPOINT'
+          return ENV[env_name] if ENV[env_name]
+
+          case mode
+          when :development
+            'http://api.knapsackpro.dev:3000'
+          when :test
+            'http://api-staging.knapsackpro.com'
+          when :production
+            'http://api.knapsackpro.com'
+          else
+            required_env(env_name)
+          end
+        end
+
+        def test_suite_token
+          required_env('KNAPSACK_PRO_TEST_SUITE_TOKEN')
+        end
+
+        def test_suite_token_rspec
+          ENV['KNAPSACK_PRO_TEST_SUITE_TOKEN_RSPEC']
+        end
+
+        def test_suite_token_minitest
+          ENV['KNAPSACK_PRO_TEST_SUITE_TOKEN_MINITEST']
+        end
+
+        def test_suite_token_cucumber
+          ENV['KNAPSACK_PRO_TEST_SUITE_TOKEN_CUCUMBER']
+        end
+
+        def mode
+          mode = ENV['KNAPSACK_PRO_MODE']
+          return :production if mode.nil?
+          mode = mode.to_sym
+          if [:development, :test, :production].include?(mode)
+            mode
+          else
+            raise ArgumentError.new('Wrong mode name')
+          end
+        end
+
+        def ci_env_for(env_name)
+          value = nil
+          ci_list = KnapsackPro::Config::CI.constants - [:Base]
+          ci_list.each do |ci_name|
+            ci_class = Object.const_get("KnapsackPro::Config::CI::#{ci_name}")
+            ci = ci_class.new
+            value = ci.send(env_name)
+            break unless value.nil?
+          end
+          value
+        end
+
+        private
+
+        def required_env(env_name)
+          ENV[env_name] || raise("Missing environment variable #{env_name}")
+        end
+      end
+    end
+  end
+end

data/lib/knapsack_pro/logger_wrapper.rb

@@ -0,0 +1,15 @@
+module KnapsackPro
+  class LoggerWrapper
+    def initialize(logger)
+      @logger = ::ActiveSupport::TaggedLogging.new(logger)
+    end
+
+    private
+
+    attr_reader :logger
+
+    def method_missing(m, *args, &block)
+      logger.tagged('knapsack_pro') { logger.send(m, *args, &block) }
+    end
+  end
+end

data/lib/knapsack_pro/presenter.rb

@@ -0,0 +1,25 @@
+module KnapsackPro
+  class Presenter
+    class << self
+      def global_time
+        global_time = pretty_seconds(KnapsackPro.tracker.global_time)
+        "Global time execution for tests: #{global_time}"
+      end
+
+      def pretty_seconds(seconds)
+        sign = ''
+
+        if seconds < 0
+          seconds = seconds*-1
+          sign = '-'
+        end
+
+        return "#{sign}#{seconds}s" if seconds.abs < 1
+
+        time = Time.at(seconds).gmtime.strftime('%Hh %Mm %Ss')
+        time_without_zeros = time.gsub(/00(h|m|s)/, '').strip
+        sign + time_without_zeros
+      end
+    end
+  end
+end

data/lib/knapsack_pro/report.rb

@@ -0,0 +1,20 @@
+module KnapsackPro
+  class Report
+    def self.save
+      repository_adapter = KnapsackPro::RepositoryAdapterInitiator.call
+      action = KnapsackPro::Client::API::V1::BuildSubsets.create(
+        commit_hash: repository_adapter.commit_hash,
+        branch: repository_adapter.branch,
+        node_total: KnapsackPro::Config::Env.ci_node_total,
+        node_index: KnapsackPro::Config::Env.ci_node_index,
+        test_files: KnapsackPro.tracker.to_a,
+      )
+      connection = KnapsackPro::Client::Connection.new(action)
+      response = connection.call
+      if connection.success?
+        raise ArgumentError.new(response) if connection.errors?
+        KnapsackPro.logger.info('Saved time execution report on API server.')
+      end
+    end
+  end
+end

data/lib/knapsack_pro/repository_adapter_initiator.rb

@@ -0,0 +1,12 @@
+module KnapsackPro
+  class RepositoryAdapterInitiator
+    def self.call
+      case KnapsackPro::Config::Env.repository_adapter
+      when 'git'
+        KnapsackPro::RepositoryAdapters::GitAdapter.new
+      else
+        KnapsackPro::RepositoryAdapters::EnvAdapter.new
+      end
+    end
+  end
+end

data/lib/knapsack_pro/repository_adapters/base_adapter.rb

@@ -0,0 +1,14 @@
+module KnapsackPro
+  module RepositoryAdapters
+    class BaseAdapter
+      def commit_hash
+        raise NotImplementedError
+      end
+
+      def branch
+        raise NotImplementedError
+      end
+    end
+  end
+end
+

data/lib/knapsack_pro/repository_adapters/env_adapter.rb

@@ -0,0 +1,13 @@
+module KnapsackPro
+  module RepositoryAdapters
+    class EnvAdapter < BaseAdapter
+      def commit_hash
+        KnapsackPro::Config::Env.commit_hash
+      end
+
+      def branch
+        KnapsackPro::Config::Env.branch
+      end
+    end
+  end
+end

data/lib/knapsack_pro/repository_adapters/git_adapter.rb

@@ -0,0 +1,19 @@
+module KnapsackPro
+  module RepositoryAdapters
+    class GitAdapter < BaseAdapter
+      def commit_hash
+        `git -C "#{working_dir}" rev-parse HEAD`.strip
+      end
+
+      def branch
+        `git -C "#{working_dir}" rev-parse --abbrev-ref HEAD`.strip
+      end
+
+      private
+
+      def working_dir
+        KnapsackPro::Config::Env.project_dir
+      end
+    end
+  end
+end