unageanu-jiji 0.1.0

data/test/test_Process.rb

@@ -0,0 +1,186 @@
+#!/usr/bin/ruby
+
+$: << "../lib"
+
+
+require "runit/testcase"
+require "runit/cui/testrunner"
+require "jiji/process"
+require 'jiji/agent/agent'
+require 'jiji/agent/agent_manager'
+require 'jiji/agent/agent_registry'
+require 'jiji/agent/permitter'
+require 'jiji/agent/util'
+require "logger"
+require "testutils"
+
+# プロセスのテスト
+class ProcessTest <  RUNIT::TestCase
+
+  include Test::Constants
+
+  def setup
+    @dir = File.dirname(__FILE__) + "/ProcessTest.tmp"
+    FileUtils.mkdir_p @dir
+  end
+
+  def teardown
+    FileUtils.rm_rf @dir
+  end
+
+  def test_basic
+
+    # 新規に作成
+    agent_mng = JIJI::AgentManager.new( "aaa", RegistryMock.new, Logger.new(STDOUT))
+    agent_mng.operator = Struct.new(:trade_enable).new(true)
+    agent_mng.conf = CONF
+    agent_mng.conf.set( [:agent,:safe_level], 0)
+
+    registory_mock = Object.new
+    class << registory_mock
+      def output( agent_name, dir )
+        output = JIJI::Output.new( agent_name, "#{dir}/out" )
+        return output
+      end
+    end
+    agent_mng.registory = registory_mock
+
+    p1 = JIJI::Process.new( "1", @dir, agent_mng, {} )
+    assert_equals p1.id, "1"
+    assert_equals p1.props, {"agents"=>[]}
+    assert_equals p1["x"], nil
+    assert_equals p1.agent_manager, agent_mng
+
+    p1["x"] = "xxx"
+    assert_equals p1["x"], "xxx"
+
+    # トレードのon/offを設定
+    p1["trade_enable"] = false
+    assert_equals p1["trade_enable"], false
+    assert_equals p1.agent_manager.operator.trade_enable, false
+    p1["trade_enable"] = true
+    assert_equals p1["trade_enable"], true
+    assert_equals p1.agent_manager.operator.trade_enable, true
+    p1["trade_enable"] = false
+    assert_equals p1["trade_enable"], false
+    assert_equals p1.agent_manager.operator.trade_enable, false
+
+    # エージェントの設定を変更
+    agents = [
+      {"id"=>"aaa",
+       "name"=>"テストエージェント1",
+       "class"=>"testclass@foo.rb",
+       "class_name" => "testclass",
+       "file_name" => "foo.rb",
+       "description" => "説明",
+       "property_def" => {"id"=>"x", "default"=>85 },
+       "properties" => {"x"=>30, "y"=>40 }},
+      {"id"=>"bbb",
+       "name"=>"テストエージェント2",
+       "class"=>"testclass2@foo.rb",
+       "class_name" => "testclass",
+       "file_name" => "foo.rb",
+       "description" => "説明2",
+       "property_def" => {"id"=>"x", "default"=>85 },
+       "properties" => {"x"=>30, "y"=>41 }}
+    ]
+
+    # 追加
+    p1["agents"] = agents
+    assert_equals p1["agents"], agents
+    assert_equals agent_mng.get("aaa").agent.cl, "testclass@foo.rb"
+    assert_equals agent_mng.get("aaa").agent.properties, {"x"=>30, "y"=>40 }
+    assert_equals agent_mng.get("bbb").agent.cl, "testclass2@foo.rb"
+    assert_equals agent_mng.get("bbb").agent.properties, {"x"=>30, "y"=>41 }
+
+    # 変更
+    agents = [
+      {"id"=>"aaa",
+       "name"=>"テストエージェント1",
+       "class"=>"testclass@foo.rb",
+       "class_name" => "testclass",
+       "file_name" => "foo.rb",
+       "description" => "説明",
+       "property_def" => {"id"=>"x", "default"=>85 },
+       "properties" => {"x"=>50, "y"=>40 }},
+      {"id"=>"bbb",
+       "name"=>"テストエージェント4",
+       "class"=>"testclass2@foo.rb",
+       "class_name" => "testclass",
+       "file_name" => "foo.rb",
+       "description" => "説明2aaaaaa",
+       "property_def" => {"id"=>"x", "default"=>85 },
+       "properties" => {"x"=>60, "y"=>41 }}
+    ]
+    p1["agents"] = agents
+    assert_equals p1["agents"], agents
+    assert_equals agent_mng.get("aaa").agent.cl, "testclass@foo.rb"
+    assert_equals agent_mng.get("aaa").agent.properties, {"x"=>50, "y"=>40 }
+    assert_equals agent_mng.get("bbb").agent.cl, "testclass2@foo.rb"
+    assert_equals agent_mng.get("bbb").agent.properties, {"x"=>60, "y"=>41 }
+
+    # 削除
+    agents = [
+      {"id"=>"aaa",
+       "name"=>"テストエージェント1",
+       "class"=>"testclass@foo.rb",
+       "class_name" => "testclass",
+       "file_name" => "foo.rb",
+       "description" => "説明",
+       "property_def" => {"id"=>"x", "default"=>85 },
+       "properties" => {"x"=>50, "y"=>41 }}
+    ]
+    p1["agents"] = agents
+    assert_equals p1["agents"], agents
+    assert_equals agent_mng.get("aaa").agent.cl, "testclass@foo.rb"
+    assert_equals agent_mng.get("aaa").agent.properties, {"x"=>50, "y"=>41 }
+    assert_equals agent_mng.get("bbb"), nil
+
+
+    # 再読み込み / ファイルに保存された設定値が読み込まれる。
+    p1 = JIJI::Process.new( "1",  @dir, agent_mng )
+    assert_equals p1.id, "1"
+    assert_equals p1.agent_manager, agent_mng
+    assert_equals p1["x"], "xxx"
+    assert_equals p1["agents"], agents
+    assert_equals agent_mng.get("aaa").agent.cl, "testclass@foo.rb"
+    assert_equals agent_mng.get("aaa").agent.properties, {"x"=>50, "y"=>41 }
+    assert_equals agent_mng.get("bbb"), nil
+    assert_equals p1["trade_enable"], false
+    assert_equals p1.agent_manager.operator.trade_enable, false
+
+    # 別のプロセスを作成 / 設定値は別途保持される。
+    p2 = JIJI::Process.new( "2", @dir, agent_mng, {} )
+    assert_equals p2.id, "2"
+    assert_equals p2.agent_manager, agent_mng
+    assert_equals p2["x"], nil
+    assert_equals p2["agents"], []
+
+  end
+
+  class Agent
+    include JIJI::Agent
+
+    def initialize( cl, properties )
+      @cl = cl
+      @properties = properties
+    end
+
+    # 設定されたプロパティを取得する
+    def properties
+      @properties
+    end
+    # プロパティを設定する
+    def properties=( properties )
+      @properties = properties
+    end
+    attr :cl, true
+  end
+
+  class RegistryMock
+    def create( cl, property )
+      Agent.new( cl, property )
+    end
+  end
+
+end

data/test/test_ProcessManager.rb

@@ -0,0 +1,312 @@
+#!/usr/bin/ruby
+
+$: << "../lib"
+
+
+clazz = class << Thread; self; end
+clazz.__send__(:alias_method, :start_org, :start )
+def Thread.start(*args, &b)
+  Thread.start_org( caller, *args) {|stack, *arg|
+      Thread.current[:stack] = stack
+      yield( *arg )
+  }
+end
+def Thread.fork(*args, &b)
+  Thread.start_org( caller, *args) {|stack, *arg|
+      Thread.current[:stack] = stack
+      yield( *arg )
+  }
+end
+def Thread.new(*args, &b)
+  Thread.start_org( caller, *args) {|stack, *arg|
+      Thread.current[:stack] = stack
+      yield( *arg )
+  }
+end
+
+require "runit/testcase"
+require "runit/cui/testrunner"
+require 'jiji/registry'
+require "logger"
+require "testutils"
+
+# プロセスマネージャのテスト
+class ProcessManagerTest <  RUNIT::TestCase
+
+  include Test::Constants
+
+  def setup
+    @dir = File.dirname(__FILE__) + "/ProcessManagerTest"
+    @registry = JIJI::Registry.new(@dir , nil)
+    @mng = @registry[:process_manager]
+  end
+
+  def teardown
+    begin
+      @mng.stop
+      begin
+        @registry.permitter.close
+      ensure
+        @registry.client.close
+      end
+#      Thread.list.each {|t|
+#        puts "---#{t}"
+#        puts t[:stack]
+#      }
+    ensure
+      FileUtils.rm_rf "#{@dir}/logs"
+      FileUtils.rm_rf "#{@dir}/process_logs"
+    end
+  end
+
+  def test_basic
+    @mng.start
+    agents = [{
+      "name" => "aaa",
+      "class" => "MovingAverageAgent@moving_average.rb",
+      "id"=> "44c0d256-8994-4240-a6c6-8d9546aef8c4",
+      "properties" =>  {
+          "period" => 10,
+          "short" => 25,
+          "long" => 75
+       }
+    }]
+
+    # プロセスを追加
+    pid = @mng.create_back_test( "test", "memo", Time.gm( 2008, 8, 21 ).to_i, Time.gm( 2008, 8, 22 ).to_i, agents )["id"]
+    sleep 0.1
+
+    # プロセスを取得
+    process = @mng.get( pid )
+    assert_equals process.state, :RUNNING
+    assert_not_nil process.progress
+    assert_equals process.id, pid
+    assert_equals process["name"], "test"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+    # 実行完了を待つ
+    sleep 1 while   @mng.get( pid ).state == :RUNNING
+
+    assert_equals process.state, :FINISHED
+    assert_not_nil process.progress
+    assert_equals process.id, pid
+    assert_equals process["name"], "test"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+    #別のプロセスを追加
+    pid2 = @mng.create_back_test( "test2", "memo", Time.gm( 2008, 8, 21 ).to_i, Time.gm( 2008, 8, 22 ).to_i, agents )["id"]
+    sleep 1
+
+    process = @mng.get( pid )
+    assert_equals process.state, :FINISHED
+    assert_not_nil process.progress
+    assert_equals process.id, pid
+    assert_equals process["name"], "test"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+    process = @mng.get( pid2 )
+    assert_equals process.state, :RUNNING
+    assert_not_nil process.progress
+    assert_equals process.id, pid2
+    assert_equals process["name"], "test2"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+    
+    # 停止
+    @mng.stop
+    
+    process = @mng.get( pid )
+    assert_equals process.state, :FINISHED
+    assert_not_nil process.progress
+    assert_equals process.id, pid
+    assert_equals process["name"], "test"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+    process = @mng.get( pid2 )
+    assert_equals process.state, :CANCELED
+    assert_not_nil process.progress
+    assert_equals process.id, pid2
+    assert_equals process["name"], "test2"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+
+    # マネージャを再作成 / プロセスがローカルのファイルより復元される
+    recreate_registry
+    @mng = @registry[:process_manager]
+
+    process = @mng.get( pid )
+    assert_equals process.state, :FINISHED
+    assert_not_nil process.progress
+    assert_equals process.id, pid
+    assert_equals process["name"], "test"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+    process = @mng.get( pid2 )
+    assert_equals process.state, :CANCELED
+    assert_not_nil process.progress
+    assert_equals process.id, pid2
+    assert_equals process["name"], "test2"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+
+    # プロセスを削除
+    @mng.delete_back_test( pid )
+    assert_process_not_found(pid)
+
+    process = @mng.get( pid2 )
+    assert_equals process.state, :CANCELED
+    assert_not_nil process.progress
+    assert_equals process.id, pid2
+    assert_equals process["name"], "test2"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+    @mng.delete_back_test( pid2 )
+    assert_process_not_found(pid2)
+
+    # マネージャを再作成 / 削除されたデータは消えている筈
+    recreate_registry
+    @mng = @registry[:process_manager]
+
+    assert_process_not_found(pid)
+    assert_process_not_found(pid2)
+
+    # 実行中に削除
+    pid3 = @mng.create_back_test( "test3", "memo", Time.gm( 2008, 8, 21 ).to_i, Time.gm( 2008, 8, 22 ).to_i, agents )["id"]
+    sleep 0.1
+
+    process = @mng.get( pid3 )
+    assert_equals process.state, :RUNNING
+    assert_not_nil process.progress
+    assert_equals process.id, pid3
+    assert_equals process["name"], "test3"
+    assert_equals process["memo"], "memo"
+    assert_equals process["start_date"], Time.gm( 2008, 8, 21 ).to_i
+    assert_equals process["end_date"], Time.gm( 2008, 8, 22 ).to_i
+    assert_not_nil process["create_date"]
+
+    @mng.delete_back_test( pid3 )
+    assert_process_not_found(pid3)
+
+    # マネージャを再作成 / 削除されたデータは消えている筈
+    recreate_registry
+    @mng = @registry[:process_manager]
+    assert_process_not_found(pid3)
+
+  end
+#
+#  #
+#  #複数プロセスを同時に起動するテスト。
+#  #
+#  def test_multi_process
+#    @mng.start
+#    agents = [{
+#      "name" => "aaa",
+#      "class" => "MovingAverageAgent@moving_average.rb",
+#      "id"=> "44c0d256-8994-4240-a6c6-8d9546aef8c4",
+#      "properties" =>  {
+#          "period" => 10,
+#          "short" => 25,
+#          "long" => 75
+#       }
+#    }]
+#
+#    # プロセスを追加
+#    pid = @mng.create_back_test( "test", "memo", Time.gm( 2008, 8, 21 ).to_i, Time.gm( 2008, 8, 21, 3 ).to_i, agents )["id"]
+#    sleep 10
+#
+#    # プロセスを取得
+#    process = @mng.get( pid )
+#    assert_equals process.state, :RUNNING
+#    assert_not_nil process.progress
+#    assert_equals process.id, pid
+#
+#    #別のプロセスを追加
+#    pid2 = @mng.create_back_test( "test2", "memo", Time.gm( 2008, 8, 21 ).to_i, Time.gm( 2008, 8, 21, 3 ).to_i, agents )["id"]
+#    sleep 1
+#
+#    #別のプロセスは待機中になる。
+#    process = @mng.get( pid2 )
+#    assert_equals process.state, :WAITING
+#    assert_not_nil process.progress
+#    assert_equals process.id, pid2
+#
+#    process = @mng.get( pid )
+#    assert_equals process.state, :RUNNING
+#    assert_not_nil process.progress
+#    assert_equals process.id, pid
+#
+#    # 実行完了を待つ
+#    sleep 1 while   @mng.get( pid ).state == :RUNNING
+#
+#    process = @mng.get( pid )
+#    assert_equals process.state, :FINISHED
+#    assert_not_nil process.progress
+#    assert_equals process.id, pid
+#
+#    sleep 1
+#
+#    #別のプロセスが開始される。
+#    process = @mng.get( pid2 )
+#    assert_equals process.state, :RUNNING
+#    assert_not_nil process.progress
+#    assert_equals process.id, pid2
+#
+#    sleep 1 while   @mng.get( pid2 ).state == :RUNNING
+#
+#    process = @mng.get( pid2 )
+#    assert_equals process.state, :FINISHED
+#    assert_not_nil process.progress
+#    assert_equals process.id, pid2
+#
+#  end
+
+  def recreate_registry
+    begin
+      @mng.stop
+       begin
+         @registry.permitter.close
+      ensure
+      @registry.client.close
+      end
+    ensure
+      return @registry = JIJI::Registry.new(@dir , nil)
+    end
+  end
+
+  # プロセスが存在しないことを確認する。
+  def assert_process_not_found( pid )
+     begin
+      @mng.get( pid )
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+  end
+
+end