invoker 1.2.0.pre1 → 1.2.0.pre2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b5f7948dc467e955d06dc212305b0d58dea40073
-  data.tar.gz: 1c4a51d9cc8657a266abcc5c20ec994a5419cbb9
+  metadata.gz: 0522d780189e4589e2b918f4b3618ca6b84b0759
+  data.tar.gz: fd335e383fde399c6970a287c07d5b24ca5a7f6e
 SHA512:
-  metadata.gz: 80c642caaff5297ecc6cacbadc33c0e3a5261f61e72d621e0c5e512cec982e296c1e3434805da300f3a3c4a83a320d70e66fc2ffed3187f6a567c9965021c4c3
-  data.tar.gz: e55662e5c00a63f54223ba93a550ae25a71d7e8eb89d778dd03b6495145ef9295460288b171456392c1a0fc028dedb904fb106b1b4a73817025672fae9856a45
+  metadata.gz: cc0847a296deb3db14ac14342a3c7acdd42bd1642954aff5807333bf71699793d29d488f89477df843a12fb451af4e6dc4b379b02926b70dd050c4bfaf8680c1
+  data.tar.gz: 2a24694f1ebf7262478a376b7b23a4fcdf3386cfbf55101e0380299e34414ed8805e993f3f4bb45c0b2e429890922907f55d65c38c735417eede8336e8fb5436

data/Gemfile

@@ -3,7 +3,7 @@ source "https://rubygems.org"
 gemspec
 
 gem 'pry'
-gem "rubydns", git: "git@github.com:ioquatix/rubydns.git"
+gem "rubydns", git: "https://github.com/ioquatix/rubydns.git"
 
 gem 'coveralls', require: false
 gem "simplecov", require: false

data/invoker.gemspec

@@ -33,6 +33,7 @@ Gem::Specification.new do |s|
   s.add_dependency("em-proxy", "~> 0.1")
   s.add_dependency("rubydns", "~> 0.7")
   s.add_dependency("uuid", "~> 2.3")
+  s.add_dependency("facter", "~> 2.0")
   s.add_dependency("http-parser-lite", "~> 0.6")
   s.add_dependency("dotenv", "~> 0.9")
   s.add_development_dependency("rspec")

data/lib/invoker.rb

@@ -8,6 +8,7 @@ require "uuid"
 require "json"
 require "rainbow"
 require "rainbow/ext/string"
+require "etc"
 
 require "invoker/version"
 require "invoker/logger"
@@ -64,7 +65,7 @@ module Invoker
     end
 
     def can_run_balancer?(throw_warning = true)
-      return true if File.exist?(Invoker::Power::Config::CONFIG_LOCATION)
+      return true if File.exist?(Invoker::Power::Config.config_file)
 
       if throw_warning
         Invoker::Logger.puts("Invoker has detected setup has not been run. Domain feature will not work without running setup command.".color(:red))
@@ -73,18 +74,18 @@ module Invoker
     end
 
     def setup_config_location
-      config_location = File.join(Dir.home, '.invoker')
-      return config_location if Dir.exist?(config_location)
+      config_dir = Invoker::Power::Config.config_dir
+      return config_dir if Dir.exist?(config_dir)
 
-      if File.exist?(config_location)
-        old_config = File.read(config_location)
-        FileUtils.rm_f(config_location)
+      if File.exist?(config_dir)
+        old_config = File.read(config_dir)
+        FileUtils.rm_f(config_dir)
       end
 
-      FileUtils.mkdir(config_location)
+      FileUtils.mkdir(config_dir)
 
-      migrate_old_config(old_config, config_location) if old_config
-      config_location
+      migrate_old_config(old_config, config_dir) if old_config
+      config_dir
     end
 
     def run_without_bundler
@@ -116,5 +117,18 @@ module Invoker
         file.write(old_config)
       end
     end
+
+    # On some platforms `Dir.home` or `ENV['HOME']` does not return home directory of user.
+    # this is especially true, after effective and real user id of process
+    # has been changed.
+    #
+    # @return [String] home directory of the user
+    def home
+      if File.writable?(Dir.home)
+        Dir.home
+      else
+        Etc.getpwuid(Process.uid).dir
+      end
+    end
   end
 end

data/lib/invoker/daemon.rb

@@ -24,7 +24,7 @@ module Invoker
     end
 
     def pid_file
-      File.join(Dir.home, ".invoker", "#{process_name}.pid")
+      File.join(Invoker.home, ".invoker", "#{process_name}.pid")
     end
 
     def pid
@@ -32,7 +32,7 @@ module Invoker
     end
 
     def log_file
-      File.join(Dir.home, ".invoker", "#{process_name}.log")
+      File.join(Invoker.home, ".invoker", "#{process_name}.log")
     end
 
     def daemonize

data/lib/invoker/power/balancer.rb

@@ -1,5 +1,6 @@
 require 'em-proxy'
 require 'http-parser'
+require "invoker/power/http_parser"
 
 module Invoker
   module Power
@@ -19,37 +20,6 @@ module Invoker
       end
     end
 
-    class BalancerParser
-      attr_accessor :host, :parser
-      def initialize
-        @parser = HTTP::Parser.new()
-        @header = {}
-        @parser.on_headers_complete { headers_received() }
-        @parser.on_header_field { |field_name|
-          @last_key = field_name
-        }
-        @parser.on_header_value { |value| header_value_received(value) }
-      end
-
-      def headers_received
-        @header_completion_callback.call(@header)
-      end
-
-      def header_value_received(value)
-        @header[@last_key] = value
-      end
-
-      def on_headers_complete(&block)
-        @header_completion_callback = block
-      end
-
-      def reset; @parser.reset(); end
-
-      def <<(data)
-        @parser << data
-      end
-    end
-
     class Balancer
       attr_accessor :connection, :http_parser, :session, :protocol
       DEV_MATCH_REGEX = /([\w-]+)\.dev(\:\d+)?$/
@@ -71,49 +41,42 @@ module Invoker
       def initialize(connection, protocol)
         @connection = connection
         @protocol = protocol
-        @http_parser = BalancerParser.new()
+        @http_parser = HttpParser.new(protocol)
         @session = nil
         @buffer = []
       end
 
-      # insert X_FORWARDED_PROTO_ so as rails can identify the request as coming from
-      # https
-      def insert_forwarded_proto_header(data)
-        if data =~ /\r\n\r\n/ && data !~ /X_FORWARDED_PROTO/i && protocol == 'https'
-          data.gsub(/\r\n\r\n/, "\r\nX_FORWARDED_PROTO: #{protocol}\r\n\r\n")
-        else
-          data
-        end
-      end
-
       def install_callbacks
-        http_parser.on_headers_complete { |header| headers_received(header) }
+        http_parser.on_headers_complete { |headers| headers_received(headers) }
+        http_parser.on_message_complete { |full_message| complete_message_received(full_message) }
         connection.on_data { |data| upstream_data(data) }
         connection.on_response { |backend, data| backend_data(backend, data) }
         connection.on_finish { |backend, name| frontend_disconnect(backend, name) }
       end
 
-      def headers_received(header)
+      def complete_message_received(full_message)
+        connection.relay_to_servers(full_message)
+        http_parser.reset
+      end
+
+      def headers_received(headers)
+        if @session
+          return
+        end
         @session = UUID.generate()
-        dns_check_response = select_backend_config(header['Host'])
+        dns_check_response = select_backend_config(headers['Host'])
         if dns_check_response && dns_check_response.port
           connection.server(session, host: '0.0.0.0', port: dns_check_response.port)
-          connection.relay_to_servers(insert_forwarded_proto_header(@buffer.join))
-          @buffer = []
         else
           return_error_page(404)
+          http_parser.reset
           connection.close_connection_after_writing
         end
       end
 
       def upstream_data(data)
-        unless session
-          @buffer << data
-          append_for_http_parsing(data)
-          nil
-        else
-          insert_forwarded_proto_header(data)
-        end
+        append_for_http_parsing(data)
+        nil
       end
 
       def append_for_http_parsing(data)
@@ -131,7 +94,6 @@ module Invoker
       def frontend_disconnect(backend, name)
         http_parser.reset
         unless @backend_data
-          Invoker::Logger.puts("\nApplication #{name} not running. Returning error page.".color(:red))
           return_error_page(503)
         end
         @backend_data = false

data/lib/invoker/power/config.rb

@@ -6,32 +6,38 @@ module Invoker
     class ConfigExists < StandardError; end
 
     class Config
-      CONFIG_LOCATION = File.join(Dir.home, ".invoker", "config")
-
       def self.has_config?
-        File.exist?(CONFIG_LOCATION)
+        File.exist?(config_file)
       end
 
       def self.create(options = {})
         if has_config?
-          raise ConfigExists, "Config file already exists at location #{CONFIG_LOCATION}"
+          raise ConfigExists, "Config file already exists at location #{config_file}"
         end
         config = new(options)
         config.save
       end
 
       def self.delete
-        if File.exist?(CONFIG_LOCATION)
-          File.delete(CONFIG_LOCATION)
+        if File.exist?(config_file)
+          File.delete(config_file)
         end
       end
 
+      def self.config_file
+        File.join(Invoker.home, ".invoker", "config")
+      end
+
+      def self.config_dir
+        File.join(Invoker.home, ".invoker")
+      end
+
       def initialize(options = {})
         @config = options
       end
 
       def self.load_config
-        config_hash = File.open(CONFIG_LOCATION, "r") { |fl| YAML.load(fl) }
+        config_hash = File.open(config_file, "r") { |fl| YAML.load(fl) }
         new(config_hash)
       end
 
@@ -60,7 +66,7 @@ module Invoker
       end
 
       def save
-        File.open(CONFIG_LOCATION, "w") do |fl|
+        File.open(self.class.config_file, "w") do |fl|
           YAML.dump(@config, fl)
         end
         self

data/lib/invoker/power/http_parser.rb

@@ -0,0 +1,68 @@
+module Invoker
+  module Power
+    class HttpParser
+      attr_accessor :host, :parser, :protocol
+
+      def initialize(protocol)
+        @protocol = protocol
+        @parser = HTTP::Parser.new
+        @header = {}
+        initialize_message_content
+        parser.on_headers_complete { complete_headers_received }
+        parser.on_header_field { |field_name| @last_key = field_name }
+        parser.on_header_value { |field_value| header_value_received(field_value) }
+
+        parser.on_message_complete { complete_message_received }
+      end
+
+      # define a callback for invoking when complete header is parsed
+      def on_headers_complete(&block)
+        @on_headers_complete_callback = block
+      end
+
+      def header_value_received(value)
+        @header[@last_key] = value
+      end
+
+      # define a callback to invoke when a full http message is received
+      def on_message_complete(&block)
+        @on_message_complete_callback = block
+      end
+
+      def reset
+        @header = {}
+        initialize_message_content
+        parser.reset
+      end
+
+      def << data
+        @full_message.write(data)
+        parser << data
+      end
+
+      private
+
+      def complete_message_received
+        full_message_string = @full_message.string.dup
+        if full_message_string =~ /\r\n\r\n/
+          full_message_string.sub!(/\r\n\r\n/, "\r\nX_FORWARDED_PROTO: #{protocol}\r\n\r\n")
+        end
+        if @on_message_complete_callback
+          @on_message_complete_callback.call(full_message_string)
+        end
+      end
+
+      def initialize_message_content
+        @full_message = StringIO.new
+        @full_message.set_encoding('ASCII-8BIT')
+      end
+
+      # gets invoker when complete header is received
+      def complete_headers_received
+        if @on_headers_complete_callback
+          @on_headers_complete_callback.call(@header)
+        end
+      end
+    end
+  end
+end

data/lib/invoker/power/setup.rb

@@ -44,7 +44,7 @@ module Invoker
       end
 
       def check_if_setup_can_run?
-        !File.exists?(Invoker::Power::Config::CONFIG_LOCATION)
+        !File.exists?(Invoker::Power::Config.config_file)
       end
     end
   end

data/lib/invoker/power/setup/distro/base.rb

@@ -0,0 +1,41 @@
+module Invoker
+  module Power
+    module Distro
+      class Base
+        RESOLVER_FILE = "/etc/dnsmasq.d/dev-tld"
+        RINETD_FILE = "/etc/rinetd.conf"
+
+        def self.distro_installer
+          case Facter[:operatingsystem].value
+          when "Ubuntu"
+            require "invoker/power/setup/distro/ubuntu"
+            Ubuntu.new
+          when "Fedora"
+            require "invoker/power/setup/distro/redhat"
+            Redhat.new
+          else
+            raise "Your selected distro is not supported by Invoker"
+          end
+        end
+
+        def resolver_file
+          RESOLVER_FILE
+        end
+
+        def rinetd_file
+          RINETD_FILE
+        end
+
+        # Install required software
+        def install_required_software
+          raise "Unimplemented"
+        end
+
+        def restart_services
+          system("service rinetd restart")
+          system("service dnsmasq restart")
+        end
+      end
+    end
+  end
+end

data/lib/invoker/power/setup/distro/redhat.rb

@@ -0,0 +1,17 @@
+module Invoker
+  module Power
+    module Distro
+      class Redhat < Base
+        def install_required_software
+          system("yum --assumeyes install dnsmasq rinetd")
+        end
+
+        def restart_services
+          system("systemctl enable rinetd")
+          system("service rinetd restart")
+          system("service dnsmasq restart")
+        end
+      end
+    end
+  end
+end

data/lib/invoker/power/setup/distro/ubuntu.rb

@@ -0,0 +1,11 @@
+module Invoker
+  module Power
+    module Distro
+      class Ubuntu < Base
+        def install_required_software
+          system("apt-get --assume-yes install dnsmasq rinetd")
+        end
+      end
+    end
+  end
+end

data/lib/invoker/power/setup/linux_setup.rb

@@ -1,16 +1,19 @@
+require "invoker/power/setup/distro/base"
+require "facter"
+
 module Invoker
   module Power
     class LinuxSetup < Setup
-      RESOLVER_FILE = "/etc/dnsmasq.d/dev-tld"
-      RINETD_FILE = "/etc/rinetd.conf"
+      attr_accessor :distro_installer
 
       def setup_invoker
         if get_user_confirmation?
+          initialize_distro_installer
           find_open_ports
-          install_required_software
+          distro_installer.install_required_software
           install_resolver
           install_port_forwarder
-          restart_services
+          distro_installer.restart_services
           drop_to_normal_user
           create_config_file
         else
@@ -34,23 +37,18 @@ module Invoker
 
       private
 
-      def restart_services
-        system("/etc/init.d/rinetd restart")
-        system("/etc/init.d/dnsmasq restart")
-      end
-
-      def install_required_software
-        system("apt-get --assume-yes install dnsmasq rinetd")
+      def initialize_distro_installer
+        @distro_installer =  Invoker::Power::Distro::Base.distro_installer
       end
 
       def install_resolver
-        File.open(RESOLVER_FILE, "w") do |fl|
+        File.open(distro_installer.resolver_file, "w") do |fl|
           fl.write(tld_setup)
         end
       end
 
       def install_port_forwarder
-        File.open(RINETD_FILE, "a") do |fl|
+        File.open(distro_installer.rinetd_file, "a") do |fl|
           fl << "\n"
           fl << rinetd_setup(port_finder.http_port, port_finder.https_port)
         end

data/lib/invoker/version.rb

@@ -1,3 +1,3 @@
 module Invoker
-  VERSION = "1.2.0.pre1"
+  VERSION = "1.2.0.pre2"
 end

data/spec/invoker/config_spec.rb

@@ -114,7 +114,7 @@ command = ls
     end
 
     it "loads config if platform is darwin and power config file exists" do
-      File.open(Invoker::Power::Config::CONFIG_LOCATION, "w") { |fl| fl.puts "sample" }
+      File.open(Invoker::Power::Config.config_file, "w") { |fl| fl.puts "sample" }
       Invoker::Power::Config.expects(:load_config).once
       Invoker::Parsers::Config.new(@file.path, 9000)
     end

data/spec/invoker/invoker_spec.rb

@@ -19,7 +19,7 @@ describe "Invoker" do
     end
 
     it "should return true if setup was run properly" do
-      File.open(Invoker::Power::Config::CONFIG_LOCATION, "w") {|fl|
+      File.open(Invoker::Power::Config.config_file, "w") {|fl|
         fl.write("hello")
       }
       expect(Invoker.can_run_balancer?).to be_true
@@ -61,4 +61,10 @@ describe "Invoker" do
       end
     end
   end
+
+  describe "#home" do
+    it "should return home directory using etc module" do
+      expect(Invoker.home).to eql ENV['HOME']
+    end
+  end
 end

data/spec/invoker/power/config_spec.rb

@@ -3,22 +3,15 @@ require "spec_helper"
 describe "Invoker Power configuration" do
   describe "#create" do
     it "should create a config file given a hash" do
-      begin
-        File.exists?(Invoker::Power::Config::CONFIG_LOCATION) &&
-          File.delete(Invoker::Power::Config::CONFIG_LOCATION)
-        config = Invoker::Power::Config.create(
-          dns_port: 1200, http_port: 1201, ipfw_rule_number: 010
-        )
-        expect(File.exists?(Invoker::Power::Config::CONFIG_LOCATION)).to be_true
+      config = Invoker::Power::Config.create(
+        dns_port: 1200, http_port: 1201, ipfw_rule_number: 010
+      )
+      expect(File.exist?(Invoker::Power::Config.config_file)).to be_true
 
-        config = Invoker::Power::Config.load_config()
-        expect(config.dns_port).to eq(1200)
-        expect(config.http_port).to eq(1201)
-        expect(config.ipfw_rule_number).to eq(010)
-      ensure
-        File.exists?(Invoker::Power::Config::CONFIG_LOCATION) &&
-          File.delete(Invoker::Power::Config::CONFIG_LOCATION)
-      end
+      config = Invoker::Power::Config.load_config()
+      expect(config.dns_port).to eq(1200)
+      expect(config.http_port).to eq(1201)
+      expect(config.ipfw_rule_number).to eq(010)
     end
   end
 end

data/spec/invoker/power/http_parser_spec.rb

@@ -0,0 +1,32 @@
+require "spec_helper"
+
+describe Invoker::Power::HttpParser do
+  let(:parser) { Invoker::Power::HttpParser.new('https') }
+
+  describe "complete message received" do
+    before { parser.reset }
+    it "should call header received with full header" do
+      @header = nil
+      parser.on_headers_complete { |header| @header = header }
+      parser << "HTTP/1.1 200 OK\r\n"
+      parser << "Content-Type: text/plain;charset=utf-8\r\n"
+      parser << "Content-Length: 5\r\n"
+      parser << "Connection: close\r\n\r\n"
+      parser << "hello"
+
+      expect(@header['Content-Type']).to eql "text/plain;charset=utf-8"
+    end
+
+    it "should return complete message with x_forwarded added" do
+      complete_message = nil
+      parser.on_message_complete { |message| complete_message = message }
+      parser.on_headers_complete { |header| @header = header }
+      parser << "HTTP/1.1 200 OK\r\n"
+      parser << "Content-Type: text/plain;charset=utf-8\r\n"
+      parser << "Content-Length: 5\r\n"
+      parser << "Connection: close\r\n\r\n"
+      parser << "hello"
+      expect(complete_message).to match(/X_FORWARDED_PROTO:/i)
+    end
+  end
+end

data/spec/invoker/power/setup/linux_setup_spec.rb

@@ -1,16 +1,23 @@
 require "spec_helper"
+require "invoker/power/setup/distro/ubuntu"
 
 describe Invoker::Power::LinuxSetup do
   let(:invoker_setup) { Invoker::Power::LinuxSetup.new }
+  let(:distro_installer) { Invoker::Power::Distro::Ubuntu.new }
+
   describe "should only proceed after user confirmation" do
+    before { invoker_setup.distro_installer = distro_installer }
+
     it "should create config file with port" do
+      invoker_setup.expects(:initialize_distro_installer).returns(true)
       invoker_setup.expects(:get_user_confirmation?).returns(true)
-      invoker_setup.expects(:install_required_software).returns(true)
       invoker_setup.expects(:install_resolver).returns(true)
       invoker_setup.expects(:install_port_forwarder).returns(true)
-      invoker_setup.expects(:restart_services).returns(true)
       invoker_setup.expects(:drop_to_normal_user).returns(true)
 
+      distro_installer.expects(:install_required_software)
+      distro_installer.expects(:restart_services)
+
       invoker_setup.setup_invoker
 
       config = Invoker::Power::Config.load_config
@@ -21,21 +28,25 @@ describe Invoker::Power::LinuxSetup do
   end
 
   describe "configuring dnsmasq and rinetd" do
+    before { invoker_setup.distro_installer = distro_installer }
+
     it "should create proper config file" do
+      invoker_setup.expects(:initialize_distro_installer).returns(true)
       invoker_setup.expects(:get_user_confirmation?).returns(true)
-      invoker_setup.expects(:install_required_software).returns(true)
-      invoker_setup.expects(:restart_services).returns(true)
       invoker_setup.expects(:drop_to_normal_user).returns(true)
 
+      distro_installer.expects(:install_required_software)
+      distro_installer.expects(:restart_services)
+
       invoker_setup.setup_invoker
 
       config = Invoker::Power::Config.load_config
 
-      dnsmasq_content = File.read(Invoker::Power::LinuxSetup::RESOLVER_FILE)
+      dnsmasq_content = File.read(distro_installer.resolver_file)
       expect(dnsmasq_content.strip).to_not be_empty
       expect(dnsmasq_content).to match(/dev/)
 
-      rinetd_content = File.read(Invoker::Power::LinuxSetup::RINETD_FILE)
+      rinetd_content = File.read(distro_installer.rinetd_file)
       expect(rinetd_content.strip).to_not be_empty
       expect(rinetd_content.strip).to match(/#{config.https_port}/)
       expect(rinetd_content.strip).to match(/#{config.http_port}/)

data/spec/invoker/power/setup/osx_setup_spec.rb

@@ -20,7 +20,7 @@ describe Invoker::Power::OsxSetup do
 
   describe "when a setup file exists" do
     it "should throw error about existing file" do
-      File.open(Invoker::Power::Config::CONFIG_LOCATION, "w") {|fl|
+      File.open(Invoker::Power::Config.config_file, "w") {|fl|
         fl.write("foo test")
       }
       Invoker::Power::Setup.any_instance.expects(:setup_invoker).never

data/spec/support/mock_setup_file.rb

@@ -23,16 +23,14 @@ module MockSetupFile
   private
 
   def setup_invoker_config
-    @old_config = Invoker::Power::Config::CONFIG_LOCATION
-    Invoker::Power::Config.const_set(:CONFIG_LOCATION, "/tmp/.invoker/config")
+    Invoker::Power::Config.stubs(:config_file).returns("/tmp/.invoker/config")
+    Invoker::Power::Config.stubs(:config_dir).returns("/tmp/.invoker")
     safe_make_directory("/tmp/.invoker")
-    safe_remove_file(Invoker::Power::Config::CONFIG_LOCATION)
+    safe_remove_file(Invoker::Power::Config.config_file)
   end
 
   def restore_invoker_config
-    safe_remove_file(Invoker::Power::Config::CONFIG_LOCATION)
-    Invoker::Power::Config.const_set(:CONFIG_LOCATION, @old_config)
-    $VERBOSE = @original_verbosity
+    safe_remove_file(Invoker::Power::Config.config_file)
   end
 
   def restore_osx_resolver_setup
@@ -51,18 +49,16 @@ module MockSetupFile
   end
 
   def setup_linux_resolver_path
-    @old_linux_resolver = Invoker::Power::LinuxSetup::RESOLVER_FILE
-    @old_rinetd_config = Invoker::Power::LinuxSetup::RINETD_FILE
-    Invoker::Power::LinuxSetup.const_set(:RESOLVER_FILE, "/tmp/dev-tld")
-    Invoker::Power::LinuxSetup.const_set(:RINETD_FILE, "/tmp/rinetd.conf")
-    safe_remove_file(Invoker::Power::LinuxSetup::RESOLVER_FILE)
-    safe_remove_file(Invoker::Power::LinuxSetup::RINETD_FILE)
+    @old_linux_resolver = Invoker::Power::Distro::Base::RESOLVER_FILE
+    @old_rinetd_config = Invoker::Power::Distro::Base::RINETD_FILE
+    Invoker::Power::Distro::Base.const_set(:RESOLVER_FILE, "/tmp/dev-tld")
+    Invoker::Power::Distro::Base.const_set(:RINETD_FILE, "/tmp/rinetd.conf")
   end
 
   def restore_linux_resolver_path
-    safe_remove_file(Invoker::Power::LinuxSetup::RESOLVER_FILE)
-    safe_remove_file(Invoker::Power::LinuxSetup::RINETD_FILE)
-    Invoker::Power::LinuxSetup.const_set(:RESOLVER_FILE, @old_linux_resolver)
-    Invoker::Power::LinuxSetup.const_set(:RINETD_FILE, @old_rinetd_config)
+    safe_remove_file(Invoker::Power::Distro::Base::RESOLVER_FILE)
+    safe_remove_file(Invoker::Power::Distro::Base::RINETD_FILE)
+    Invoker::Power::Distro::Base.const_set(:RESOLVER_FILE, @old_linux_resolver)
+    Invoker::Power::Distro::Base.const_set(:RINETD_FILE, @old_rinetd_config)
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: invoker
 version: !ruby/object:Gem::Version
-  version: 1.2.0.pre1
+  version: 1.2.0.pre2
 platform: ruby
 authors:
 - Hemant Kumar
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-26 00:00:00.000000000 Z
+date: 2014-06-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -123,6 +123,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.3'
+- !ruby/object:Gem::Dependency
+  name: facter
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
 - !ruby/object:Gem::Dependency
   name: http-parser-lite
   requirement: !ruby/object:Gem::Requirement
@@ -245,11 +259,15 @@ files:
 - lib/invoker/power/balancer.rb
 - lib/invoker/power/config.rb
 - lib/invoker/power/dns.rb
+- lib/invoker/power/http_parser.rb
 - lib/invoker/power/http_response.rb
 - lib/invoker/power/port_finder.rb
 - lib/invoker/power/power.rb
 - lib/invoker/power/powerup.rb
 - lib/invoker/power/setup.rb
+- lib/invoker/power/setup/distro/base.rb
+- lib/invoker/power/setup/distro/redhat.rb
+- lib/invoker/power/setup/distro/ubuntu.rb
 - lib/invoker/power/setup/linux_setup.rb
 - lib/invoker/power/setup/osx_setup.rb
 - lib/invoker/power/templates/404.html
@@ -276,6 +294,7 @@ files:
 - spec/invoker/ipc/unix_client_spec.rb
 - spec/invoker/power/balancer_spec.rb
 - spec/invoker/power/config_spec.rb
+- spec/invoker/power/http_parser_spec.rb
 - spec/invoker/power/http_response_spec.rb
 - spec/invoker/power/port_finder_spec.rb
 - spec/invoker/power/setup/linux_setup_spec.rb
@@ -309,5 +328,31 @@ rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Something small for Process management
-test_files: []
+test_files:
+- spec/invoker/cli/pinger_spec.rb
+- spec/invoker/cli/tail_watcher_spec.rb
+- spec/invoker/cli_spec.rb
+- spec/invoker/command_worker_spec.rb
+- spec/invoker/commander_spec.rb
+- spec/invoker/config_spec.rb
+- spec/invoker/daemon_spec.rb
+- spec/invoker/event/manager_spec.rb
+- spec/invoker/invoker_spec.rb
+- spec/invoker/ipc/client_handler_spec.rb
+- spec/invoker/ipc/dns_check_command_spec.rb
+- spec/invoker/ipc/message/list_response_spec.rb
+- spec/invoker/ipc/message_spec.rb
+- spec/invoker/ipc/unix_client_spec.rb
+- spec/invoker/power/balancer_spec.rb
+- spec/invoker/power/config_spec.rb
+- spec/invoker/power/http_parser_spec.rb
+- spec/invoker/power/http_response_spec.rb
+- spec/invoker/power/port_finder_spec.rb
+- spec/invoker/power/setup/linux_setup_spec.rb
+- spec/invoker/power/setup/osx_setup_spec.rb
+- spec/invoker/power/setup_spec.rb
+- spec/invoker/process_manager_spec.rb
+- spec/invoker/reactor_spec.rb
+- spec/spec_helper.rb
+- spec/support/mock_setup_file.rb
 has_rdoc: