unageanu-jiji 0.1.0

data/test/all_tests.rb

@@ -0,0 +1,26 @@
+#!/usr/bin/ruby
+
+$: << "../lib"
+
+
+require "runit/testcase"
+require "runit/cui/testrunner"
+
+require "test_BackTestCollector"
+require "test_Collector"
+require "test_SingleClickClient"
+require "test_BlockToSession"
+require "test_AgentManager"
+
+require "test_AgentRegistory"
+require "test_Configuration"
+require "test_CSV"
+require "test_Operator"
+require "test_Output"
+require "test_Output_registry"
+require "test_PeriodicallyAgent"
+require "test_Process"
+require "test_ProcessManager"
+require "test_RateDao"
+require "test_Permitter"
+require "test_TradeResultDao"

data/test/service/all_service_tests.rb

@@ -0,0 +1,10 @@
+#!/usr/bin/ruby
+
+$: << "../lib"
+
+
+require "runit/testcase"
+require "runit/cui/testrunner"
+require "test_AgentService"
+
+

data/test/service/test_AgentService.rb

@@ -0,0 +1,185 @@
+#!/usr/bin/ruby
+
+$: << "../../lib"
+$: << "../../ext_lib"
+$: << "../"
+
+require "runit/testcase"
+require "runit/cui/testrunner"
+require "testutils"
+
+# AgentService のテスト
+class AgentServiceTest < Test::AbstractServiceTest
+  
+  def setup
+    super
+    @as = Test::JsonRpcRequestor.new("agent")
+    
+    # 登録
+    @as.put_agent_class( "AgentServiceTest_1.rb", BODY_O1 )
+    @as.put_agent_class( "AgentServiceTest_2.rb", BODY_O2 )      
+  end
+
+  def teardown
+    
+    # 削除
+    begin
+      @as.delete_agent_class( "AgentServiceTest_1.rb" )
+    ensure
+      @as.delete_agent_class( "AgentServiceTest_2.rb" )      
+    end
+      
+    super
+  end
+  
+  # 基本動作のテスト
+  def test_hello
+    hello_service = Test::JsonRpcRequestor.new("hello")
+    assert_equals hello_service.hello, "hello."
+  end
+  
+  # 
+  def test_basic
+    
+    # 一覧
+    list = @as.list_agent_class
+    agent1   = list.find {|a| a["class_name"] == "TestAgent@AgentServiceTest_1.rb" }
+    agent1_2 = list.find {|a| a["class_name"] == "Test::TestAgent2@AgentServiceTest_1.rb" }
+    agent2   = list.find {|a| a["class_name"] == "TestAgent@AgentServiceTest_2.rb" }
+    
+    assert_equals agent1["description"], "テスト\nテスト"
+    assert_equals agent1["properties"], [
+      {"default"=>10,"type"=>"number","id"=>"period","name"=>"レートの通知を受け取る間隔(分)"},
+      {"default"=>1.0,"type"=>"string","id"=>"a","name"=>"aaa"},
+      {"default"=>"foo","type"=>"string","id"=>"b","name"=>"bbb"}
+    ]
+    assert_equals agent1_2["description"], "テスト2"
+    assert_equals agent1_2["properties"], []
+    assert_equals agent2["description"], "テスト"
+    assert_equals agent2["properties"], [
+      {"default"=>10,"type"=>"number","id"=>"period","name"=>"レートの通知を受け取る間隔(分)"},
+      {"default"=>1.0,"type"=>"string","id"=>"x","name"=>"aaa"},
+      {"default"=>"foo","type"=>"string","id"=>"y","name"=>"bbb"}
+    ]
+    
+    # 削除
+    @as.delete_agent_class( "AgentServiceTest_1.rb" )
+    list = @as.list_agent_class
+    agent1   = list.find {|a| a["class_name"] == "TestAgent@AgentServiceTest_1.rb" }
+    agent1_2 = list.find {|a| a["class_name"] == "Test::TestAgent2@AgentServiceTest_1.rb" }
+    agent2   = list.find {|a| a["class_name"] == "TestAgent@AgentServiceTest_2.rb" }
+    assert_nil agent1
+    assert_nil agent1_2
+    assert_not_nil agent2
+    
+    # 再登録
+    @as.put_agent_class( "AgentServiceTest_1.rb", BODY_O1 )
+    
+    # RMTのエージェントとして登録
+    begin
+	    @as.add_agent( "rmt", "AgentServiceTest_その1", 
+	      "TestAgent@AgentServiceTest_1.rb", {"a"=>100, "b"=>"value", "period"=>610} )
+	    @as.add_agent( "rmt", "AgentServiceTest_その2",
+	      "TestAgent@AgentServiceTest_2.rb", {"x"=>200} )
+	    
+	    agents = @as.list_agent("rmt")
+      a1 = agents.find {|a| a["name"] == "AgentServiceTest_その1" }
+      a2 = agents.find {|a| a["name"] == "AgentServiceTest_その2" }
+      
+      assert_equals a1["properties"], [
+        {"id"=>"period", "value"=>610, "info"=>{"default"=>10,"type"=>"number","id"=>"period","name"=>"レートの通知を受け取る間隔(分)"}},
+        {"id"=>"a",  "value"=>100, "info"=>{"default"=>1.0,"type"=>"string","id"=>"a","name"=>"aaa"}},
+        {"id"=>"b",  "value"=>"value", "info"=>{"default"=>"foo","type"=>"string","id"=>"b","name"=>"bbb"}}
+      ]
+      assert_equals a1["active"], true
+      assert_equals a1["description"], "テスト\nテスト"
+      
+      assert_equals a2["properties"], [
+        {"id"=>"period", "value"=>nil, "info"=>{"default"=>10,"type"=>"number","id"=>"period","name"=>"レートの通知を受け取る間隔(分)"}},
+        {"id"=>"x",  "value"=>200, "info"=>{"default"=>1.0,"type"=>"string","id"=>"x","name"=>"aaa"}},
+        {"id"=>"y",  "value"=>nil, "info"=>{"default"=>"foo","type"=>"string","id"=>"y","name"=>"bbb"}}
+      ]
+      assert_equals a2["active"], true
+      assert_equals a2["description"], "テスト"      
+      
+      # 状態変更
+      assert_equals @as.on?( "rmt", "AgentServiceTest_その1" ), true
+      assert_equals @as.on?( "rmt", "AgentServiceTest_その2" ), true
+      
+      @as.off( "rmt", "AgentServiceTest_その1" )
+      assert_equals @as.on?( "rmt", "AgentServiceTest_その1" ), false
+      assert_equals @as.on?( "rmt", "AgentServiceTest_その2" ), true
+      
+      @as.on( "rmt", "AgentServiceTest_その1" )
+      assert_equals @as.on?( "rmt", "AgentServiceTest_その1" ), true
+      assert_equals @as.on?( "rmt", "AgentServiceTest_その2" ), true
+      
+    ensure
+      begin
+	      @as.remove_agent( "rmt", "AgentServiceTest_その1" )
+	      agents = @as.list_agent("rmt")
+	      a1 = agents.find {|a| a["name"] == "AgentServiceTest_その1" }
+	      a2 = agents.find {|a| a["name"] == "AgentServiceTest_その2" }      
+	      assert_nil a1
+	      assert_not_nil a2
+      ensure
+	      @as.remove_agent( "rmt", "AgentServiceTest_その2" )
+	      agents = @as.list_agent("rmt")
+	      a1 = agents.find {|a| a["name"] == "AgentServiceTest_その1" }
+	      a2 = agents.find {|a| a["name"] == "AgentServiceTest_その2" }      
+	      assert_nil a1
+	      assert_nil a2
+      end
+    end
+  end
+  
+  BODY_O1 =<<-BODY
+    class TestAgent < JIJI::PeriodicallyAgent
+      include JIJI::Agent
+      def initialize
+        @a = 1
+        @b = "foo"
+      end
+      def property_infos
+        super().concat [
+          Property.new( "a", "aaa", 1 ),
+          Property.new( "b", "bbb", "foo" )
+        ]
+      end
+      def description
+        "テスト\nテスト"
+      end
+      attr :a, true
+      attr :b, true
+    end
+    module Test
+      class TestAgent2
+        include JIJI::Agent
+        def description
+          "テスト2"
+        end
+      end
+    end
+  BODY
+  BODY_O2 =<<-BODY
+    class TestAgent < JIJI::PeriodicallyAgent
+      include JIJI::Agent
+      def initialize
+        @x = 1.0
+        @y = "foo"
+      end
+      def property_infos
+        super().concat [
+          Property.new( "x", "aaa", 1.0 ),
+          Property.new( "y", "bbb", "foo" )
+        ]
+      end
+      def description
+        "テスト"
+      end
+      attr :x, true
+      attr :y, true
+    end
+  BODY
+end
+

data/test/service/test_OutputService.rb

@@ -0,0 +1,33 @@
+#!/usr/bin/ruby
+
+$: << "../../lib"
+$: << "../../ext_lib"
+$: << "../"
+
+require "runit/testcase"
+require "runit/cui/testrunner"
+require "testutils"
+
+# OutputService のテスト
+class OutputServiceTest < Test::AbstractServiceTest
+  
+  def setup
+    super
+    @s = Test::JsonRpcRequestor.new("output")     
+  end
+
+  def teardown
+    super
+  end
+  
+  # レート一覧取得
+  def test_basic
+      
+    puts "---"
+    
+    list = @s.list_outputs("rmt")
+    puts list.join(",")
+    
+  end
+end
+

data/test/service/test_RateService.rb

@@ -0,0 +1,56 @@
+#!/usr/bin/ruby
+
+$: << "../../lib"
+$: << "../../ext_lib"
+$: << "../"
+
+require "runit/testcase"
+require "runit/cui/testrunner"
+require "testutils"
+
+# RateService のテスト
+class RateServiceTest < Test::AbstractServiceTest
+  
+  def setup
+    super
+    @rs = Test::JsonRpcRequestor.new("rate")     
+  end
+
+  def teardown
+    super
+  end
+  
+  # レート一覧取得
+  def test_basic
+    
+    ["1m", "10m", "6h","1d", "5d"].each {|s|
+      
+      puts "---" + s
+      
+      start = Time.now
+      list = @rs.list( :EURJPY, s, Time.local(2008,8, 23).to_i, 120)
+      puts "time:" + ( Time.now.to_f - start.to_f ).to_s
+      
+      puts list.length
+      p list[0]
+      
+      #assert_equals list.length, 10
+      list.each {|item|
+        assert_equals item.length, 6
+        #p item
+      }
+      
+#      list = @rs.list_next( :EURJPY, s, Time.local(2008,8,24).to_i )
+#      
+#      puts list.length
+#      p list[0]      
+#      assert_equals list.length > 0, true
+#      list.each {|item|
+#        assert_equals item.length, 6
+#        #p item
+#      }
+    }
+    
+  end
+end
+

data/test/test_AgentManager.rb

@@ -0,0 +1,307 @@
+#!/usr/bin/ruby
+
+$: << "../lib"
+
+
+require "runit/testcase"
+require "runit/cui/testrunner"
+require 'jiji/agent/agent'
+require 'jiji/agent/agent_manager'
+require 'jiji/agent/agent_registry'
+require 'jiji/agent/permitter'
+require 'jiji/agent/util'
+require "jiji/configuration"
+require "jiji/collector"
+require "jiji/dao/trade_result_dao"
+require "fileutils"
+require "testutils"
+require "logger"
+
+# AgentManager のテスト
+class AgentManagerTest <  RUNIT::TestCase
+
+  include JIJI::AgentUtil
+  include Test::Constants
+
+  def setup
+    @dir = File.dirname(__FILE__) + "/agents"
+    FileUtils.mkdir_p @dir
+
+    @registry = JIJI::AgentRegistry.new( "#{@dir}/agents", "#{@dir}/sharedLib" )
+    @registry.conf = CONF
+
+    @permitter = JIJI::Permitter.new( 5, 0 )
+    logger = Logger.new STDOUT
+    logger = @permitter.proxy( logger, [/^(info|debug|warn|error|fatal)$/] )
+    @client = JIJI::SingleClickClient.new( ACCOUNT, CONF, logger )
+
+    @registory_mock = Object.new
+    class << @registory_mock
+      def output( agent_name, dir )
+        output = JIJI::Output.new( agent_name, "#{dir}/out" )
+        output = @permitter.proxy( output, [/^(get|each|put|<<)$/], [/^get$/] )
+        return output
+      end
+      def permitter=(permitter)
+        @permitter = permitter
+      end
+    end
+    @registory_mock.permitter = @permitter
+
+    operator = JIJI::RmtOperator.new( @client, logger, nil )
+    operator.conf = CONF
+    operator = @permitter.proxy( operator, [/^(sell|buy|commit)$/],[/^(sell|buy)$/])
+    trade_result_dao = JIJI::Dao::TradeResultDao.new( "#{@dir}/trade_result" )
+
+    @m = JIJI::AgentManager.new( "test_agentManager", @registry, logger)
+    @m.operator = operator
+    @m.conf = CONF
+    @m.trade_result_dao = trade_result_dao
+    @m.client = @client
+    @m.registory = @registory_mock
+  end
+
+  def teardown
+    @client.close
+    @permitter.close
+    FileUtils.rm_rf @dir
+  end
+
+  # 基本動作のテスト
+  def test_basic
+    str =<<-AGENT_STR
+      class Foo
+        include JIJI::Agent
+        def initialize( )
+          @log = []
+        end
+        def next_rates( rates )
+          @log << rates[:EURJPY].bid
+        end
+        attr :log, true
+      end
+    AGENT_STR
+    @registry.add_file( "foo.rb", str )
+
+    agent1 = @registry.create( "Foo@foo.rb" )
+    agent2 = @registry.create( "Foo@foo.rb" )
+
+    @m.add( "テスト", agent1 )
+    @m.add( "テスト2", agent2 )
+
+    # 追加直後はactive
+    assert_equals @m.on?( "テスト" ), true
+    assert_equals @m.on?( "テスト2" ), true
+
+    pair_infos = {
+      ClickClient::FX::EURJPY => Info.new( 10000 )
+    }
+    @m.next_rates( JIJI::Rates.new( pair_infos, {
+      :EURJPY => Rate.new( 100.0,100.05,-200,200,100 )
+    }))
+    @m.next_rates( JIJI::Rates.new( pair_infos, {
+      :EURJPY => Rate.new( 110.0,110.05,-200,200,100 )
+    }))
+    assert_equals [[100.0, 110.0],[100.0, 110.0]], safe(4) {
+      value = [agent1.log, agent2.log]
+      agent1.log = []
+      agent2.log = []
+      value
+    }
+
+    # agent1をoff
+    @m.off( "テスト" )
+    assert_equals @m.on?( "テスト" ), false
+    assert_equals @m.on?( "テスト2" ), true
+    @m.next_rates( JIJI::Rates.new( pair_infos, {
+      :EURJPY => Rate.new( 100.0,100.05,-200,200,100 )
+    }))
+    assert_equals [[],[100.0]], safe(4) {
+      value = [agent1.log, agent2.log]
+      agent1.log = []
+      agent2.log = []
+      value
+    }
+
+    # agent1を再度onにする
+    @m.on( "テスト" )
+    assert_equals @m.on?( "テスト" ), true
+    assert_equals @m.on?( "テスト2" ), true
+    @m.next_rates( JIJI::Rates.new( pair_infos, {
+      :EURJPY => Rate.new( 100.0,100.05,-200,200,100 )
+    }))
+    assert_equals [[100.0],[100.0]], safe(4) {
+      value = [agent1.log, agent2.log]
+      agent1.log = []
+      agent2.log = []
+      value
+    }
+
+    # agent1を削除
+    @m.remove( "テスト" )
+    assert_equals @m.on?( "テスト2" ), true
+    @m.next_rates( JIJI::Rates.new( pair_infos, {
+      :EURJPY => Rate.new( 100.0,100.05,-200,200,100 )
+    }))
+    assert_equals [[],[100.0]], safe(4) {
+      value = [agent1.log, agent2.log]
+      agent1.log = []
+      agent2.log = []
+      value
+    }
+  end
+
+  def test_error
+
+    # 存在しないエージェントの操作
+    begin
+      @m.remove( "not found" )
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @m.on( "not found" )
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @m.off( "not found" )
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @m.on?( "not found" )
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+
+    # エージェントの2重登録
+    @m.add( "テスト", TestAgent.new.taint )
+    begin
+      @m.add( "テスト", TestAgent.new.taint )
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_ALREADY_EXIST, $!.code
+    end
+  end
+
+  class TestAgent
+    include JIJI::Agent
+  end
+
+  # outputsとoperatorを使ってみるテスト
+  def test_use_outputs
+    str =<<-AGENT_STR
+      class Foo < JIJI::PeriodicallyAgent
+        include JIJI::Agent
+        def initialize( )
+          super(0.1)
+        end
+        def next_period_rates( rates )
+
+          out1 = output.get( "foo" )
+          out1.put( :a, "aaaa" )
+          out1.put( :b, "bbbb" )
+
+          out2 = output.get( "var", :graph )
+          out2 << 100
+          out2 << 200
+
+          logger.info "test"
+          logger.warn IOError.new
+          logger.error IOError.new
+          logger.debug "test"
+
+#          p1 = operator.sell(1)
+#          p1.commit
+#          p2 = operator.buy(1)
+#          p2.commit
+        end
+        attr :log, true
+      end
+    AGENT_STR
+    @registry.add_file( "foo.rb", str )
+
+    agent = @registry.create( "Foo@foo.rb" )
+    @m.add( "テスト", agent )
+
+    pair_infos = {
+      ClickClient::FX::EURJPY => Info.new( 10000 )
+    }
+    2.times {
+	    @m.next_rates( JIJI::Rates.new( pair_infos, {
+	      :EURJPY => Rate.new( 100.0,100.05,-200,200,100 )
+	    }))
+      sleep 0.5
+    }
+
+  end
+
+  # 循環参照のテスト
+  def test_recursive
+    @registry.add_file( "AgentServiceTest_1.rb", BODY_O1 )
+    @registry.add_file( "AgentServiceTest_2.rb", BODY_O2 )
+    @m.each {|a|}
+    @registry.inject([]) {|list,name|
+      list << {
+        "class_name"=>name,
+        "properties"=>@registry.get_property_infos(name),
+        "description"=>@registry.get_description(name)
+      }
+      list
+    }
+  end
+  BODY_O1 =<<-BODY
+    class TestAgent < JIJI::PeriodicallyAgent
+      include JIJI::Agent
+      def initialize
+        @a = 1
+        @b = "foo"
+      end
+      def property_infos
+        super().concat [
+          Property.new( "a", "aaa", 1 ),
+          Property.new( "b", "bbb", "foo" )
+        ]
+      end
+      def description
+        "テスト\nテスト"
+      end
+      attr :a, true
+      attr :b, true
+    end
+    module Test
+      class TestAgent2
+        include JIJI::Agent
+        def description
+          "テスト2"
+        end
+      end
+    end
+  BODY
+  BODY_O2 =<<-BODY
+    class TestAgent < JIJI::PeriodicallyAgent
+      include JIJI::Agent
+      def initialize
+        @x = 1.0
+        @y = "foo"
+      end
+      def property_infos
+        super().concat [
+          Property.new( "x", "aaa", 1.0 ),
+          Property.new( "y", "bbb", "foo" )
+        ]
+      end
+      def description
+        "テスト"
+      end
+      attr :x, true
+      attr :y, true
+    end
+  BODY
+end
+