newrelic_rpm 3.9.3.241 → 3.9.4.245

data/test/flaky_proxy/lib/flaky_proxy/proxy.rb

@@ -1,58 +0,0 @@
-# encoding: utf-8
-# This file is distributed under New Relic's license terms.
-# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
-
-require 'socket'
-
-module FlakyProxy
-  class Proxy
-    def initialize(options)
-      @listen_host = options.fetch(:listen_host)
-      @listen_port = options.fetch(:listen_port)
-      @backend_server = Server.new(options.fetch(:target_host), options.fetch(:target_port))
-      @rules = RuleSet.build('')
-      @rules_path = options[:rules_path]
-      @listen_socket = nil
-      reload_rules_file
-    end
-
-    def service_connection(client_socket)
-      with_connection_logging(client_socket) do
-        Connection.new(client_socket, @backend_server, @rules).service
-      end
-    rescue => e
-      FlakyProxy.logger.error("Error servicing connection: #{e}, from #{e.backtrace.join("\n")}")
-    end
-
-    def reload_rules_file
-      if @rules_path
-        mtime = File.stat(@rules_path).mtime
-        if @last_rules_mtime.nil? || mtime > @last_rules_mtime
-          FlakyProxy.logger.info("Reloading rules file at #{@rules_path}")
-          @rules = RuleSet.build(File.read(@rules_path))
-          @last_rules_mtime = mtime
-        end
-      end
-    rescue => e
-      FlakyProxy.logger.error("Error reloading rules file at #{@rules_path}: #{e}\n#{e.backtrace.join("\n")}")
-    end
-
-    def with_connection_logging(client_socket)
-      peer_info = client_socket.peeraddr(:hostname)
-      client_str = "#{peer_info[2]}:#{peer_info[1]}"
-      FlakyProxy.logger.info("Accepted connection from #{client_str}")
-      yield
-      FlakyProxy.logger.info("Finished servicing connection from #{client_str}")
-    end
-
-    def run
-      FlakyProxy.logger.info("Starting FlakyProxy on #{@listen_host}:#{@listen_port} -> #{@backend_server.to_s}")
-      @listen_socket = TCPServer.new(@listen_host, @listen_port)
-      loop do
-        client_socket = @listen_socket.accept
-        service_connection(client_socket)
-        reload_rules_file
-      end
-    end
-  end
-end

data/test/flaky_proxy/lib/flaky_proxy/rule.rb

@@ -1,72 +0,0 @@
-# encoding: utf-8
-# This file is distributed under New Relic's license terms.
-# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
-
-module FlakyProxy
-  class Rule
-    class ActionBuilder
-      attr_reader :actions
-
-      def initialize
-        @actions = []
-      end
-
-      def pass
-        @actions << [:pass]
-      end
-
-      def respond(response_spec)
-        @actions << [:respond, response_spec]
-      end
-
-      def delay(amount)
-        @actions << [:delay, amount]
-      end
-
-      def close
-        @actions << [:close]
-      end
-    end
-
-    def initialize(url_regex=nil, builder=nil, &blk)
-      @url_regex = url_regex
-      @action = blk
-      @builder = builder || ActionBuilder.new
-      @default_action = [:pass]
-    end
-
-    def match?(request)
-      @url_regex.nil? || @url_regex.match(request.request_url)
-    end
-
-    def next_action
-      @builder.actions.shift || @default_action
-    end
-
-    def relay(request, connection)
-      request.relay_to(connection.server_socket)
-      response = Response.read_from(connection.server_socket)
-      response.relay_to(connection.client_socket)
-    end
-
-    def evaluate(request, connection)
-      @builder.instance_exec(request, connection, &@action) if @action
-      action, *params = next_action
-      FlakyProxy.logger.info("    [#{action.upcase}] #{request.request_method} #{request.request_url}")
-      case action
-      when :pass
-        relay(request, connection)
-      when :respond
-        response_spec = params.first
-        response = Response.build(response_spec)
-        response.relay_to(connection.client_socket)
-      when :delay
-        amount = params.first
-        sleep(amount)
-        relay(request, connection)
-      when :close
-        connection.client_socket.close
-      end
-    end
-  end
-end

data/test/flaky_proxy/lib/flaky_proxy/rule_set.rb

@@ -1,45 +0,0 @@
-# encoding: utf-8
-# This file is distributed under New Relic's license terms.
-# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
-
-module FlakyProxy
-  class RuleSet
-    class Builder
-      attr_reader :rules
-
-      def initialize(ruleset, &blk)
-        @ruleset = ruleset
-        @rules = []
-      end
-
-      def match(criteria, sequence=nil, &blk)
-        if blk
-          @ruleset.rules << Rule.new(criteria, &blk)
-        else
-          @ruleset.rules << Rule.new(criteria, sequence.builder)
-        end
-      end
-
-      def sequence(&blk)
-        Sequence.new(&blk)
-      end
-    end
-
-    def self.build(text=nil, &blk)
-      ruleset = self.new
-      Builder.new(ruleset).instance_eval(text, &blk)
-      ruleset
-    end
-
-    attr_accessor :rules
-
-    def initialize
-      @rules = []
-      @default_rule = Rule.new { pass }
-    end
-
-    def match(request)
-      @rules.detect { |rule| rule.match?(request) } || @default_rule
-    end
-  end
-end

data/test/flaky_proxy/lib/flaky_proxy/sequence.rb

@@ -1,14 +0,0 @@
-# encoding: utf-8
-# This file is distributed under New Relic's license terms.
-# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
-
-module FlakyProxy
-  class Sequence
-    attr_reader :builder
-
-    def initialize(&blk)
-      @builder = Rule::ActionBuilder.new
-      @builder.instance_eval(&blk)
-    end
-  end
-end

data/test/flaky_proxy/lib/flaky_proxy/server.rb

@@ -1,22 +0,0 @@
-# encoding: utf-8
-# This file is distributed under New Relic's license terms.
-# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
-
-module FlakyProxy
-  class Server
-    attr_reader :host, :port
-
-    def initialize(host, port)
-      @host = host
-      @port = port
-    end
-
-    def open_socket
-      TCPSocket.new(@host, @port)
-    end
-
-    def to_s
-      "#{@host}:#{@port}"
-    end
-  end
-end

data/test/flaky_proxy/script/flaky_proxy

@@ -1,39 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'optparse'
-require File.join(File.dirname(__FILE__), '..', 'lib', 'flaky_proxy')
-
-DEFAULTS = {
-  :listen_host => 'localhost',
-  :listen_port => 8888,
-  :target_host => 'localhost',
-  :target_port => 8081
-}
-
-options = DEFAULTS.dup
-parser = OptionParser.new do |opts|
-  opts.banner = "Usage: #{$0} [options] [<rules file>]"
-
-  opts.on("-l", "--listen [<HOST>:]<PORT>", "Specify the host and port to listen on") do |listen_spec|
-    if listen_spec =~ /^([^:]+):(\d+)$/
-      options[:listen_host] = $1
-      options[:listen_port] = $2.to_i
-    else
-      options[:listen_port] = listen_spec.to_i
-    end
-  end
-
-  opts.on("-t", "--target [<HOST>:]<PORT>", "Specify the host and port to forward to") do |target_spec|
-    if target_spec =~ /^([^:]+):(\d+)$/
-      options[:target_host] = $1
-      options[:target_port] = $2
-    else
-      options[:target_port] = target_spec.to_i
-    end
-  end
-end
-parser.parse!
-
-options[:rules_path] = ARGV[0] if ARGV[0]
-
-FlakyProxy::Proxy.new(options).run