pione 0.1.0

data/test/tuple-space/spec_tuple-space-provider.rb

@@ -0,0 +1,36 @@
+require_relative '../test-util'
+
+describe 'Pione::TupleSpace::TupleSpaceProvider' do
+  before do
+    DRb.start_service
+    Global.front = Front::ClientFront.new(self)
+  end
+
+  after do
+    DRb.stop_service
+  end
+
+  it 'should get provider' do
+    p1 = TupleSpaceProvider.instance
+    p2 = TupleSpaceProvider.instance
+    p3 = TupleSpaceProvider.instance
+    p1.uuid.should == p2.uuid
+    p3.uuid.should == p3.uuid
+  end
+
+  it 'should terminate' do
+    provider = TupleSpaceProvider.new
+    should.not.raise { provider.terminate }
+    provider.terminated?
+    provider = TupleSpaceProvider.new
+    should.not.raise { provider.terminate }
+    provider.terminated?
+  end
+
+  it "should add tuple space server" do
+    tuple_space_server = TupleSpaceServer.new
+    provider = TupleSpaceProvider.new
+    provider.add_tuple_space_server(tuple_space_server)
+    provider.tuple_space_servers.map{|ts| ts.uuid}.first.should == tuple_space_server.uuid
+  end
+end

data/test/tuple-space/spec_tuple-space-receiver.rb

@@ -0,0 +1,32 @@
+require_relative '../test-util'
+
+describe "Pione::TupleSpace::TupleSpaceReceiver" do
+  after do
+    DRb.start_service
+  end
+
+  after do
+    DRb.stop_service
+  end
+
+  it "should get the receiver" do
+    should.not.raise do
+      receiver = TupleSpaceReceiver.new
+      receiver.terminate
+      receiver.should.terminated
+    end
+  end
+
+  it "should receive tuple space servers" do
+    tuple_space_server = TupleSpaceServer.new
+    provider = TupleSpaceProvider.new
+    receiver = TupleSpaceReceiver.new
+    provider.add_tuple_space_server(tuple_space_server)
+    sleep 1 # wait to received the packet...
+    servers = receiver.tuple_space_servers
+    servers.size.should == 1
+    servers.first.uuid.should == tuple_space_server.uuid
+    provider.terminate
+    receiver.terminate
+  end
+end

data/test/tuple-space/spec_tuple-space-server.rb

@@ -0,0 +1,49 @@
+require_relative '../test-util'
+
+describe "Pione::TupleSpace::TupleSpaceServer" do
+  before do
+    DRb.start_service
+    @server = TupleSpaceServer.new
+    sleep 1
+  end
+
+  after do
+    DRb.stop_service
+  end
+
+  it "should create be alive" do
+    @server.should.alive
+  end
+
+  it "should count tuple size" do
+    Tuple.define_format [:test, :sym]
+    @server.write([:test, :a])
+    @server.count_tuple([:test, nil]).should == 1
+    @server.write([:test, :b])
+    @server.count_tuple([:test, nil]).should == 2
+    Tuple.delete_format(:test)
+  end
+
+  it "should count workers" do
+    @server.current_task_worker_size.should == 0
+    t1 = Tuple[:agent].new(agent_type: :task_worker, uuid: Util.generate_uuid)
+    @server.write(t1)
+    @server.current_task_worker_size.should == 1
+    t2 = Tuple[:agent].new(agent_type: :task_worker, uuid: Util.generate_uuid)
+    @server.write(t2)
+    @server.current_task_worker_size.should == 2
+    t3 = Tuple[:agent].new(agent_type: :task_worker, uuid: Util.generate_uuid)
+    @server.write(t3)
+    @server.current_task_worker_size.should == 3
+    @server.take(t1)
+    @server.current_task_worker_size.should == 2
+    @server.take(t2)
+    @server.current_task_worker_size.should == 1
+    @server.take(t3)
+    @server.current_task_worker_size.should == 0
+  end
+
+  it "should know worker resource" do
+    @server.task_worker_resource.should == 1
+  end
+end

data/test/tuple/spec_basic-tuple.rb

@@ -0,0 +1,87 @@
+require_relative '../test-util'
+
+class TestTuple < Pione::Tuple::BasicTuple
+  define_format([:test,
+      [:attr1, Integer],
+      [:attr2, Symbol],
+      [:attr3, Object]])
+end
+
+describe "Pione::Tuple::BasicTuple" do
+  it "should get a tuple class" do
+    Tuple[:test].should.not.be.nil
+    Tuple[:test].should == TestTuple
+    Tuple[:test].ancestors.should.include Pione::Tuple::BasicTuple
+    Tuple[:test].format.should == [:test,
+      [:attr1, Integer],
+      [:attr2, Symbol],
+      [:attr3, Object]
+    ]
+  end
+
+  it "should get a tuple object" do
+    t1 = Tuple[:test].new(attr1: 1, attr2: :a, attr3: 3)
+    t2 = Tuple[:test].new(attr1: 3, attr2: :a, attr3: 1)
+    t3 = Tuple[:test].new(attr2: :a, attr3: 3, attr1: 1)
+    t4 = Tuple[:test].new(1, :a, 3)
+    t5 = Tuple[:test].new(3, :a, 1)
+    [t1, t2, t3, t4, t5].each {|t| t.should.be.kind_of? Tuple[:test]}
+    t1.should.not == t2
+    t1.should == t3
+    t1.should == t4
+    t1.should.not == t5
+    t2.should.not == t3
+    t2.should.not == t4
+    t2.should == t5
+    t3.should == t4
+    t3.should.not == t5
+    t4.should.not == t5
+  end
+
+  it "should get values of tuple objects" do
+    t1 = Tuple[:test].new(attr1: 1, attr2: :a, attr3: true)
+    t1.attr1.should == 1
+    t1.attr2.should == :a
+    t1.attr3.should == true
+    t2 = Tuple[:test].new(attr1: 2, attr3: false)
+    t2.attr1.should == 2
+    t2.attr2.should.be.nil
+    t2.attr3.should == false
+  end
+
+  it "should set new data in each field" do
+    t = Tuple[:test].new(attr1: 1, attr2: :a, attr3: true)
+    t.attr1 = 2
+    t.attr2 = :b
+    t.attr3 = false
+    t.attr1.should.equal 2
+    t.attr2.should.equal :b
+    t.attr3.should.equal false
+  end
+
+  it "should get any form" do
+    any = Tuple[:test].any
+    any.should == Tuple[:test].new
+    any.attr1.should.be.nil
+    any.attr2.should.be.nil
+    any.attr3.should.be.nil
+  end
+
+  it "should get tuple object from array" do
+    t = Tuple.from_array([:test, 1, :a, true])
+    t.should.be.kind_of TestTuple
+    t.attr1.should.== 1
+    t.attr2.should.== :a
+    t.attr3.should.== true
+    t.should.== Tuple[:test].new(attr1: 1, attr2: :a, attr3: true)
+  end
+
+  it "should raise exception when tuple is redefined by different format" do
+    # different format
+    should.raise(ScriptError) do
+      class TestTuple
+        define_format([:test, :attr3, :attr2, :attr1])
+      end
+    end
+  end
+end

data/test/tuple/spec_data-tuple.rb

@@ -0,0 +1,85 @@
+require_relative '../test-util'
+
+describe 'Pione::Tuple::TaskTuple' do
+  before do
+    @domain = "A"
+    @name = Model::DataExpr.new("a.txt")
+    @uri = "local:/home/keita/"
+    @time = Time.now
+    @data = Tuple::DataTuple.new(@domain, @name, @uri, @time)
+  end
+
+  after do
+    @data = nil
+  end
+
+  it 'should get identifier' do
+    @data.identifier.should == :data
+  end
+
+  it 'should get the domain' do
+    @data.domain.should == @domain
+  end
+
+  it 'should set the domain' do
+    domain = "B"
+    @data.domain = domain
+    @data.domain.should == domain
+  end
+
+  it 'should get name' do
+    @data.name.should == @name
+  end
+
+  it 'should set name' do
+    name = "b.txt"
+    @data.name = name
+    @data.name.should == name
+  end
+
+  it 'should get URI' do
+    @data.uri.should == @uri
+  end
+
+  it 'should set URI' do
+    uri = "local:./output"
+    @data.uri = uri
+    @data.uri.should == uri
+  end
+
+  it 'should get time' do
+    @data.time.should == @time
+  end
+
+  it 'should set time' do
+    time = Time.now
+    @data.time = time
+    @data.time.should == time
+  end
+
+  it 'should raise FormatError' do
+    should.raise(Tuple::FormatError) do
+      Tuple::DataTuple.new(true, @name, @uri, @time)
+    end
+
+    should.raise(Tuple::FormatError) do
+      Tuple::DataTuple.new(@domain, true, @uri, @time)
+    end
+
+    should.raise(Tuple::FormatError) do
+      Tuple::DataTuple.new(@domain, @name, true, @time)
+    end
+
+    should.raise(Tuple::FormatError) do
+      Tuple::DataTuple.new(@domain, @name, @uri, true)
+    end
+  end
+
+  it 'should get any tuple' do
+    any = Tuple::DataTuple.any
+    any.domain.should == nil
+    any.name.should == nil
+    any.uri.should == nil
+    any.time.should == nil
+  end
+end

data/test/tuple/spec_finished-tuple.rb

@@ -0,0 +1,61 @@
+require_relative '../test-util'
+
+describe 'Pione::Tuple::FinishedTuple' do
+  before do
+    @domain = "A"
+    @status = :success
+
+    name = Model::DataExpr.new("a.txt")
+    uri = "local:/home/keita/"
+    time = Time.now
+    data = Tuple::DataTuple.new(@domain, name, uri, time)
+
+    @outputs = [data]
+    @digest = "A"
+    @data = Tuple::FinishedTuple.new(@domain, @status, @outputs, @digest)
+  end
+
+  after do
+    @data = nil
+  end
+
+  it 'should get domain' do
+    @data.domain.should == @domain
+  end
+
+  it 'should set domain' do
+    domain = "B"
+    @data.domain = domain
+    @data.domain.should == domain
+  end
+
+  it 'should get status' do
+    @data.status.should == @status
+  end
+
+  it 'should set status' do
+    status = :failed
+    @data.status = status
+    @data.status.should == status
+  end
+
+  it 'should get outputs' do
+    @data.outputs.should == @outputs
+  end
+
+  it 'should set outputs' do
+    outputs = []
+    @data.outputs = outputs
+    @data.outputs.should == outputs
+  end
+
+  it 'should get digest' do
+    @data.digest.should == @digest
+  end
+
+  it 'should set digest' do
+    digest = "B"
+    @data.digest = digest
+    @data.digest.should == digest
+  end
+end

data/test/tuple/spec_task-tuple.rb

@@ -0,0 +1,127 @@
+require_relative '../test-util'
+
+describe 'Pione::Tuple::TaskTuple' do
+  before do
+    domain = "A"
+    data = Tuple[:data].new(domain, "a.txt", "local:/home/keita/", Time.now)
+
+    @rule_path = "main_Main"
+    @inputs = [data]
+    @params = Parameters.new(Model::Variable.new("X") => Model::PioneInteger.new(1))
+    @features = Feature.empty
+    @domain = domain
+    @call_stack = []
+
+    args = [@rule_path, @inputs, @params, @features, @domain, @call_stack]
+    @task = Tuple::TaskTuple.new(*args)
+  end
+
+  after do
+    @task = nil
+  end
+
+  it 'should get the class from identifier' do
+    Tuple[:task].should == Tuple::TaskTuple
+  end
+
+  it 'should get identifier' do
+    @task.identifier.should == :task
+  end
+
+  it 'should get rule path' do
+    @task.rule_path.should == @rule_path
+  end
+
+  it 'should set rule path' do
+    rule_path = "B"
+    @task.rule_path = rule_path
+    @task.rule_path.should == rule_path
+  end
+
+  it 'should get inputs' do
+    @task.inputs.should == @inputs
+  end
+
+  it 'should set inputs' do
+    @task.inputs = []
+    @task.inputs.should == []
+  end
+
+  it 'should get params' do
+    @task.params.should == @params
+  end
+
+  it 'should set params' do
+    params = Parameters.new(Model::Variable.new("Y") => Model::PioneInteger.new(2))
+    @task.params = params
+    @task.params.should == params
+  end
+
+  it 'should get features' do
+    @task.features.should == @features
+  end
+
+  it 'should set features' do
+    features = Model::Feature::RequestExpr.new("X")
+    @task.features = features
+    @task.features.should == features
+  end
+
+  it 'should get call-stack' do
+    @task.call_stack.should == @call_stack
+  end
+
+  it 'should set call-stack' do
+    call_stack = ["a", "b" , "c"]
+    @task.call_stack = call_stack
+    @task.call_stack.should == call_stack
+  end
+
+  it 'should get digest' do
+    @task.digest.should == "main_Main([a.txt],{X:1})"
+  end
+
+  it 'should raise format error' do
+    should.raise(Tuple::FormatError) do
+      args = [true, @inputs, @params, @features, @domain, @call_stack]
+      Tuple::TaskTuple.new(*args)
+    end
+
+    should.raise(Tuple::FormatError) do
+      args = [@rule_path, true, @params, @features, @domain, @call_stack]
+      Tuple::TaskTuple.new(*args)
+    end
+
+    should.raise(Tuple::FormatError) do
+      args = [@rule_path, @inputs, true, @features, @domain, @call_stack]
+      Tuple::TaskTuple.new(*args)
+    end
+
+    should.raise(Tuple::FormatError) do
+      args = [@rule_path, @inputs, @params, true, @domain, @call_stack]
+      Tuple::TaskTuple.new(*args)
+    end
+
+    should.raise(Tuple::FormatError) do
+      args = [@rule_path, @inputs, @params, @features, true, @call_stack]
+      Tuple::TaskTuple.new(*args)
+    end
+
+    should.raise(Tuple::FormatError) do
+      args = [@rule_path, @inputs, @params, @features, @domain, true]
+      Tuple::TaskTuple.new(*args)
+    end
+  end
+
+  it 'should get any tuple' do
+    any = Tuple::TaskTuple.any
+    any.identifier.should == :task
+    any.rule_path.should == nil
+    any.inputs.should == nil
+    any.params.should == nil
+    any.features.should == nil
+    any.domain.should == nil
+    any.domain.should == nil
+  end
+end
+