drbqs 0.0.14 → 0.0.15

data/spec/node/task_client_spec.rb

@@ -61,6 +61,21 @@ describe DRbQS::Node::TaskClient do
     end
   end
 
+  context "when getting new task" do
+    before(:all) do
+      init_task_client(7)
+    end
+
+    it "should return nil" do
+      subject.add_new_task.should be_nil
+    end
+
+    it "should return nil" do
+      add_task_to_tuplespace(100, [[1, 3, 5, 7], :size, []])
+      subject.add_new_task.should be_true
+    end
+  end
+
   context "when adding new task" do
     before(:all) do
       init_task_client(7)

data/spec/server/history_spec.rb

@@ -11,59 +11,56 @@ describe DRbQS::Server::History do
     end
   end
 
+  subject do
+    DRbQS::Server::History.new
+  end
+
   context "when setting events" do
     before(:all) do
-      @history = DRbQS::Server::History.new
-      @history.set(1, :abc)
-      @history.set(1, :def)
-      @history.set(2, 'ABC', 'DEF')
-      @history.set(3, :ghi)
-      @history.set(1, :jkl)
-      @history.set(2, 123, 456)
+      subject.set(1, :abc)
+      subject.set(1, :def)
+      subject.set(2, 'ABC', 'DEF')
+      subject.set(3, :ghi)
+      subject.set(1, :jkl)
+      subject.set(2, 123, 456)
     end
 
     it "should have 3 items." do
-      @history.should have(3).items
+      subject.should have(3).items
     end
 
     it "should 3 events of ID 1." do
-      @history.number_of_events(1).should == 3
+      subject.number_of_events(1).should == 3
     end
 
     it "should 2 events of ID 2" do
-      @history.number_of_events(2).should == 2
+      subject.number_of_events(2).should == 2
     end
 
     it "should 1 event of ID 3" do
-      @history.number_of_events(3).should == 1
+      subject.number_of_events(3).should == 1
     end
 
     it "should check events of ID 1" do
-      events = @history.events(1)
+      events = subject.events(1)
       check_event(events[0], :abc)
       check_event(events[1], :def)
       check_event(events[2], :jkl)
     end
 
     it "should check events of ID 2" do
-      events = @history.events(2)
+      events = subject.events(2)
       check_event(events[0], 'ABC', 'DEF')
       check_event(events[1], 123, 456)
     end
 
     it "should check events of ID 1" do
-      events = @history.events(3)
+      events = subject.events(3)
       check_event(events[0], :ghi)
     end
-
-    it "should return strings of log" do
-      @history.log_strings.should be_an_instance_of String
-    end
   end
 
   context "when executing a method" do
-    subject { DRbQS::Server::History.new }
-
     it "should add new event" do
       id = 1
       subject.set(id, :connect)
@@ -88,3 +85,37 @@ describe DRbQS::Server::History do
     end
   end
 end
+
+describe DRbQS::Server::TaskHistory do
+  subject do
+    DRbQS::Server::TaskHistory.new
+  end
+
+  it "should return strings of log" do
+    subject.set(1, :def)
+    subject.set(2, 'ABC', 'DEF')
+    subject.set(3, :ghi)
+    subject.set(1, :jkl)
+    subject.log_strings.should be_an_instance_of String
+  end
+
+  it "should return zero for finished_task_number." do
+    subject.finished_task_number.should == 0
+  end
+
+  it "should count finished task when setting :result." do
+    5.times do |i|
+      lambda do
+        subject.set(i, :result)
+      end.should change(subject, :finished_task_number).by(1)
+    end
+  end
+
+  it "should not change finished task number for :add." do
+    [:add, :requeue, :hook, :calculate].each do |sym|
+      lambda do
+        subject.set(1, sym)
+      end.should_not change(subject, :finished_task_number)
+    end
+  end
+end

data/spec/server/message_spec.rb

@@ -21,7 +21,7 @@ describe DRbQS::Server::Message do
 
   def clear_all_node
     @message_server.get_all_nodes.each do |id_num, id_str|
-      @node_list.delete(id_num)
+      @node_list.delete(id_num, :disconnect)
     end
   end
 
@@ -65,10 +65,15 @@ describe DRbQS::Server::Message do
     it "should get :node_error message" do
       node_id = 74
       @message.write([:server, :node_error, [node_id, 'Error occurs.']])
-      @node_list.should_receive(:delete).with(node_id)
+      @node_list.should_receive(:delete).with(node_id, :error)
       @message_server.get_message.should == [:node_error, node_id]
     end
 
+    it "should get :new_data message" do
+      @message.write([:server, :new_data, 'data string'])
+      @message_server.get_message.should == [:new_data, 'data string']
+    end
+
     it "should return nil for invalid message" do
       @message.write([:server, :invalid_message])
       @message_server.get_message.should be_nil

data/spec/server/node_list_spec.rb

@@ -58,7 +58,7 @@ describe DRbQS::Server::NodeList do
 
     it "should delete a node" do
       id = subject.get_new_id(create_id_string)
-      subject.delete(id)
+      subject.delete(id, :disconnect)
       subject.exist?(id).should_not be_true
     end
   end

data/spec/server/queue_spec.rb

@@ -38,6 +38,10 @@ describe DRbQS::Server::Queue do
       subject.calculating_task_number.should == 0
     end
 
+    it "should get number of stocked tasks." do
+      subject.stocked_task_number.should == 0
+    end
+
     it "should be empty." do
       subject.should be_empty
     end
@@ -45,6 +49,10 @@ describe DRbQS::Server::Queue do
     it "should be finished." do
       subject.should be_finished
     end
+
+    it "should return an array of calculating nodes." do
+      subject.calculating_nodes.should == []
+    end
   end
 
   context "when adding a task" do
@@ -65,6 +73,10 @@ describe DRbQS::Server::Queue do
       subject.calculating_task_number.should == 0
     end
 
+    it "should get number of stocked tasks." do
+      subject.stocked_task_number.should == 1
+    end
+
     it "should not be empty." do
       subject.empty?.should be_false
     end
@@ -76,6 +88,10 @@ describe DRbQS::Server::Queue do
     it "should take objects from queue." do
       @ts[:queue].take([nil, nil, nil, nil]).should be_true
     end
+
+    it "should return an array of calculating nodes." do
+      subject.calculating_nodes.should == []
+    end
   end
 
   context "when getting accept signal" do
@@ -83,7 +99,7 @@ describe DRbQS::Server::Queue do
       object_init
       @node_id = 100
       @task_id = subject.add(@task[:obj])
-      @ts[:result].write([:accept, @task_id, 100])
+      @ts[:result].write([:accept, @task_id, @node_id])
     end
 
     it "should return number of acceptances of signals." do
@@ -98,6 +114,10 @@ describe DRbQS::Server::Queue do
       subject.calculating_task_number.should == 1
     end
 
+    it "should get number of stocked tasks." do
+      subject.stocked_task_number.should == 0
+    end
+
     it "should be empty." do
       subject.should be_empty
     end
@@ -109,6 +129,10 @@ describe DRbQS::Server::Queue do
     it "should take objects from queue." do
       @ts[:result].read_all([nil, nil, nil]).size.should == 0
     end
+
+    it "should return an array of calculating nodes." do
+      subject.calculating_nodes.should == [@node_id]
+    end
   end
 
   context "when getting result" do
@@ -116,7 +140,7 @@ describe DRbQS::Server::Queue do
       object_init
       @node_id = 100
       @task_id = subject.add(@task[:obj])
-      @ts[:result].write([:accept, @task_id, 100])
+      @ts[:result].write([:accept, @task_id, @node_id])
       subject.get_accept_signal
       @ts[:result].write([:result, @task_id, @node_id, :result_object])
     end
@@ -133,6 +157,10 @@ describe DRbQS::Server::Queue do
       subject.calculating_task_number.should == 0
     end
 
+    it "should get number of stocked tasks." do
+      subject.stocked_task_number.should == 0
+    end
+
     it "should be empty." do
       subject.should be_empty
     end
@@ -140,6 +168,10 @@ describe DRbQS::Server::Queue do
     it "should not be finished." do
       subject.should be_finished
     end
+
+    it "should return an array of calculating nodes." do
+      subject.calculating_nodes.should == []
+    end
   end
 
   context "when requeueing a task" do
@@ -147,7 +179,7 @@ describe DRbQS::Server::Queue do
       object_init
       @node_id = 100
       @task_id = subject.add(@task[:obj])
-      @ts[:result].write([:accept, @task_id, 100])
+      @ts[:result].write([:accept, @task_id, @node_id])
       subject.get_accept_signal
       subject.requeue_for_deleted_node_id([@node_id])
     end
@@ -164,6 +196,10 @@ describe DRbQS::Server::Queue do
       subject.calculating_task_number.should == 0
     end
 
+    it "should get number of stocked tasks." do
+      subject.stocked_task_number.should == 1
+    end
+
     it "should be empty." do
       subject.should_not be_empty
     end
@@ -176,6 +212,54 @@ describe DRbQS::Server::Queue do
       (ary = @ts[:queue].take([nil, nil, nil, nil])).should be_true
       ary[0].should == @task_id
     end
+
+    it "should return an array of calculating nodes." do
+      subject.calculating_nodes.should == []
+    end
+  end
+
+  context "when some nodes are calculating" do
+    before(:all) do
+      object_init
+      @node_ids = [200, 300]
+      @tasks = [DRbQS::Task.new([1, 2, 3], :size, []),
+                DRbQS::Task.new([-1, -2], :size, []),
+                DRbQS::Task.new([8, 9, 10, 11], :size, [])]
+      @task_ids = @tasks.map do |t|
+        subject.add(t)
+      end
+      @ts[:result].write([:accept, @task_ids[0], @node_ids[0]])
+      @ts[:result].write([:accept, @task_ids[1], @node_ids[1]])
+      subject.get_accept_signal
+    end
+
+    it "should return number of acceptances of signals." do
+      subject.get_accept_signal.should == 0
+    end
+
+    it "should return number of acceptances of resultss." do
+      subject.get_result(@server_dummy).should == 0
+    end
+
+    it "should return number of calculating task." do
+      subject.calculating_task_number.should == 2
+    end
+
+    it "should get number of stocked tasks." do
+      subject.stocked_task_number.should == 1
+    end
+
+    it "should be empty." do
+      subject.should_not be_empty
+    end
+
+    it "should not be finished." do
+      subject.should_not be_finished
+    end
+
+    it "should return an array of calculating nodes." do
+      subject.calculating_nodes.should == @node_ids.sort
+    end
   end
 
   context "when executing hook of task" do
@@ -189,9 +273,10 @@ describe DRbQS::Server::Queue do
     end
 
     it "should execute hook of task" do
+      @node_id = 100
       task_id = subject.add(@task[:obj])
       @task_id = subject.add(@task[:obj])
-      @ts[:result].write([:accept, @task_id, 100])
+      @ts[:result].write([:accept, @task_id, @node_id])
       subject.get_accept_signal
       @ts[:result].write([:result, @task_id, @node_id, :result_object])
       subject.get_result(@server_dummy)
@@ -227,6 +312,10 @@ describe DRbQS::Server::Queue do
       subject.calculating_task_number.should == 1
     end
 
+    it "should get number of stocked tasks." do
+      subject.stocked_task_number.should == 1
+    end
+
     it "should be empty." do
       subject.should_not be_empty
     end

data/spec/server/server_hook_spec.rb

@@ -5,6 +5,10 @@ require 'drbqs/server/server_hook'
 describe DRbQS::Server::Hook do
   subject { DRbQS::Server::Hook.new }
 
+  it "should return 0 for nonexistent key." do
+    subject.number_of_hook(:nonexistent_key).should == 0
+  end
+
   it "should add a hook with automatic creation of name." do
     n = subject.number_of_hook(:finish)
     name = subject.add(:finish) do |server|
@@ -75,4 +79,62 @@ describe DRbQS::Server::Hook do
     subject.exec(:finish)
     execute.should be_true
   end
+
+  context "when checking block before executing hook" do
+    it "should execute hooks." do
+      exec_flag = {}
+      subject.add(:finish) do |server|
+        exec_flag[:first] = true
+      end
+      subject.add(:finish) do |server|
+        exec_flag[:second] = true
+      end
+      subject.exec(:finish) do |name|
+        name.should be_an_instance_of String
+        true
+      end
+      exec_flag[:first].should be_true
+      exec_flag[:second].should be_true
+    end
+
+    it "should execute finish_exit that is special proc." do
+      execute = nil
+      subject.set_finish_exit do
+        execute = true
+      end
+      subject.exec(:finish) do |name|
+        name.should be_an_instance_of String
+        true
+      end
+      execute.should be_true
+    end
+
+    it "should not execute hooks." do
+      exec_flag = {}
+      subject.add(:finish) do |server|
+        exec_flag[:first] = true
+      end
+      subject.add(:finish) do |server|
+        exec_flag[:second] = true
+      end
+      subject.exec(:finish) do |name|
+        name.should be_an_instance_of String
+        false
+      end
+      exec_flag[:first].should be_nil
+      exec_flag[:second].should be_nil
+    end
+
+    it "should not execute finish_exit that is special proc." do
+      execute = nil
+      subject.set_finish_exit do
+        execute = true
+      end
+      subject.exec(:finish) do |name|
+        name.should be_an_instance_of String
+        false
+      end
+      execute.should be_nil
+    end
+  end
 end

data/spec/setting/base_spec.rb

@@ -0,0 +1,35 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe DRbQS::Setting::Base do
+  it "should get default registered options." do
+    obj = DRbQS::Setting::Base.new
+    obj.source.registered_keys.sort.should == [:debug].sort
+  end
+
+  it "should register built-in options." do
+    obj = DRbQS::Setting::Base.new(:all_keys_defined => true, :log_level => true, :daemon => true)
+    obj.source.registered_keys.sort.should == [:debug, :log_level, :daemon].sort
+  end
+
+  it "should return string for shell." do
+    obj = DRbQS::Setting::Base.new(:all_keys_defined => true, :log_level => true, :daemon => true)
+    obj.value.log_level 'debug'
+    obj.value.daemon '/path/to/log'
+    obj.parse!
+    str = obj.string_for_shell
+    str.should match(/--log-level/)
+    str.should match(/--daemon/)
+    str.should match(/debug/)
+    str.should match(/\/path\/to\/log/)
+  end
+
+  it "should not change when executing parse! twice." do
+    obj = DRbQS::Setting::Base.new(:all_keys_defined => true, :log_level => true, :daemon => true)
+    obj.value.log_level 'debug'
+    obj.value.daemon '/path/to/log'
+    obj.parse!
+    lambda do
+      obj.parse!
+    end.should_not change(obj, :string_for_shell)
+  end
+end

data/spec/setting/data_container_spec.rb

@@ -0,0 +1,92 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+require 'drbqs/setting/setting'
+
+describe DRbQS::Setting::Source::DataContainer do
+  subject do
+    DRbQS::Setting::Source::DataContainer.new(Array)
+  end
+
+  context "when setting values" do
+    it "should set a value." do
+      subject.one_argument 3
+      subject.__data__[:one_argument].should == [3]
+    end
+
+    it "should set values." do
+      subject.some_arguments 'hello', 123, 'world'
+      subject.__data__[:some_arguments].should == ['hello', 123, 'world']
+    end
+
+    it "should set a value by a method with '='." do
+      subject.one_argument = 3
+      subject.__data__[:one_argument].should == [3]
+    end
+
+    it "should set values by a method with '='." do
+      subject.some_arguments = 'hello', 123, 'world'
+      subject.__data__[:some_arguments].should == ['hello', 123, 'world']
+    end
+
+    it "should delete values with a string key." do
+      subject.delete_key = 'abc', 'def'
+      lambda do
+        subject.__delete__('delete_key')
+      end.should change { subject.delete_key }.from(['abc', 'def']).to(nil)
+    end
+
+    it "should delete values with a symbol key." do
+      subject.delete_key = 'abc', 'def'
+      lambda do
+        subject.__delete__(:delete_key)
+      end.should change { subject.delete_key }.from(['abc', 'def']).to(nil)
+    end
+
+    it "should set nil." do
+      subject.nil_key = nil
+      subject.__data__[:nil_key].should == [nil]
+    end
+  end
+
+  context "when setting arguments" do
+    it "should add an argument" do
+      subject.argument << 'hello'
+      subject.argument.should include('hello')
+    end
+
+    it "should set argumets" do
+      subject.argument = ['ABC', 'DEF']
+      subject.argument.should == ['ABC', 'DEF']
+    end
+
+    it "should unshift an argument" do
+      subject.argument = ['ABC', 'DEF']
+      subject.argument.unshift('first')
+      subject.argument.should == ['first', 'ABC', 'DEF']
+    end
+  end
+
+  context "when cloning" do
+    it "should have same values." do
+      subject.argument = ['hello']
+      subject.some_key1 = [1, 2]
+      subject.some_key2 = 'ABCDE'
+      container_clone = DRbQS::Setting::Source.clone_container(subject)
+      subject.argument.should == container_clone.argument
+    end
+
+    it "should have a different argument." do
+      subject.argument = ['hello']
+      container_clone = DRbQS::Setting::Source.clone_container(subject)
+      subject.argument << 'world'
+      subject.argument.should_not == container_clone.argument
+    end
+
+    it "should have different values." do
+      subject.key1 = 'abc'
+      container_clone = DRbQS::Setting::Source.clone_container(subject)
+      subject.key1 = 'def'
+      subject.key1.should_not == container_clone.key1
+    end
+  end
+end

data/spec/setting/execute_spec.rb

@@ -0,0 +1,51 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe DRbQS::Setting::Execute do
+  subject do
+    DRbQS::Setting::Execute.new
+  end
+
+  context "when there are invalid arguments" do
+    it "should raise error by a value of invalid argument size." do
+      subject.set_argument('first', 'second')
+      lambda do
+        subject.parse!
+      end.should raise_error
+    end
+
+    it "should raise error by a value of invalid key." do
+      subject.set(:invalid_key, 1, 2, 3)
+      subject.set_argument('first')
+      lambda do
+        subject.parse!
+      end.should raise_error
+    end
+  end
+
+  context "when parsing" do
+    it "should not raise error." do
+      subject.set(:port, 123)
+      subject.set(:server, 'server')
+      subject.set(:node, 'node')
+      subject.set(:no_server)
+      subject.set(:no_node)
+      subject.set_argument('one')
+      lambda do
+        subject.parse!
+      end.should_not raise_error
+    end
+
+    it "should not change string for shell." do
+      subject.set(:port, 123)
+      subject.set(:server, 'server')
+      subject.set(:node, 'node')
+      subject.set(:no_server)
+      subject.set(:no_node)
+      subject.set_argument('one')
+      subject.parse!
+      lambda do
+        subject.parse!
+      end.should_not change(subject, :string_for_shell)
+    end
+  end
+end

data/spec/setting/manage_spec.rb

@@ -0,0 +1,63 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe DRbQS::Setting::Manage do
+  subject do
+    DRbQS::Setting::Manage.new
+  end
+
+  context "when there are invalid arguments" do
+    it "should raise error by an invalid mode." do
+      subject.set_argument('first', 'second')
+      lambda do
+        subject.parse!
+      end.should raise_error
+    end
+
+    it "should raise error by a value of invalid key." do
+      subject.set(:invalid_key, 1, 2, 3)
+      subject.set_argument('first')
+      lambda do
+        subject.parse!
+      end.should raise_error
+    end
+
+    [['process', 'invalid_command'],
+     ['send', 'invalid_type', 'uri', 'string']].each do |args|
+      it "should raise error for #{args.inspect}." do
+        subject.set_argument(*args)
+        lambda do
+          subject.parse!
+        end.should raise_error
+      end
+    end
+  end
+
+  context "when parsing" do
+    [['signal', 'uri', 'server-exit'],
+     ['signal', 'uri', 'node-exit-after-task', '3'],
+     ['signal', 'uri', 'node-wake', '4'],
+     ['signal', 'uri', 'node-sleep', '5'],
+     ['status', 'uri'],
+     ['history', 'uri'],
+     ['process', 'list'],
+     ['process', 'clear'],
+     ['send', 'string', 'uri', 'string'],
+     ['send', 'file', 'uri', __FILE__],
+     ['initialize']].each do |args|
+      it "should not raise error for #{args.inspect}." do
+        subject.set_argument(*args)
+        lambda do
+          subject.parse!
+        end.should_not raise_error
+      end
+
+      it "should not change string for shell for #{args.inspect}." do
+        subject.set_argument(*args)
+        subject.parse!
+        lambda do
+          subject.parse!
+        end.should_not change(subject, :string_for_shell)
+      end
+    end
+  end
+end