Contents

module KnapsackPro
  class Allocator
    def initialize(args)
      @test_files = args.fetch(:test_files)
      @ci_node_total = args.fetch(:ci_node_total)
      @ci_node_index = args.fetch(:ci_node_index)
      @repository_adapter = args.fetch(:repository_adapter)
    end

    def test_file_paths
      connection = KnapsackPro::Client::Connection.new(build_action)
      response = connection.call
      if connection.success?
        raise ArgumentError.new(response) if connection.errors?
        prepare_test_files(response)
      else
        fallback_test_files
      end
    end

    private

    attr_reader :test_files,
      :ci_node_total,
      :ci_node_index,
      :repository_adapter

    def encrypted_test_files
      KnapsackPro::Crypto::Encryptor.call(test_files)
    end

    def encrypted_branch
      KnapsackPro::Crypto::BranchEncryptor.call(repository_adapter.branch)
    end

    def build_action
      KnapsackPro::Client::API::V1::BuildDistributions.subset(
        commit_hash: repository_adapter.commit_hash,
        branch: encrypted_branch,
        node_total: ci_node_total,
        node_index: ci_node_index,
        test_files: encrypted_test_files,
      )
    end

    def prepare_test_files(response)
      decrypted_test_files = KnapsackPro::Crypto::Decryptor.call(test_files, response['test_files'])
      KnapsackPro::TestFilePresenter.paths(decrypted_test_files)
    end

    def fallback_test_files
      test_flat_distributor = KnapsackPro::TestFlatDistributor.new(test_files, ci_node_total)
      test_files_for_node_index = test_flat_distributor.test_files_for_node(ci_node_index)
      KnapsackPro::TestFilePresenter.paths(test_files_for_node_index)
    end
  end
end

Version data entries

16 entries across 16 versions & 1 rubygems

Version	Path
knapsack_pro-0.47.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.46.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.45.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.44.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.43.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.42.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.41.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.40.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.39.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.38.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.37.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.36.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.35.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.34.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.33.0	lib/knapsack_pro/allocator.rb
knapsack_pro-0.32.0	lib/knapsack_pro/allocator.rb