ruby-druid 0.1.1

data/spec/lib/zoo_handler_spec.rb

@@ -0,0 +1,200 @@
+require "spec_helper"
+
+module ZK
+  def self.new(uri, opts = {})
+    Mock.new uri, opts
+  end
+
+  class Mock
+    include RSpec::Matchers
+
+    def initialize(uri, opts)
+      uri.should == 'test-uri'
+      opts.should == { :chroot => :check }
+      @registrations = {}
+      @paths = {
+        '/disco' => ['a', 'b'],
+        '/disco/a' => ['b1', 'm1'],
+        '/disco/b' => ['b2', 'm2']
+      }
+      @unregisters = []
+    end
+
+    def register(path, opts, &block)
+      opts.should == { :only => :child }
+      @registrations[path].should == nil
+
+      zk = self
+      block.define_singleton_method :unregister do
+        zk.instance_variable_get('@unregisters').push path
+        zk.instance_variable_get('@registrations').delete path
+      end
+      @registrations[path] = block
+    end
+
+    def unregistrations
+      result = @unregisters
+      @unregisters = []
+      result
+    end
+
+    def children(path, opts)
+      @registrations[path].should be_a(Proc)
+
+      value = @paths[path]
+      throw "no mock code for #{path}" unless value
+      value
+    end
+
+    def get(path)
+      case path
+      when '/disco/a/b1'
+        [{
+          :address => 'b1_address',
+          :port => 80
+        }.to_json]
+      when '/disco/a/m1'
+        [{
+          :address => 'm1_address',
+          :port => 81
+        }.to_json]
+      when '/disco/b/b2'
+        [{
+          :address => 'b2_address',
+          :port => 90
+        }.to_json]
+      when '/disco/b/m2'
+        [{
+          :address => 'm2_address',
+          :port => 85
+        }.to_json]
+      when '/disco/a/b3'
+        [{
+          :address => 'b3_address',
+          :port => 83
+        }.to_json]
+      else
+        throw "no mock code for #{path}"
+      end
+    end
+
+    def change(uri, values)
+      @paths[uri] = values
+      @registrations[uri].call if @registrations[uri]
+    end
+
+    def on_expired_session
+    end
+  end
+end
+
+class RestClientResponseMock
+  def initialize(code, value)
+    @code = code
+    @value = value
+  end
+
+  def code
+    @code
+  end
+
+  def to_str
+    @value.to_str
+  end
+end
+
+describe Druid::ZooHandler do
+  it 'reports services and data sources correctly' do
+    calls = []
+    RestClient::Request.stub(:execute) do |opts|
+      uri_match = opts[:url].match /^http:\/\/(.+)_address:(.+)\/druid\/v2\/datasources\/$/
+
+      host = uri_match[1]
+      port = uri_match[2].to_i
+
+      calls.push [host, port]
+
+      case host
+      when 'b1'
+        RestClientResponseMock.new(200, ['s1','s2'].to_json)
+      when 'b2'
+        RestClientResponseMock.new(200, ['s3','s4'].to_json)
+      when 'b3'
+        RestClientResponseMock.new(200, ['s5','s6'].to_json)
+      else
+        RestClientResponseMock.new(404, nil)
+      end
+    end
+
+    zk = Druid::ZooHandler.new 'test-uri', :discovery_path => '/disco'
+
+    calls.should == [
+      ['b1', 80],
+      ['m1', 81],
+      ['b2', 90],
+      ['m2', 85]
+    ]
+    zk.services.should == ['a', 'b']
+    zk.data_sources.should == {
+      'a/s1' => 'http://b1_address:80/druid/v2/',
+      'a/s2' => 'http://b1_address:80/druid/v2/',
+      'b/s3' => 'http://b2_address:90/druid/v2/',
+      'b/s4' => 'http://b2_address:90/druid/v2/'
+    }
+
+    calls = []
+    mock = zk.instance_variable_get('@zk')
+    mock.unregistrations.should == []
+
+    # unregister a whole service
+    mock.change '/disco', ['a']
+    calls.should == []
+    zk.services.should == ['a']
+    zk.data_sources.should == {
+      'a/s1' => 'http://b1_address:80/druid/v2/',
+      'a/s2' => 'http://b1_address:80/druid/v2/'
+    }
+    mock.unregistrations.should == ['/disco/b']
+    # register it again
+    mock.change '/disco', ['a', 'b']
+    calls.should == [
+      ['b2', 90],
+      ['m2', 85]
+    ]
+    zk.services.should == ['a', 'b']
+    zk.data_sources.should == {
+      'a/s1' => 'http://b1_address:80/druid/v2/',
+      'a/s2' => 'http://b1_address:80/druid/v2/',
+      'b/s3' => 'http://b2_address:90/druid/v2/',
+      'b/s4' => 'http://b2_address:90/druid/v2/'
+    }
+    mock.unregistrations.should == []
+
+    #register a new broker
+    calls = []
+    mock.change '/disco/a', ['b1', 'b3']
+    calls.should == [['b3', 83]]
+    zk.services.should == ['a', 'b']
+    zk.data_sources.should == {
+      "a/s1" => "http://b1_address:80/druid/v2/",
+      "a/s2" => "http://b1_address:80/druid/v2/",
+      "b/s3" => "http://b2_address:90/druid/v2/",
+      "b/s4" => "http://b2_address:90/druid/v2/",
+      "a/s5" => "http://b3_address:83/druid/v2/",
+      "a/s6" => "http://b3_address:83/druid/v2/"
+    }
+    mock.unregistrations.should == ['/disco/a']
+    # unregister it
+    calls = []
+    mock.change '/disco/a', ['b1']
+    calls.should == []
+    zk.services.should == ['a', 'b']
+    zk.data_sources.should == {
+      'a/s1' => 'http://b1_address:80/druid/v2/',
+      'a/s2' => 'http://b1_address:80/druid/v2/',
+      'b/s3' => 'http://b2_address:90/druid/v2/',
+      'b/s4' => 'http://b2_address:90/druid/v2/'
+    }
+    mock.unregistrations.should == ['/disco/a']
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,2 @@
+require "druid"
+require 'webmock/rspec'

metadata

@@ -0,0 +1,96 @@
+--- !ruby/object:Gem::Specification
+name: ruby-druid
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- The LiquidM Team
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-08-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: zk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Ruby client for metamx druid
+email: tech@liquidm.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Guardfile
+- LICENSE
+- README.md
+- Rakefile
+- bin/dripl
+- dot_driplrc_example
+- lib/druid.rb
+- lib/druid/client.rb
+- lib/druid/console.rb
+- lib/druid/filter.rb
+- lib/druid/having.rb
+- lib/druid/post_aggregation.rb
+- lib/druid/query.rb
+- lib/druid/response_row.rb
+- lib/druid/zoo_handler.rb
+- ruby-druid.gemspec
+- spec/lib/client_spec.rb
+- spec/lib/query_spec.rb
+- spec/lib/zoo_handler_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/madvertise/ruby-druid
+licenses: []
+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.0.5
+signing_key: 
+specification_version: 4
+summary: Ruby client for druid
+test_files:
+- spec/lib/client_spec.rb
+- spec/lib/query_spec.rb
+- spec/lib/zoo_handler_spec.rb
+- spec/spec_helper.rb