elasticsearch-transport 7.1.0 → 7.13.3

data/test/unit/connection_selector_test.rb

@@ -1,81 +0,0 @@
-# Licensed to Elasticsearch B.V. under one or more contributor
-# license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright
-# ownership. Elasticsearch B.V. licenses this file to you under
-# the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#	http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied.  See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-require 'test_helper'
-
-class Elasticsearch::Transport::Transport::Connections::SelectorTest < Minitest::Test
-  include Elasticsearch::Transport::Transport::Connections::Selector
-
-  class DummyStrategySelector
-    include Elasticsearch::Transport::Transport::Connections::Selector::Base
-  end
-
-  class BackupStrategySelector
-    include Elasticsearch::Transport::Transport::Connections::Selector::Base
-
-    def select(options={})
-      connections.reject do |c|
-        c.host[:attributes] && c.host[:attributes][:backup]
-      end.send( defined?(RUBY_VERSION) && RUBY_VERSION > '1.9' ? :sample : :choice)
-    end
-  end
-
-  context "Connection selector" do
-
-    should "be initialized with connections" do
-      assert_equal [1, 2], Random.new(:connections => [1, 2]).connections
-    end
-
-    should "have the abstract select method" do
-      assert_raise(NoMethodError) { DummyStrategySelector.new.select }
-    end
-
-    context "in random strategy" do
-      setup do
-        @selector = Random.new :connections => ['A', 'B', 'C']
-      end
-
-      should "pick a connection" do
-        assert_not_nil @selector.select
-      end
-    end
-
-    context "in round-robin strategy" do
-      setup do
-        @selector = RoundRobin.new :connections => ['A', 'B', 'C']
-      end
-
-      should "rotate over connections" do
-        assert_equal 'A', @selector.select
-        assert_equal 'B', @selector.select
-        assert_equal 'C', @selector.select
-        assert_equal 'A', @selector.select
-      end
-    end
-
-    context "with a custom strategy" do
-
-      should "return proper connection" do
-        selector = BackupStrategySelector.new :connections => [ stub(:host => { :hostname => 'host1' }),
-                                                                stub(:host => { :hostname => 'host2', :attributes => { :backup => true }}) ]
-        10.times { assert_equal 'host1', selector.select.host[:hostname] }
-      end
-
-    end
-
-  end
-end