cluster_test.rb in elasticsearch-extensions-0.0.27

- old
+ new

@@ -1,10 +1,10 @@
 require 'test_helper'
 
 require 'elasticsearch/extensions/test/cluster'
 
-class Elasticsearch::Extensions::TestClusterTest < Test::Unit::TestCase
+class Elasticsearch::Extensions::TestClusterTest < Elasticsearch::Test::UnitTestCase
   include Elasticsearch::Extensions::Test
   context "The Test::Cluster" do
     context "module" do
       should "delegate the methods to the class" do
         Cluster::Cluster
@@ -267,9 +267,14 @@
 
           IO.any_instance.expects(:read)
             .returns('Version: 2.3.0-SNAPSHOT, Build: d1c86b0/2016-03-30T10:43:20Z, JVM: 1.8.0_60')
 
           assert_equal '2.0', @subject.__determine_version
+        end
+
+        should "return version from arguments" do
+          cluster = Elasticsearch::Extensions::Test::Cluster::Cluster.new command: '/foo/bar/bin/elasticsearch', version: '5.2'
+          assert_equal '5.0', cluster.__determine_version
         end
 
         should "raise an exception when the version cannot be parsed from .jar" do
           # Incorrect jar version (no dots)
           File.expects(:exist?).with('/foo/bar/bin/../lib/').returns(true)