drbqs 0.0.5 → 0.0.6

data/spec/connection_spec.rb

@@ -0,0 +1,39 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require 'drbqs/connection.rb'
+
+describe DRbQS::ConnectionClient do
+  before(:all) do
+    @message = Rinda::TupleSpace.new
+    @connection = DRbQS::ConnectionClient.new(@message)
+    @node_id = 23
+    id_string = @connection.instance_variable_get(:@id_string)
+    @message.write([id_string, @node_id])
+  end
+
+  it "should get node ID." do
+    @connection.get_id.should == @node_id
+    @message.take([:connect, nil]).should be_true
+  end
+
+  it "should get no initialization method." do
+    @connection.get_initialization.should be_nil
+  end
+
+  it "should get initialization" do
+    ary = [:initialize, [1, 2], :size, []]
+    @message.write(ary)
+    @connection.get_initialization.should == ary[1..-1]
+  end
+
+  it "should respond :alive_p signal" do
+    @message.write([@node_id, :alive_p])
+    @connection.respond_alive_signal
+    @message.take([:alive, nil]).should be_true
+  end
+
+  it "should respond :exit signal" do
+    @message.write([@node_id, :exit])
+    @connection.respond_alive_signal.should == :exit
+  end
+end

data/spec/manage_spec.rb

@@ -0,0 +1,50 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require_relative 'test/test1.rb'
+
+describe DRbQS do
+  before(:all) do
+    @tasks = []
+    5.times do |i|
+      @tasks << DRbQS::Task.new(Test1.new, :echo, [i])
+    end
+    @process_id = fork do
+      server = DRbQS::Server.new(:port => 13501)
+
+      @tasks.each do |task|
+        server.queue.add(task)
+      end
+
+      server.set_finish_hook do |serv|
+        serv.exit
+      end
+
+      server.set_signal_trap
+      server.start
+      server.wait
+    end
+    sleep(1)
+
+    @uri = 'druby://:13501'
+
+    @manage = DRbQS::Manage.new(@uri)
+  end
+
+  it "should send exit signal" do
+    lambda do
+      @manage.send_exit_signal
+    end.should_not raise_error
+    lambda do
+      i = 0
+      while !Process.waitpid(@process_id, Process::WNOHANG)
+        i += 1
+        if i > 10
+          Process.kill(:KILL, @process_id)
+          raise "Server process does not finish."
+        end
+        sleep(1)
+      end
+    end.should_not raise_error
+  end
+
+end

data/spec/message_spec.rb

@@ -0,0 +1,71 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require 'drbqs/message.rb'
+
+describe DRbQS::MessageServer do
+  before(:all) do
+    @message = Rinda::TupleSpace.new
+    @message_server = DRbQS::MessageServer.new(@message)
+    @node_id_list = []
+  end
+
+  it "should return nil" do
+    @message_server.get_message.should be_nil
+  end
+
+  it "should have no node" do
+    @message_server.node_not_exist?.should be_true
+  end
+
+  it "should set initialization task" do
+    lambda do
+      @message.take([:initialize, nil, Symbol, nil], 0)
+    end.should raise_error Rinda::RequestExpiredError
+    @message_server.set_initialization(DRbQS::Task.new(Object.new, :object_id))
+    @message.take([:initialize, nil, Symbol, nil], 0).should be_true
+  end
+
+  it "should get :connect message" do
+    5.times do |i|
+      id_str = "connect_test_#{i}"
+      @message.write([:connect, id_str])
+      @message_server.get_message
+      (ary = @message.take([id_str, Fixnum])).should be_true
+      @node_id_list << ary[1]
+      @message_server.node_not_exist?.should be_false
+    end
+  end
+
+  it "should get :alive message" do
+    node_id = 73
+    @message.write([:alive, node_id])
+    node_list = @message_server.instance_variable_get(:@node_list)
+    node_list.should_receive(:set_alive).with(node_id)
+    @message_server.get_message
+  end
+
+  it "should get :exit_server message" do
+    @message.write([:exit_server, 'message_test'])
+    @message_server.get_message.should == :exit_server
+  end
+
+  it "should send exit message" do
+    @message_server.send_exit
+    @node_id_list.each do |id|
+      @message.take([id, :exit]).should be_true
+    end
+  end
+
+  it "should delete a node" do
+    @message_server.check_connection.should == []
+    @node_id_list.each do |id|
+      @message.take([id, :alive_p]).should be_true
+    end
+    @message_server.check_connection.should == @node_id_list
+    @node_id_list.each do |id|
+      lambda do
+        @message.take([id, :alive_p], 0)
+      end.should raise_error Rinda::RequestExpiredError
+    end
+  end
+end

data/spec/node_list_spec.rb

@@ -0,0 +1,54 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require 'drbqs/node_list'
+
+describe DRbQS::NodeList do
+  before(:all) do
+    @node_list = DRbQS::NodeList.new
+    @id_strings = 10.times.map { |i| sprintf("%05d", i) }
+    @id_list = []
+  end
+
+  it "should be empty." do
+    @node_list.empty?.should be_true
+  end
+
+  it "should get ids that are not duplicated." do
+    @id_strings.each do |str|
+      @id_list << @node_list.get_new_id(str)
+    end
+    @id_list.uniq!
+    @id_list.all? { |i| Integer === i }.should be_true
+    @id_list.size.should == @id_strings.size
+    @node_list.each do |id_num, id_str|
+      @id_strings.include?(id_str).should be_true
+      @id_list.include?(id_num).should be_true
+    end
+  end
+
+  it "should delete all ids" do
+    @node_list.empty?.should_not be_true
+    @node_list.set_check_connection
+    ids = @node_list.delete_not_alive
+    ids.sort.should == @id_list.sort
+    @node_list.empty?.should be_true
+    @id_list.clear
+  end
+
+  it "should set alive flag" do
+    alive_id_num = [3, 4, 5]
+    @id_strings.each do |str|
+      @id_list << @node_list.get_new_id(str)
+    end
+    @node_list.set_check_connection
+    alive_id_num.each do |i|
+      @node_list.set_alive(@id_list[i])
+    end
+    @node_list.delete_not_alive
+    alive_ids = alive_id_num.map { |i| @id_list[i] }
+    @node_list.each do |id_num, id_str|
+      alive_ids.include?(id_num).should be_true
+    end
+  end
+
+end

data/spec/queue_spec.rb

@@ -0,0 +1,59 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require 'drbqs/queue'
+
+describe DRbQS::QueueServer do
+  before(:all) do
+    @ts = {
+      :queue => Rinda::TupleSpace.new,
+      :result => Rinda::TupleSpace.new
+    }
+    @queue_server = DRbQS::QueueServer.new(@ts[:queue], @ts[:result])
+    @task = { :obj => DRbQS::Task.new([1, 2, 3], :size, []), :id => nil }
+    @node_id = 100
+  end
+
+  it "should be empty" do
+    @queue_server.calculating_task_number.should == 0
+    @queue_server.empty?.should be_true
+    @queue_server.finished?.should be_true
+  end
+
+  it "should add a task" do
+    @task[:id] = @queue_server.add(@task[:obj])
+    @queue_server.calculating_task_number.should == 0
+    @queue_server.empty?.should be_false
+    @queue_server.finished?.should be_false
+    @ts[:queue].take([nil, nil, nil, nil]).should be_true
+  end
+
+  it "should get accept signal" do
+    @ts[:result].write([:accept, @task[:id], @node_id])
+    @queue_server.get_accept_signal.should == 1
+    @queue_server.calculating_task_number.should == 1
+    @queue_server.empty?.should be_true
+    @queue_server.finished?.should be_false
+    @ts[:result].read_all([nil, nil, nil]).size.should == 0
+  end
+
+  it "should get result" do
+    @ts[:result].write([:result, @task[:id], :result_object])
+    @queue_server.get_result
+    @queue_server.calculating_task_number.should == 0
+    @queue_server.empty?.should be_true
+    @queue_server.finished?.should be_true
+  end
+
+  it "should delete node" do
+    @task[:id] = @queue_server.add(@task[:obj])
+    @ts[:queue].take([nil, nil, nil, nil]).should be_true
+    @ts[:result].write([:accept, @task[:id], 100])
+    @queue_server.get_accept_signal.should == 1
+    @queue_server.requeue_for_deleted_node_id([@node_id])
+    @queue_server.calculating_task_number.should == 0
+    @queue_server.empty?.should be_false
+    @queue_server.finished?.should be_false
+    (ary = @ts[:queue].take([nil, nil, nil, nil])).should be_true
+    ary[0].should == @task[:id]
+  end
+end

data/spec/server_define_spec.rb

@@ -0,0 +1,34 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe DRbQS::ServerDefinition do
+  context "when we call class methods" do
+    before(:all) do
+      @server_definition = DRbQS.class_variable_get(:@@server_def)
+    end
+
+    it "should define server" do
+      @server_definition.should_receive(:define_server)
+      DRbQS.define_server do |server, argv, opts|
+        server.set_finish_hook do |serv|
+          serv.exit
+        end
+      end
+    end
+
+    it "should set parser of options" do
+      @server_definition.should_receive(:option_parser)
+      DRbQS.option_parser do |opt, hash|
+        opt.on('--test') do |v|
+          hash[:test] = true
+        end
+      end
+    end
+
+    it "should parse options" do
+      @server_definition.should_receive(:parse_option)
+      DRbQS.parse_option(['--test'])
+    end
+
+  end
+
+end

data/spec/task_client_spec.rb

@@ -0,0 +1,53 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require 'drbqs/task_client'
+
+describe DRbQS::TaskClient do
+  before(:all) do
+    @node_id = 4
+    @ts_queue = Rinda::TupleSpace.new
+    @ts_result = Rinda::TupleSpace.new
+    @task_client = DRbQS::TaskClient.new(@node_id, @ts_queue, @ts_result)
+    @task_example = [[1, 2, 3], :size, []]
+    @task_id = 10
+  end
+
+  it "should be empty" do
+    @task_client.task_empty?.should be_true
+    @task_client.result_empty?.should be_true
+    @task_client.calculating_task.should be_nil
+  end
+
+  it "should queue a task" do
+    @ts_queue.write([@task_id] + @task_example)
+    @task_client.add_new_task
+    @task_client.task_empty?.should_not be_true
+    @task_client.result_empty?.should be_true
+    @task_client.calculating_task.should == @task_id
+    @ts_queue.read_all([nil, nil, nil, nil]).size.should == 0
+    @ts_result.take([:accept, nil, nil], 0).should be_true
+  end
+
+  it "should dequeue a task" do
+    ary = @task_client.dequeue_task
+    ary.should == @task_example
+    @task_client.task_empty?.should be_true
+    @task_client.result_empty?.should be_true
+    @task_client.calculating_task.should == @task_id
+  end
+
+  it "should queue result" do
+    @task_client.queue_result(@task_example[0].__send__(@task_example[1], *@task_example[2]))
+    @task_client.task_empty?.should be_true
+    @task_client.result_empty?.should_not be_true
+    @task_client.calculating_task.should == @task_id
+  end
+
+  it "should send result" do
+    @task_client.send_result
+    @task_client.task_empty?.should be_true
+    @task_client.result_empty?.should be_true
+    @task_client.calculating_task.should be_nil
+    @ts_result.take([:result, nil, nil], 0).should be_true
+  end
+end

data/spec/task_generator_spec.rb

@@ -0,0 +1,32 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require 'drbqs/task_generator'
+
+describe DRbQS::TaskGenerator do
+  def check_task_ary(tasks, num)
+    tasks.should have(num).items
+    tasks.all? { |t| DRbQS::Task === t }.should be_true
+  end
+
+  subject { DRbQS::TaskGenerator.new(:abc => 'ABC', :def => 123, :data => [1, 2, 3]) }
+
+  it "should initialize instance varibles" do
+    subject.instance_variable_get('@abc').should == 'ABC'
+    subject.instance_variable_get('@def').should == 123
+  end
+
+  it "should create new tasks" do
+    subject.set(2) do
+      @data.each do |i|
+        create_add_task(i, :to_s)
+      end
+    end
+    check_task_ary(subject.new_tasks, 2)
+    check_task_ary(subject.new_tasks, 1)
+    subject.new_tasks.should be_nil
+  end
+
+  it "should debug generator" do
+    subject.debug_all_tasks.should be_true
+  end
+end

data/spec/task_spec.rb

@@ -0,0 +1,21 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe DRbQS::Task do
+  it "should not set hook" do
+    task = DRbQS::Task.new([1, 2, 3], :size)
+    task.hook.should be_nil
+  end
+
+  it "should set hook" do
+    task = DRbQS::Task.new([1, 2, 3], :size) do |server, ret|
+      p ret
+    end
+    task.hook.should be_an_instance_of Proc
+  end
+
+  it "should have same targets" do
+    task1 = DRbQS::Task.new([1, 2, 3], :concat, [3, 4, 5])
+    task2 = DRbQS::Task.new([1, 2, 3], :concat, [3, 4, 5])
+    task1.same_target?(task2).should be_true
+  end
+end

data/spec/test/test1.rb

@@ -0,0 +1,13 @@
+class Test1
+  @@execute_echo_number = 0
+
+  def echo(*args)
+    puts "execute Test1#echo(*#{args.inspect.strip})"
+    @@execute_echo_number += 1
+    args
+  end
+
+  def self.get_execute_echo_number
+    @@execute_echo_number
+  end
+end

data/spec/test1_spec.rb

@@ -0,0 +1,80 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require_relative '../lib/drbqs.rb'
+require_relative 'test/test1.rb'
+
+describe DRbQS do
+  before(:all) do
+    @tasks = []
+    5.times do |i|
+      @tasks << DRbQS::Task.new(Test1.new, :echo, [i])
+    end
+    @process_id = fork do
+      server = DRbQS::Server.new(:port => 13501)
+
+      @tasks.each do |task|
+        server.queue.add(task)
+      end
+
+      server.set_finish_hook do |serv|
+        serv.exit
+      end
+
+      server.set_signal_trap
+      server.start
+      server.wait
+    end
+    sleep(1)
+
+    @uri = 'druby://:13501'
+    @client = DRbQS::Client.new(@uri, :log_file => $stdout, :continue => true)
+  end
+
+  it "should have nil instance variables" do
+    @client.instance_variable_get(:@task_client).should be_nil
+    @client.instance_variable_get(:@connection).should be_nil
+    @client.connect
+  end
+
+  it "should initialize @task_client" do
+    task_client = @client.instance_variable_get(:@task_client)
+    task_client.should be_an_instance_of DRbQS::TaskClient
+    task_client.node_id.should be_an_instance_of Fixnum
+    task_client.task_empty?.should be_true
+    task_client.result_empty?.should be_true
+  end
+
+  it "should initialize @connection" do
+    connection = @client.instance_eval { @connection }
+    connection.should be_an_instance_of DRbQS::ConnectionClient
+    connection.instance_variable_get(:@id_number).should be_an_instance_of Fixnum
+    connection.instance_variable_get(:@id_string).should be_an_instance_of String
+  end
+
+  it "should calculate" do
+    task_client = @client.instance_eval { @task_client }
+    # *** Too late ***
+    # task_client.should_receive(:add_new_task).at_least(:once)
+    # task_client.should_receive(:transit).exactly(5).times
+    # task_client.should_receive(:send_result).exactly(5).times
+    lambda do
+      @client.calculate
+    end.should_not raise_error
+    Test1.get_execute_echo_number.should == @tasks.size
+  end
+
+  after(:all) do
+    lambda do
+      i = 0
+      while !Process.waitpid(@process_id, Process::WNOHANG)
+        i += 1
+        if i > 10
+          Process.kill(:KILL, @process_id)
+          raise "Server process does not finish."
+        end
+        sleep(1)
+      end
+    end.should_not raise_error
+  end
+
+end

data/spec/test2_spec.rb

@@ -0,0 +1,75 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+require_relative '../lib/drbqs.rb'
+require_relative 'test/test1.rb'
+
+describe DRbQS do
+  before(:all) do
+    @tasks = []
+    @task_generator = DRbQS::TaskGenerator.new(:iterate => 3)
+    @task_generator.set do
+      @iterate.times do |i|
+        create_add_task(Test1.new, :echo, [i])
+      end
+    end
+    @process_id = fork do
+      server = DRbQS::Server.new(:port => 13501)
+
+      server.set_task_generator(@task_generator)
+
+      server.set_finish_hook do |serv|
+        serv.exit
+      end
+
+      server.set_signal_trap
+      server.start
+      server.wait
+    end
+    sleep(1)
+
+    @uri = 'druby://:13501'
+    @client = DRbQS::Client.new(@uri, :log_file => $stdout, :continue => true)
+  end
+
+  it "should have nil instance variables" do
+    @client.instance_variable_get(:@task_client).should be_nil
+    @client.instance_variable_get(:@connection).should be_nil
+    @client.connect
+  end
+
+  it "should initialize @task_client" do
+    task_client = @client.instance_variable_get(:@task_client)
+    task_client.should be_an_instance_of DRbQS::TaskClient
+    task_client.node_id.should be_an_instance_of Fixnum
+    task_client.task_empty?.should be_true
+    task_client.result_empty?.should be_true
+  end
+
+  it "should initialize @connection" do
+    connection = @client.instance_eval { @connection }
+    connection.should be_an_instance_of DRbQS::ConnectionClient
+    connection.instance_variable_get(:@id_number).should be_an_instance_of Fixnum
+    connection.instance_variable_get(:@id_string).should be_an_instance_of String
+  end
+
+  it "should calculate" do
+    lambda do
+      @client.calculate
+    end.should_not raise_error
+  end
+
+  after(:all) do
+    lambda do
+      i = 0
+      while !Process.waitpid(@process_id, Process::WNOHANG)
+        i += 1
+        if i > 10
+          Process.kill(:KILL, @process_id)
+          raise "Server process does not finish."
+        end
+        sleep(1)
+      end
+    end.should_not raise_error
+  end
+
+end