elasticsearch_update 1.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 8184b6ab6ae68ea0bee8b38f1cc54059ed3bbb7d
+  data.tar.gz: f8b5bca19aed12f5e68ef3aaea2c9ff4170f2042
+SHA512:
+  metadata.gz: 8d17edc14813b94caa113d2d3ca5b790e7df47093b8330fc63c10b511390aceb9713c781badc00ec85fcea0b664da25322ede0003c6232542d901b713d64e670
+  data.tar.gz: f6fce5f86ddb2f53e19077cc24e00d5d1a31d74a18972597d0289a17d4f433ee97ec268865890c6c339c8db5fc9eb7eec1d8f9e68bc64bd6474fd4f724610d52

data/.gitignore

@@ -0,0 +1,14 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log

data/.travis.yml

@@ -0,0 +1,15 @@
+language: ruby
+rvm:
+  - "2.0.0"
+  - "2.1.0"
+  - "2.2.0"
+
+before_install:
+  - gem update --system
+  - gem --version
+
+gemfile:
+  - Gemfile
+
+notifications:
+  email: false

data/Gemfile

@@ -0,0 +1,10 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in elasticsearch_update.gemspec
+gemspec
+gem 'highline'
+gem 'rake', group: :test, require: nil
+gem 'minitest', group: :test, require: nil
+gem 'bundler', group: :test, require: nil
+gem 'rubyzip'
+gem "codeclimate-test-reporter", group: :test, require: nil

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Kevin Kirsche
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,50 @@
+[![Build Status](https://travis-ci.org/kkirsche/elasticsearch_update.svg?branch=master)](https://travis-ci.org/kkirsche/elasticsearch_update) [![Code Climate](https://codeclimate.com/github/kkirsche/elasticsearch_update/badges/gpa.svg)](https://codeclimate.com/github/kkirsche/elasticsearch_update) [![Test Coverage](https://codeclimate.com/github/kkirsche/elasticsearch_update/badges/coverage.svg)](https://codeclimate.com/github/kkirsche/elasticsearch_update) [![Dependency Status](https://gemnasium.com/kkirsche/elasticsearch_update.svg)](https://gemnasium.com/kkirsche/elasticsearch_update)
+# Elasticsearch Update v1.0.1
+
+This gem allows users to easily update their 1.0 and later Elasticsearch instance on the local machine from one of the following formats.
+
+1. .zip (in progress)
+2. .tar.gz (in progress)
+3. .deb (complete)
+4. .rpm (complete)
+
+Installation follows the upgrade recommendations of Elasticsearch found [here](http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/setup-upgrade.html).
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'elasticsearch_update'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install elasticsearch_update
+
+## Usage
+
+After installing the gem, execute the `elasticsearch_update` command and it will walk you through the upgrade process.
+
+For development, execute `bundle exec bin/elasticsearch_update` and the gem will walk you through the upgrade process.
+
+For testing, execute `rake test` and the gem will run the local tests.
+
+## Versioning
+
+For transparency into my release cycle and in striving to maintain backward compatibility, Elasticsearch Update is maintained under [the Semantic Versioning guidelines](http://semver.org/). Sometimes I screw up, but I'll adhere to those rules whenever possible.
+
+## Contributing
+
+1. Fork it ( https://github.com/kkirsche/elasticsearch_update/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request
+
+## Warning
+Before performing an upgrade, it’s a good idea to back up the data on your system. This will allow you to roll back in the event of a problem with the upgrade. The upgrades sometimes include upgrades to the Lucene libraries used by Elasticsearch to access the index files, and after an index file has been updated to work with a new version of Lucene, it may not be accessible to the versions of Lucene present in earlier Elasticsearch releases. I am not responsible for any loss of data, damaged data, etc. caused by use of this gem.

data/Rakefile

@@ -0,0 +1,11 @@
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test/"
+  t.libs << "bin/"
+  t.libs << "lib/"
+  t.pattern = "test/elasticsearch_update/*_spec.rb"
+end
+
+task :default => :test

data/bin/elasticsearch_update

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+require 'elasticsearch_update'
+
+args = ARGV.dup
+ARGV.clear
+command = args.shift.strip rescue 'help'
+
+ElasticsearchUpdate::Cli.start

data/bin/update.rb

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+require 'elasticsearch_update'
+
+args = ARGV.dup
+ARGV.clear
+command = args.shift.strip rescue 'help'
+
+ElasticsearchUpdate::Cli.start

data/elasticsearch_update.gemspec

@@ -0,0 +1,25 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'elasticsearch_update/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "elasticsearch_update"
+  spec.version       = ElasticsearchUpdate::VERSION
+  spec.authors       = ["Kevin Kirsche"]
+  spec.email         = ["kev.kirsche@gmail.com"]
+  spec.summary       = %q{Updates the elasticsearch instance on the local machine.}
+  spec.description   = %q{Updates the elasticsearch instance from a deb or rpm on the local machine. Assumes elasticsearch is a service.}
+  spec.homepage      = "https://github.com/kkirsche/elasticsearch_update"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   << 'elasticsearch_update'
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.7"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "minitest", "5.4.3"
+  spec.add_runtime_dependency 'highline', '1.6.21'
+end

data/lib/elasticsearch_update.rb

@@ -0,0 +1,11 @@
+require 'elasticsearch_update/version'
+require 'elasticsearch_update/cli'
+require 'elasticsearch_update/downloader'
+require 'elasticsearch_update/elasticsearch'
+require 'elasticsearch_update/installer'
+require 'elasticsearch_update/version'
+require 'elasticsearch_update/wizard'
+
+# This module updates a debian Elasticsearch instance
+module ElasticsearchUpdate
+end

data/lib/elasticsearch_update/cli.rb

@@ -0,0 +1,32 @@
+require 'logger'
+
+module ElasticsearchUpdate
+  # This class is in charge of retrieving and downloading data.
+  class Cli
+    def self.start
+      @log = Logger.new(STDOUT)
+      @log.level = Logger::INFO
+
+      @log.debug('Logger created for CLI.')
+
+      wizard = Wizard.new
+
+      es_client = Elasticsearch.new wizard.es_location_hash
+      es_client.cluster_routing_allocation('none')
+      es_client.shutdown_local_node
+
+      downloader = Downloader.new(wizard.download_hash)
+      file = downloader.download_file
+      downloader.verify_update_file
+
+      installer = Installer.new(wizard.sudo_password,
+                                downloader.extension)
+      installer.install_file(file)
+
+      es_client.start_elasticsearch(installer)
+      @log.info('Waiting for Elasticsearch to start.')
+      sleep 10
+      es_client.cluster_routing_allocation('all')
+    end
+  end
+end

data/lib/elasticsearch_update/downloader.rb

@@ -0,0 +1,83 @@
+require 'logger'
+require 'digest/sha1'
+require 'tempfile'
+require 'open-uri'
+
+module ElasticsearchUpdate
+  # This class is in charge of retrieving and downloading data.
+  class Downloader
+    attr_reader :extension, :base, :version, :download_url, :verify_url
+    attr_accessor :update_file
+    def initialize(hash, test = false)
+      @log = Logger.new(STDOUT)
+      if test
+        @log.level = Logger::FATAL
+      else
+        @log.level = Logger::INFO
+      end
+
+      @log.debug('Logger created for Downloader.')
+
+      @extension = hash[:extension]
+      @base = hash[:base_url]
+      @version = hash[:version]
+      @download_url = 'https://' + @base +
+                      '/elasticsearch/elasticsearch/elasticsearch-' + @version +
+                      @extension
+      @verify_url = 'https://' + @base +
+                    '/elasticsearch/elasticsearch/elasticsearch-' + @version +
+                    @extension + '.sha1.txt'
+    end
+
+    def write_file_from_url(file, url)
+      Net::HTTP.start(@base) do |http|
+        begin
+          http.request_get(url) do |resp|
+            resp.read_body do |segment|
+              file.write(segment)
+            end
+          end
+        ensure
+          file.close
+        end
+      end
+    end
+
+    def download_file(test = false)
+      @update_file = Tempfile.new(['elasticsearch_update_file', @extension])
+
+      @log.info('Downloading file from url.')
+
+      write_file_from_url(@update_file, @download_url) unless test
+
+      @update_file
+    end
+
+    def verify_update_file
+      @log.info('Beginning integrity check of downloaded file .')
+      @file_sha1 = Digest::SHA1.file(@update_file.path).hexdigest
+
+      @log.info('Verifying integrity of downloaded file.')
+
+      if download_remote_sha1 == @file_sha1
+        @log.info('Integrity verified.')
+        true
+      else
+        abort('File was not downloaded correctly. Please try again.')
+      end
+    end
+
+    def download_remote_sha1
+      @log.info('Downloading Elasticsearch SHA1.')
+
+      @remote_sha1 = ''
+      open(@verify_url) do |file|
+        @remote_sha1 = file.read
+      end
+
+      @remote_sha1 = @remote_sha1.split(/\s\s/)[0]
+
+      @remote_sha1
+    end
+  end
+end

data/lib/elasticsearch_update/elasticsearch.rb

@@ -0,0 +1,72 @@
+require 'logger'
+require 'json'
+require 'net/http'
+
+module ElasticsearchUpdate
+  # This class is in charge of retrieving and downloading data.
+  class Elasticsearch
+    attr_reader :es_host, :es_port
+    def initialize(hash = { host: 'localhost', port: 9200 }, test = false)
+      @log = Logger.new(STDOUT)
+      if test
+        @log.level = Logger::FATAL
+      else
+        @log.level = Logger::INFO
+      end
+
+      @log.debug('Logger created for Elasticsearch.')
+
+      @es_host = hash[:host]
+      @es_port = hash[:port]
+    end
+
+    def cluster_routing_allocation(type)
+      @log.info('Disabling cluster routing allocation')
+
+      begin
+        req = Net::HTTP::Put.new('/_cluster/settings',
+                                 'Content-Type' => 'application/javascript')
+        req.body = { transient: { 'cluster.routing.allocation.enable' => type } }
+        req.body = req.body.to_json
+        response = Net::HTTP.new(@es_host, @es_port).start {|http| http.request(req) }
+      rescue Errno::ECONNREFUSED
+        puts 'Connection could not be made to Elasticsearch at'
+        puts @es_host + ':' + @es_port + '/_cluster/settings'
+        abort('Please verify that Elasticsearch is available at this address.')
+      end
+
+      response
+    end
+
+    def shutdown_local_node
+      @log.info('Shutting down local node')
+      @shutdown_uri = URI('http://' + @es_host + ':' + @es_port.to_s + '/_cluster/nodes/_local/_shutdown')
+      response = Net::HTTP.post_form(@shutdown_uri, {})
+
+      response
+    end
+
+    def start_elasticsearch_service(password)
+      @log.info('Starting elasticsearch service')
+      system('echo ' + password + ' | sudo -S service elasticsearch start')
+    end
+
+    def start_elasticsearch_binary(path)
+      @log.info('Starting elasticsearch binary')
+      system(path + 'bin/elasticsearch')
+    end
+
+    def start_elasticsearch(installer_obj)
+      case installer_obj.extension
+      when '.zip'
+        start_elasticsearch_binary(wizard.test)
+      when '.deb'
+        start_elasticsearch_service(installer_obj.sudo_password)
+      when '.rpm'
+        start_elasticsearch_service(installer_obj.sudo_password)
+      when '.tar.gz'
+
+      end
+    end
+  end
+end

data/lib/elasticsearch_update/installer.rb

@@ -0,0 +1,49 @@
+require 'logger'
+
+module ElasticsearchUpdate
+  # This class is in charge of retrieving and downloading data.
+  class Installer
+    attr_accessor :sudo_password, :extension
+    def initialize(password, extension, test = false)
+      @log = Logger.new(STDOUT)
+      if test
+        @log.level = Logger::FATAL
+      else
+        @log.level = Logger::INFO
+      end
+
+      @log.debug('Logger created for Installer.')
+
+      @sudo_password = password
+      @extension = extension
+    end
+
+    def install_file(file)
+      case @extension
+      when '.zip'
+        true
+      when '.deb'
+        install_update_file(file, @extension)
+      when '.rpm'
+        install_update_file(file, @extension)
+      when '.tar.gz'
+        true
+      end
+    end
+
+    def install_update_file(file, extension)
+      @log.info('Installing' + extension + 'file.')
+      command = 'echo ' + @sudo_password + ' | '
+      case extension
+      when '.deb'
+        command += 'sudo -S dpkg -i "' + file.path + '"'
+      when '.rpm'
+        command += 'sudo -S rpm -i "' + file.path + '"'
+      end
+      Kernel.system(command)
+    end
+
+    def unzip_file(file)
+    end
+  end
+end

data/lib/elasticsearch_update/version.rb

@@ -0,0 +1,3 @@
+module ElasticsearchUpdate
+  VERSION = "1.0.1"
+end

data/lib/elasticsearch_update/wizard.rb

@@ -0,0 +1,61 @@
+require 'highline/import'
+
+module ElasticsearchUpdate
+  # This class is in charge of retrieving and downloading data.
+  class Wizard
+    def extension
+      choose do |menu|
+        menu.prompt = 'Which type of upgrade are we doing?  '
+
+        menu.choice(:deb) { @choice = '.deb' }
+        # menu.choice(:zip) { @choice = '.zip' }
+        menu.choice(:rpm) { @choice = '.rpm' }
+        # menu.choice(:tar) { @choice = '.tar.gz' }
+      end
+
+      @choice
+    end
+
+    def version
+      ask('What version of Elasticsearch should we update to? (major.minor.path) ', String) { |q| q.validate = /\d\.\d\.\d/ }
+    end
+
+    def host
+      response = ask('What is your Elasticsearch hostname? (Default: localhost) ', String)
+      response = 'localhost' if response == ''
+
+      response
+    end
+
+    def port
+      response = ask('What is your Elasticsearch port? (Default: 9200) ', String)
+
+      response = '9200' if response == ''
+
+      response
+    end
+
+    def es_location_hash
+      {
+        host: host,
+        port: port
+      }
+    end
+
+    def elasticsearch_fs_location
+      ask('In what directory does Elasticsearch run? ', String)
+    end
+
+    def sudo_password
+      ask('What password should be used while updating Elasticsearch? ') { |q| q.echo = "x" }
+    end
+
+    def download_hash
+      {
+        base_url: 'download.elasticsearch.org',
+        version: version,
+        extension: extension
+      }
+    end
+  end
+end

data/test/elasticsearch_update/downloader_spec.rb

@@ -0,0 +1,134 @@
+require 'minitest_helper'
+
+module ElasticsearchUpdate
+  class TestDownloader < Downloader
+    def open(_url, &b)
+      b.call StringIO.new "d377e39343e5cc277104beee349e1578dc50f7f8  elasticsearch-1.4.2.deb"
+    end
+
+    def abort(string)
+      string
+    end
+  end
+
+  # The TestDownloader class below tests the Downloader class from the library
+  class TestDownloader
+    describe 'Downloader', 'Used to download and verify Elasticsearch file' do
+
+      it 'should initialize and correctly assign values.' do
+        hash =
+        {
+          base_url: 'download.elasticsearch.org',
+          version: '1.4.2',
+          extension: '.deb'
+        }
+
+        @downloader = TestDownloader.new(hash, false)
+        assert_kind_of ElasticsearchUpdate::Downloader, @downloader
+
+        @downloader.base.must_equal 'download.elasticsearch.org'
+        @downloader.extension.must_equal '.deb'
+        @downloader.version.must_equal '1.4.2'
+        @downloader.download_url.must_equal 'https://download.elasticsearch.org/elasticsearch/elasticsearch/elasticsearch-1.4.2.deb'
+        @downloader.verify_url.must_equal 'https://download.elasticsearch.org/elasticsearch/elasticsearch/elasticsearch-1.4.2.deb.sha1.txt'
+      end
+
+      it 'should retrieve the correct SHA1 value from Elasticsearch\'s website' do
+        hash =
+        {
+          base_url: 'download.elasticsearch.org',
+          version: '1.4.2',
+          extension: '.deb'
+        }
+
+        @mock_file = Minitest::Mock.new
+        @mock_file.expect(:read, 'd377e39343e5cc277104beee349e1578dc50f7f8  elasticsearch-1.4.2.deb')
+
+        Kernel.stub :open, nil, @mock_file do
+          @downloader = TestDownloader.new(hash, true)
+          @downloader.download_remote_sha1.must_equal 'd377e39343e5cc277104beee349e1578dc50f7f8'
+        end
+      end
+
+      it "should retrieve the verify the file's SHA1 value against Elasticsearch's" do
+        hash =
+        {
+          base_url: 'download.elasticsearch.org',
+          version: '1.4.2',
+          extension: '.deb'
+        }
+
+        @mock_file_obj = Minitest::Mock.new
+        @mock_result = 'd377e39343e5cc277104beee349e1578dc50f7f8'
+        @mock_file_obj.expect(:hexdigest, @mock_result)
+
+        @mock_file = Minitest::Mock.new
+        @mock_file.expect(:path, 'fake/path')
+
+        Digest::SHA1.stub :file, @mock_file_obj do
+          @downloader = TestDownloader.new(hash, true)
+          @downloader.update_file = @mock_file
+          @downloader.verify_update_file.must_equal true
+        end
+      end
+
+      it "should retrieve the verify the file's SHA1 value against Elasticsearch's" do
+        hash =
+        {
+          base_url: 'download.elasticsearch.org',
+          version: '1.4.2',
+          extension: '.deb'
+        }
+
+        @mock_file_obj = Minitest::Mock.new
+        @mock_result = '0'
+        @mock_file_obj.expect(:hexdigest, @mock_result)
+
+        @mock_file = Minitest::Mock.new
+        @mock_file.expect(:path, 'fake/path')
+
+        Digest::SHA1.stub :file, @mock_file_obj do
+          @downloader = TestDownloader.new(hash, true)
+          @downloader.update_file = @mock_file
+          @downloader.verify_update_file.must_equal 'File was not downloaded correctly. Please try again.'
+        end
+      end
+
+      it 'creates a tempfile to store the Elasticsearch deb file' do
+        hash =
+        {
+          base_url: 'download.elasticsearch.org',
+          version: '1.4.2',
+          extension: '.deb'
+        }
+
+        @downloader = TestDownloader.new(hash, true)
+        assert_kind_of Tempfile, @downloader.download_file(true)
+      end
+
+      it 'writes a downloaded file' do
+        hash =
+        {
+          base_url: 'download.elasticsearch.org',
+          version: '1.4.2',
+          extension: '.deb'
+        }
+
+        @mock_resp = Minitest::Mock.new
+        @mock_resp.expect(:read_body, 'Test')
+
+        @mock_http = Minitest::Mock.new
+        @mock_http.expect(:request_get, @mock_resp, [String])
+
+        @mock_file = Minitest::Mock.new
+        @mock_file.expect(:write, true, [String])
+        @mock_file.expect(:close, true)
+
+        @downloader = TestDownloader.new(hash, true)
+        Net::HTTP.stub :start, true, @mock_http do
+          @downloader.write_file_from_url(@mock_file, 'http://test.org')
+        end
+      end
+    end
+  end
+end

data/test/elasticsearch_update/elasticsearch_spec.rb

@@ -0,0 +1,92 @@
+require 'minitest_helper'
+
+module ElasticsearchUpdate
+  class TestElasticsearch < Elasticsearch
+    def system(command)
+      command
+    end
+  end
+  # The TestInstaller class below tests the Downloader class from the library
+  class TestElasticsearch
+    describe 'Elasticsearch', 'Used to interact with Elasticsearch' do
+
+      it 'should initialize without errors' do
+        @es_client = TestElasticsearch.new
+        assert_kind_of Elasticsearch, @es_client
+
+        @es_client.es_port.must_equal 9200
+        @es_client.es_host.must_equal 'localhost'
+      end
+
+      it 'should disable cluster allocation' do
+        @es_client = TestElasticsearch.new({ host: 'localhost', port: 9200 }, true)
+
+        @mock_http = Minitest::Mock.new
+        @mock_http.expect(:start, 200)
+
+        Net::HTTP.stub :new, @mock_http do
+          @es_client.cluster_routing_allocation('none').must_equal 200
+        end
+      end
+
+      it 'should enable cluster allocation' do
+        @es_client = TestElasticsearch.new({ host: 'localhost', port: 9200 }, true)
+
+        @mock_http = Minitest::Mock.new
+        @mock_http.expect(:start, 200)
+
+        Net::HTTP.stub :new, @mock_http do
+          @es_client.cluster_routing_allocation('all').must_equal 200
+        end
+      end
+
+      it 'should shutdown the local node' do
+        @es_client = TestElasticsearch.new({ host: 'localhost', port: 9200 }, true)
+
+        Net::HTTP.stub :post_form, 200 do
+          @es_client.shutdown_local_node.must_equal 200
+        end
+      end
+
+      it 'start elasticsearch service' do
+        @es_client = TestElasticsearch.new({ host: 'localhost', port: 9200 }, true)
+
+        result = @es_client.start_elasticsearch_service('test')
+
+        result.must_equal 'echo test | sudo -S service elasticsearch start'
+      end
+
+      it 'start elasticsearch binary' do
+        @es_client = TestElasticsearch.new({ host: 'localhost', port: 9200 }, true)
+
+        result = @es_client.start_elasticsearch_binary('/path/to/elasticsearch/')
+
+        result.must_equal '/path/to/elasticsearch/bin/elasticsearch'
+      end
+
+      it 'start elasticsearch service from the deb extension' do
+        @mock = Minitest::Mock.new
+        @mock.expect(:extension, '.deb')
+        @mock.expect(:sudo_password, 'test')
+
+        @es_client = TestElasticsearch.new({ host: 'localhost', port: 9200 }, true)
+
+        result = @es_client.start_elasticsearch(@mock)
+
+        result.must_equal 'echo test | sudo -S service elasticsearch start'
+      end
+
+      it 'start elasticsearch service from the rpm extension' do
+        @mock = Minitest::Mock.new
+        @mock.expect(:extension, '.rpm')
+        @mock.expect(:sudo_password, 'test')
+
+        @es_client = TestElasticsearch.new({ host: 'localhost', port: 9200 }, true)
+
+        result = @es_client.start_elasticsearch(@mock)
+
+        result.must_equal 'echo test | sudo -S service elasticsearch start'
+      end
+    end
+  end
+end

data/test/elasticsearch_update/installer_spec.rb

@@ -0,0 +1,65 @@
+require 'minitest_helper'
+
+module ElasticsearchUpdate
+  # The TestInstaller class below tests the Downloader class from the library
+  class TestInstaller < Minitest::Test
+    describe 'Installer', 'Used to install Elasticsearch file' do
+
+      it 'should initialize and correctly assign values.' do
+        @installer = ElasticsearchUpdate::Installer.new('test_password', '.deb', false)
+        assert_kind_of ElasticsearchUpdate::Installer, @installer
+
+        @installer.sudo_password.must_equal 'test_password'
+        @installer.extension.must_equal '.deb'
+      end
+
+      it 'should install the .deb file' do
+        @mock_file = Minitest::Mock.new
+        @mock_file.expect(:path, '/path/to/file.deb')
+
+        Kernel.stub :system, true do
+          @installer = ElasticsearchUpdate::Installer.new('test_password', '.deb', true)
+          response = @installer.install_file(@mock_file)
+
+          response.must_equal true
+        end
+      end
+
+      it 'should install the .rpm file' do
+        @mock_file = Minitest::Mock.new
+        @mock_file.expect(:path, '/path/to/file.rpm')
+
+        Kernel.stub :system, true do
+          @installer = ElasticsearchUpdate::Installer.new('test_password', '.rpm', true)
+          response = @installer.install_file(@mock_file)
+
+          response.must_equal true
+        end
+      end
+
+      it 'should install the .zip file' do
+        @mock_file = Minitest::Mock.new
+        @mock_file.expect(:path, '/path/to/file.rpm')
+
+        Kernel.stub :system, true do
+          @installer = ElasticsearchUpdate::Installer.new('test_password', '.zip', true)
+          response = @installer.install_file(@mock_file)
+
+          response.must_equal true
+        end
+      end
+
+      it 'should install the .rpm file' do
+        @mock_file = Minitest::Mock.new
+        @mock_file.expect(:path, '/path/to/file.rpm')
+
+        Kernel.stub :system, true do
+          @installer = ElasticsearchUpdate::Installer.new('test_password', '.tar.gz', true)
+          response = @installer.install_file(@mock_file)
+
+          response.must_equal true
+        end
+      end
+    end
+  end
+end

data/test/elasticsearch_update/wizard_spec.rb

@@ -0,0 +1,94 @@
+require 'minitest_helper'
+
+module ElasticsearchUpdate
+  class TestWizard < Wizard
+    def ask(_question, _type = String, &_block)
+      'Question asked.'
+    end
+
+    def choose(&_block)
+      @choice = '.deb'
+    end
+  end
+  # The TestInstaller class below tests the Downloader class from the library
+  class TestWizard
+    describe 'Wizard', 'Used to ask questions on the command line' do
+
+      it 'should initialize without errors' do
+        wizard = TestWizard.new
+        assert_kind_of Wizard, wizard
+      end
+
+      it "should ask for Elasticsearch's host" do
+        wizard = TestWizard.new
+
+        result = wizard.host
+
+        result.must_equal 'Question asked.'
+      end
+
+      it 'should ask for the desired Elasticsearch version' do
+        wizard = TestWizard.new
+
+        result = wizard.version
+
+        result.must_equal 'Question asked.'
+      end
+
+      it "should ask for Elasticsearch's port" do
+        wizard = TestWizard.new
+
+        result = wizard.port
+
+        result.must_equal 'Question asked.'
+      end
+
+      it "should ask for Elasticsearch's location info" do
+        wizard = TestWizard.new
+
+        result = wizard.es_location_hash
+
+        result.must_equal(host: 'Question asked.', port: 'Question asked.')
+      end
+
+      it "should ask for Elasticsearch's file system location" do
+        wizard = TestWizard.new
+
+        result = wizard.elasticsearch_fs_location
+
+        result.must_equal 'Question asked.'
+      end
+
+      it "should ask for the system's sudo password" do
+        wizard = TestWizard.new
+
+        result = wizard.sudo_password
+
+        result.must_equal 'Question asked.'
+      end
+
+      it 'should ask for the desired extension of the update file' do
+        wizard = TestWizard.new
+
+        @mock_menu = Minitest::Mock.new
+        @mock_menu.expect(:prompt, true)
+        @mock_menu.expect(:choice, '.deb')
+
+        TestWizard.stub :choose, '.deb', @mock_menu do
+          result = wizard.extension
+          result.must_equal '.deb'
+        end
+      end
+
+      it 'should create the download hash' do
+        wizard = TestWizard.new
+
+        result = wizard.download_hash
+
+        result.must_equal(base_url: 'download.elasticsearch.org',
+                          version: 'Question asked.',
+                          extension: '.deb')
+      end
+    end
+  end
+end

data/test/minitest_helper.rb

@@ -0,0 +1,9 @@
+require 'codeclimate-test-reporter'
+CodeClimate::TestReporter.start
+require 'minitest/autorun'
+require 'elasticsearch_update'
+
+# ElasticsearchUpdate houses all tests for the
+# Elasticsearch Updater and all associated library files
+module ElasticsearchUpdate
+end

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: elasticsearch_update
+version: !ruby/object:Gem::Version
+  version: 1.0.1
+platform: ruby
+authors:
+- Kevin Kirsche
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-02-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 5.4.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 5.4.3
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.6.21
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.6.21
+description: Updates the elasticsearch instance from a deb or rpm on the local machine.
+  Assumes elasticsearch is a service.
+email:
+- kev.kirsche@gmail.com
+executables:
+- elasticsearch_update
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/elasticsearch_update
+- bin/update.rb
+- elasticsearch_update.gemspec
+- lib/elasticsearch_update.rb
+- lib/elasticsearch_update/cli.rb
+- lib/elasticsearch_update/downloader.rb
+- lib/elasticsearch_update/elasticsearch.rb
+- lib/elasticsearch_update/installer.rb
+- lib/elasticsearch_update/version.rb
+- lib/elasticsearch_update/wizard.rb
+- test/elasticsearch_update/downloader_spec.rb
+- test/elasticsearch_update/elasticsearch_spec.rb
+- test/elasticsearch_update/installer_spec.rb
+- test/elasticsearch_update/wizard_spec.rb
+- test/minitest_helper.rb
+homepage: https://github.com/kkirsche/elasticsearch_update
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Updates the elasticsearch instance on the local machine.
+test_files:
+- test/elasticsearch_update/downloader_spec.rb
+- test/elasticsearch_update/elasticsearch_spec.rb
+- test/elasticsearch_update/installer_spec.rb
+- test/elasticsearch_update/wizard_spec.rb
+- test/minitest_helper.rb