elasticsearch-watcher 0.0.1

data/lib/elasticsearch/watcher/api/actions/execute_watch.rb

@@ -0,0 +1,27 @@
+module Elasticsearch
+  module API
+    module Watcher
+      module Actions
+
+        # Force the execution of the watch actions (eg. for testing)
+        #
+        # @option arguments [String] :id Watch ID (*Required*)
+        # @option arguments [Hash] :body Execution control
+        #
+        # @see http://www.elastic.co/guide/en/watcher/current/appendix-api-execute-watch.html
+        #
+        def execute_watch(arguments={})
+          raise ArgumentError, "Required argument 'id' missing" unless arguments[:id]
+          valid_params = [
+             ]
+          method = 'PUT'
+          path   = "_watcher/watch/#{arguments[:id]}/_execute"
+          params = {}
+          body   = arguments[:body]
+
+          perform_request(method, path, params, body).body
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/watcher/api/actions/get_watch.rb

@@ -0,0 +1,31 @@
+module Elasticsearch
+  module API
+    module Watcher
+      module Actions
+
+        # Get a specific watch
+        #
+        # @option arguments [String] :id Watch ID (*Required*)
+        #
+        # @see http://www.elastic.co/guide/en/watcher/current/appendix-api-get-watch.html
+        #
+        def get_watch(arguments={})
+          raise ArgumentError, "Required argument 'id' missing" unless arguments[:id]
+          valid_params = [
+             ]
+          method = 'GET'
+          path   = "_watcher/watch/#{arguments[:id]}"
+          params = {}
+          body   = nil
+
+          perform_request(method, path, params, body).body
+        rescue Exception => e
+          # NOTE: Use exception name, not full class in Elasticsearch::Client to allow client plugability
+          if Array(arguments[:ignore]).include?(404) && e.class.to_s =~ /NotFound/; false
+          else raise(e)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/watcher/api/actions/info.rb

@@ -0,0 +1,23 @@
+module Elasticsearch
+  module API
+    module Watcher
+      module Actions
+
+        # Return information about the installed Watcher plugin
+        #
+        # @see http://www.elastic.co/guide/en/watcher/current/appendix-api-info.html
+        #
+        def info(arguments={})
+          valid_params = [
+             ]
+          method = 'GET'
+          path   = "/_watcher/"
+          params = {}
+          body   = nil
+
+          perform_request(method, path, params, body).body
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/watcher/api/actions/put_watch.rb

@@ -0,0 +1,29 @@
+module Elasticsearch
+  module API
+    module Watcher
+      module Actions
+
+        # Create a new watch or update an existing one
+        #
+        # @option arguments [String] :id Watch ID (*Required*)
+        # @option arguments [Hash] :body The watch (*Required*)
+        # @option arguments [Boolean] :pretty Pretty the output
+        #
+        # @see http://www.elastic.co/guide/en/watcher/current/appendix-api-put-watch.html
+        #
+        def put_watch(arguments={})
+          raise ArgumentError, "Required argument 'id' missing" unless arguments[:id]
+          raise ArgumentError, "Required argument 'body' missing" unless arguments[:body]
+          valid_params = [
+            :master_timeout ]
+          method = 'PUT'
+          path   = "_watcher/watch/#{arguments[:id]}"
+          params = Utils.__validate_and_extract_params arguments, valid_params
+          body   = arguments[:body]
+
+          perform_request(method, path, params, body).body
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/watcher/api/actions/restart.rb

@@ -0,0 +1,24 @@
+module Elasticsearch
+  module API
+    module Watcher
+      module Actions
+
+        # Restart the watcher service
+        #
+        #
+        # @see http://www.elastic.co/guide/en/watcher/current/appendix-api-service.html
+        #
+        def restart(arguments={})
+          valid_params = [
+             ]
+          method = 'PUT'
+          path   = "/_watcher/_restart"
+          params = {}
+          body   = nil
+
+          perform_request(method, path, params, body).body
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/watcher/api/actions/start.rb

@@ -0,0 +1,23 @@
+module Elasticsearch
+  module API
+    module Watcher
+      module Actions
+
+        # Start the watcher service
+        #
+        # @see http://www.elastic.co/guide/en/watcher/current/appendix-api-service.html
+        #
+        def start(arguments={})
+          valid_params = [
+             ]
+          method = 'PUT'
+          path   = "/_watcher/_start"
+          params = {}
+          body   = nil
+
+          perform_request(method, path, params, body).body
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/watcher/api/actions/stats.rb

@@ -0,0 +1,23 @@
+module Elasticsearch
+  module API
+    module Watcher
+      module Actions
+
+        # Return statistical information about the watcher service
+        #
+        # @see http://www.elastic.co/guide/en/watcher/current/appendix-api-stats.html
+        #
+        def stats(arguments={})
+          valid_params = [
+             ]
+          method = 'GET'
+          path   = "/_watcher/stats"
+          params = {}
+          body   = nil
+
+          perform_request(method, path, params, body).body
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/watcher/api/actions/stop.rb

@@ -0,0 +1,23 @@
+module Elasticsearch
+  module API
+    module Watcher
+      module Actions
+
+        # Stop the watcher service
+        #
+        # @see http://www.elastic.co/guide/en/watcher/current/appendix-api-service.html
+        #
+        def stop(arguments={})
+          valid_params = [
+             ]
+          method = 'PUT'
+          path   = "/_watcher/_stop"
+          params = {}
+          body   = nil
+
+          perform_request(method, path, params, body).body
+        end
+      end
+    end
+  end
+end

data/lib/elasticsearch/watcher/version.rb

@@ -0,0 +1,5 @@
+module Elasticsearch
+  module Watcher
+    VERSION = "0.0.1"
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,87 @@
+JRUBY = defined?(JRUBY_VERSION)
+
+if ENV['COVERAGE'] || ENV['CI']
+  require 'simplecov'
+  SimpleCov.start { add_filter "/test|test_" }
+end
+
+at_exit { Elasticsearch::Test::IntegrationTestCase.__run_at_exit_hooks }
+
+require 'test/unit'
+require 'shoulda-context'
+require 'mocha/setup'
+
+require 'minitest/reporters'
+Minitest::Reporters.use! Minitest::Reporters::SpecReporter.new
+
+require 'elasticsearch'
+require 'elasticsearch/extensions/test/cluster'
+require 'elasticsearch/extensions/test/startup_shutdown'
+
+require 'elasticsearch/watcher'
+
+module Elasticsearch
+  module Test
+    class IntegrationTestCase < ::Test::Unit::TestCase
+      include Elasticsearch::Extensions::Test
+      extend  StartupShutdown
+
+      startup do
+        Cluster.start(nodes: 1) if ENV['SERVER'] \
+                                && ! Elasticsearch::Extensions::Test::Cluster.running?
+      end
+
+      shutdown do
+        Cluster.stop if ENV['SERVER'] \
+                     && started?      \
+                     && Elasticsearch::Extensions::Test::Cluster.running?
+      end
+
+      def setup
+        @port = (ENV['TEST_CLUSTER_PORT'] || 9250).to_i
+
+        @logger =  Logger.new(STDERR)
+        @logger.formatter = proc do |severity, datetime, progname, msg|
+          color = case severity
+            when /INFO/ then :green
+            when /ERROR|WARN|FATAL/ then :red
+            when /DEBUG/ then :cyan
+            else :white
+          end
+          ANSI.ansi(severity[0] + ' ', color, :faint) + ANSI.ansi(msg, :white, :faint) + "\n"
+        end
+
+        @client = Elasticsearch::Client.new host: "localhost:#{@port}", logger: @logger
+      end
+
+      def teardown
+        @client.indices.delete index: '_all'
+      end
+    end
+  end
+end
+
+module Elasticsearch
+  module Test
+    class FakeClient
+      include Elasticsearch::API::Watcher
+
+      def perform_request(method, path, params, body)
+        puts "PERFORMING REQUEST:", "--> #{method.to_s.upcase} #{path} #{params} #{body}"
+        FakeResponse.new(200, 'FAKE', {})
+      end
+    end
+
+    FakeResponse = Struct.new(:status, :body, :headers) do
+      def status
+        values[0] || 200
+      end
+      def body
+        values[1] || '{}'
+      end
+      def headers
+        values[2] || {}
+      end
+    end
+  end
+end

data/test/unit/ack_watch_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class WatcherAckWatchTest < ::Test::Unit::TestCase
+
+      context "Watcher: Ack watch" do
+        subject { FakeClient.new }
+
+        should "perform correct request" do
+          subject.expects(:perform_request).with do |method, url, params, body|
+            assert_equal 'PUT', method
+            assert_equal '_watcher/watch/foo/_ack', url
+            assert_equal Hash.new, params
+            assert_nil   body
+            true
+          end.returns(FakeResponse.new)
+
+          subject.watcher.ack_watch id: 'foo'
+        end
+
+      end
+
+    end
+  end
+end

data/test/unit/delete_watch_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class WatcherDeleteWatchTest < ::Test::Unit::TestCase
+
+      context "Watcher: Delete watch" do
+        subject { FakeClient.new }
+
+        should "perform correct request" do
+          subject.expects(:perform_request).with do |method, url, params, body|
+            assert_equal 'DELETE', method
+            assert_equal "_watcher/watch/foo", url
+            assert_equal Hash.new, params
+            assert_nil   body
+            true
+          end.returns(FakeResponse.new)
+
+          subject.watcher.delete_watch id: 'foo'
+        end
+
+      end
+
+    end
+  end
+end

data/test/unit/execute_watch_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class WatcherExecuteWatchTest < ::Test::Unit::TestCase
+
+      context "Watcher: Execute watch" do
+        subject { FakeClient.new }
+
+        should "perform correct request" do
+          subject.expects(:perform_request).with do |method, url, params, body|
+            assert_equal 'PUT', method
+            assert_equal "_watcher/watch/foo/_execute", url
+            assert_equal Hash.new, params
+            assert_equal Hash.new, body
+            true
+          end.returns(FakeResponse.new)
+
+          subject.watcher.execute_watch id: 'foo', body: {}
+        end
+
+      end
+
+    end
+  end
+end

data/test/unit/get_watch_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class WatcherGetWatchTest < ::Test::Unit::TestCase
+
+      context "Watcher: Get watch" do
+        subject { FakeClient.new }
+
+        should "perform correct request" do
+          subject.expects(:perform_request).with do |method, url, params, body|
+            assert_equal 'GET', method
+            assert_equal "_watcher/watch/foo", url
+            assert_equal Hash.new, params
+            assert_nil   body
+            true
+          end.returns(FakeResponse.new)
+
+          subject.watcher.get_watch id: 'foo'
+        end
+
+      end
+
+    end
+  end
+end

data/test/unit/info_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class WatcherInfoTest < ::Test::Unit::TestCase
+
+      context "Watcher: Info" do
+        subject { FakeClient.new }
+
+        should "perform correct request" do
+          subject.expects(:perform_request).with do |method, url, params, body|
+            assert_equal 'GET', method
+            assert_equal '/_watcher/', url
+            assert_equal Hash.new, params
+            assert_nil   body
+            true
+          end.returns(FakeResponse.new)
+
+          subject.watcher.info
+        end
+
+      end
+
+    end
+  end
+end