jiji 1.2.5

data/test/agent/test_PeriodicallyAgent.rb

@@ -0,0 +1,180 @@
+#!/usr/bin/ruby
+
+$: << "../lib"
+
+require 'rubygems'
+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/collector'
+
+# PeriodicallyAgent のテスト
+class PeriodicallyAgentTest <  RUNIT::TestCase
+
+  def setup
+  end
+
+  def teardown
+  end
+  
+  # 基本動作のテスト
+  def test_basic
+    
+    agent = TestAgent.new( 2 )
+    assert_equals agent.log, []
+    
+    pair_infos = {:USDJPY=>"info"}
+    
+    agent.next_rates(  JIJI::Rates.new( pair_infos, {
+      :USDJPY => R.new( 100.0, 100.05, -200, 222 ),
+      :AUDUSD => R.new( 100.0, 100.05, -200, 200 )
+    }, Time.local(2008, 8, 1, 10, 0, 0)))
+    agent.next_rates( JIJI::Rates.new( pair_infos, {
+      :USDJPY => R.new( 100.4, 100.09, -200, 200 ),
+      :AUDUSD => R.new(  90.5, 100.00, -330, 330 )
+    }, Time.local(2008, 8, 1, 10, 0, 1)))
+    agent.next_rates( JIJI::Rates.new( pair_infos, {
+      :USDJPY => R.new( 90.6, 110.01, -220, 220 ),
+      :AUDUSD => R.new( 90.5, 100.00, -300, 300 )
+    }, Time.local(2008, 8, 1, 10, 1, 0)))
+    
+    # まだ通知されない
+    assert_equals agent.log, []
+    
+    # さらに待ち、データを追加
+    agent.next_rates( JIJI::Rates.new( pair_infos, {
+      :USDJPY => R.new( 90.7, 110.02, -219, 222 ),
+      :AUDUSD => R.new( 90.6, 100.03, -303, 303 )
+    }, Time.local(2008, 8, 1, 10, 2, 1)))
+    # 通知される
+    assert_equals agent.log.length, 1
+    r = agent.log[0]
+    assert_not_nil r.time
+    assert_not_nil r.start_time
+    assert_not_nil r.end_time
+    assert_equals r.pair_infos, {:USDJPY=>"info"}
+    
+    assert_equals r[:USDJPY].bid.max,   100.4
+    assert_equals r[:USDJPY].bid.min,   90.6
+    assert_equals r[:USDJPY].bid.start, 100.0
+    assert_equals r[:USDJPY].bid.end,   90.7
+    
+    assert_equals r[:USDJPY].ask.max,   110.02
+    assert_equals r[:USDJPY].ask.min,   100.05
+    assert_equals r[:USDJPY].ask.start, 100.05
+    assert_equals r[:USDJPY].ask.end,   110.02
+    
+    assert_equals r[:USDJPY].sell_swap.max,   -200
+    assert_equals r[:USDJPY].sell_swap.min,   -220
+    assert_equals r[:USDJPY].sell_swap.start, -200
+    assert_equals r[:USDJPY].sell_swap.end,   -219
+    
+    assert_equals r[:USDJPY].buy_swap.max,   222
+    assert_equals r[:USDJPY].buy_swap.min,   200
+    assert_equals r[:USDJPY].buy_swap.start, 222
+    assert_equals r[:USDJPY].buy_swap.end,   222
+    
+    assert_equals r[:AUDUSD].bid.max,   100.0
+    assert_equals r[:AUDUSD].bid.min,   90.5
+    assert_equals r[:AUDUSD].bid.start, 100.0
+    assert_equals r[:AUDUSD].bid.end,   90.6
+    
+    assert_equals r[:AUDUSD].ask.max,   100.05
+    assert_equals r[:AUDUSD].ask.min,   100.00
+    assert_equals r[:AUDUSD].ask.start, 100.05
+    assert_equals r[:AUDUSD].ask.end,   100.03
+    
+    assert_equals r[:AUDUSD].sell_swap.max,   -200
+    assert_equals r[:AUDUSD].sell_swap.min,   -330
+    assert_equals r[:AUDUSD].sell_swap.start, -200
+    assert_equals r[:AUDUSD].sell_swap.end,   -303
+    
+    assert_equals r[:AUDUSD].buy_swap.max,   330
+    assert_equals r[:AUDUSD].buy_swap.min,   200
+    assert_equals r[:AUDUSD].buy_swap.start, 200
+    assert_equals r[:AUDUSD].buy_swap.end,   303
+    
+    agent.log.clear
+    
+    
+    # データを追加
+    agent.next_rates( JIJI::Rates.new( pair_infos, {
+      :USDJPY => R.new( 100.4, 100.09, -200, 200 ),
+      :AUDUSD => R.new(  90.5, 100.00, -330, 330 )
+    }, Time.local(2008, 8, 1, 10, 3, 0)))
+    
+    assert_equals agent.log, []
+    
+    agent.next_rates( JIJI::Rates.new( pair_infos, {
+      :USDJPY => R.new( 90.6, 110.01, -220, 220 ),
+      :AUDUSD => R.new( 90.5, 100.00, -300, 300 )
+    }, Time.local(2008, 8, 1, 10, 4, 2)))
+    
+    # 通知される
+    assert_equals agent.log.length, 1
+    r = agent.log[0]
+    assert_not_nil r.time
+    assert_not_nil r.start_time
+    assert_not_nil r.end_time
+    assert_equals r.pair_infos, {:USDJPY=>"info"}
+    
+    assert_equals r[:USDJPY].bid.max,   100.4
+    assert_equals r[:USDJPY].bid.min,   90.6
+    assert_equals r[:USDJPY].bid.start, 100.4
+    assert_equals r[:USDJPY].bid.end,   90.6
+    
+    assert_equals r[:USDJPY].ask.max,   110.01
+    assert_equals r[:USDJPY].ask.min,   100.09
+    assert_equals r[:USDJPY].ask.start, 100.09
+    assert_equals r[:USDJPY].ask.end,   110.01
+    
+    assert_equals r[:USDJPY].sell_swap.max,   -200
+    assert_equals r[:USDJPY].sell_swap.min,   -220
+    assert_equals r[:USDJPY].sell_swap.start, -200
+    assert_equals r[:USDJPY].sell_swap.end,   -220
+    
+    assert_equals r[:USDJPY].buy_swap.max,   220
+    assert_equals r[:USDJPY].buy_swap.min,   200
+    assert_equals r[:USDJPY].buy_swap.start, 200
+    assert_equals r[:USDJPY].buy_swap.end,   220
+    
+    assert_equals r[:AUDUSD].bid.max,   90.5
+    assert_equals r[:AUDUSD].bid.min,   90.5
+    assert_equals r[:AUDUSD].bid.start, 90.5
+    assert_equals r[:AUDUSD].bid.end,   90.5
+    
+    assert_equals r[:AUDUSD].ask.max,   100.00
+    assert_equals r[:AUDUSD].ask.min,   100.00
+    assert_equals r[:AUDUSD].ask.start, 100.00
+    assert_equals r[:AUDUSD].ask.end,   100.00
+    
+    assert_equals r[:AUDUSD].sell_swap.max,   -300
+    assert_equals r[:AUDUSD].sell_swap.min,   -330
+    assert_equals r[:AUDUSD].sell_swap.start, -330
+    assert_equals r[:AUDUSD].sell_swap.end,   -300
+    
+    assert_equals r[:AUDUSD].buy_swap.max,   330
+    assert_equals r[:AUDUSD].buy_swap.min,   300
+    assert_equals r[:AUDUSD].buy_swap.start, 330
+    assert_equals r[:AUDUSD].buy_swap.end,   300
+    
+  end
+  
+end
+
+class TestAgent < JIJI::PeriodicallyAgent
+  def initialize( period )
+    super
+    @log = []
+  end
+  def next_period_rates( rates )
+    @log << rates
+  end
+  attr :log, true 
+end
+
+R = Struct.new( :bid, :ask, :sell_swap, :buy_swap )

data/test/agent/test_Permitter.rb

@@ -0,0 +1,217 @@
+#!/usr/bin/ruby
+
+$: << "../lib"
+
+require 'rubygems'
+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 'fileutils'
+require 'test_utils'
+require 'logger'
+
+# Permitter のテスト
+class PermitterTest <  RUNIT::TestCase
+
+  include JIJI::AgentUtil
+  include Test::Constants
+
+  def setup
+    @dir = File.dirname(__FILE__) + "/PermitterTest"
+    FileUtils.mkdir_p @dir
+
+    @logger = Logger.new STDOUT
+    @registry = JIJI::AgentRegistry.new( "#{@dir}/agents", "#{@dir}/shared" )
+    @registry.conf = CONF
+
+    @r = JIJI::Permitter.new(5, 2)
+  end
+
+  def teardown
+    @r.close
+    FileUtils.rm_rf @dir
+  end
+
+  # 基本動作のテスト
+  def test_basic
+
+    untaint = UntaintClass.new
+
+    # 汚染されたクラスのインスタンスを生成
+    PermitterTestTaint.taint
+    test = safe(4) {
+      str =<<-STR
+      class TaintClass
+        def test(args)
+          return args.test("a")
+        end
+        def test2(args)
+          return args.test2("a")
+        end
+      end
+      STR
+      PermitterTestTaint.module_eval(str)
+      PermitterTestTaint::TaintClass.new
+    }
+
+    # test.testはセーフレベル4で実行されるため、
+    # untaint内のFile.exist?でセキュリティエラーになる
+    begin
+      safe(1) {
+	      test.test untaint
+	      fail
+      }
+    rescue SecurityError
+    end
+
+    # プロキシをはさむ
+    with_proxy = @r.proxy( untaint, [/^test$/] )
+    # 実行できるようになる
+    assert_equals "ab", safe(4) {
+      test.test( with_proxy )
+    }
+    # 許可されていないメソッドは使えない
+    begin
+      safe(4) {
+        test.test2( with_proxy )
+        fail
+      }
+    rescue SecurityError
+    end
+
+    # 例外を返すパターン
+    with_proxy2 = @r.proxy( UntaintClass2.new, [/^test$/] )
+    safe(4) {
+      begin
+        test.test( with_proxy2 )
+        fail
+      rescue IOError
+      end
+    }
+    with_proxy3 = @r.proxy( UntaintClass3.new, [/^test$/] )
+    safe(4) {
+      begin
+        test.test( with_proxy3 )
+        fail
+      rescue LoadError
+      end
+    }
+  end
+
+  # 戻り値にプロキシを適用するテスト
+  def test_proxy_result
+    untaint = UntaintClass.new
+
+    # 汚染されたクラスのインスタンスを生成
+    PermitterTestTaint.taint
+    test = safe(4) {
+      str =<<-STR
+      class TaintClass2
+        def test(args)
+          result = args.return_value
+          result.test("x")
+        end
+      end
+      STR
+      PermitterTestTaint.module_eval(str)
+      PermitterTestTaint::TaintClass2.new
+    }
+
+    # プロキシをはさむ
+    with_proxy = @r.proxy( untaint, [/^(return\_value|test)$/], [/^return\_value$/] )
+    assert_equals "xb", safe(4) {
+      test.test( with_proxy )
+    }
+
+    # 戻り値に適用されていない場合は実行できない
+    untaint = UntaintClass.new
+    with_proxy = @r.proxy( untaint, [/^(return_value|test)$/] )
+    begin
+      safe(4) {
+        test.test with_proxy
+        fail
+      }
+    rescue SecurityError
+    end
+  end
+
+  # セーフレベルの高い環境で、プロキシを使ってみるテスト
+  def test_use_proxy_in_safe4
+
+
+    # 汚染されたクラスのインスタンスを生成
+    PermitterTestTaint.taint
+    test = safe(4) {
+      str =<<-STR
+      class Foo
+        def test
+          File.exist? "./test"
+        end
+      end
+      class TaintClass3
+        def test(req)
+          # プロキシを設定して、セーフレベルを下げて実行しようとする
+          req.proxy( Foo.new, [/.*/] ).test
+        end
+      end
+      STR
+      PermitterTestTaint.module_eval(str)
+      PermitterTestTaint::TaintClass3.new
+    }
+
+    # エラーになるのを確認
+    safe(4) {
+      begin
+        test.test(@r)
+        fail
+      rescue SecurityError
+      end
+    }
+  end
+
+end
+
+class UntaintClass
+  def test(args)
+#
+
+    # 汚れたクラスのインスタンス内で作成されたオブジェクトは汚染されている。
+    # そのままではFile.exist?などの引数としては使えないので、必要に応じて
+    # 安全性をチェックした後untaintすること。
+    begin
+      File.exist? args
+      raise "fail"
+    rescue SecurityError
+    end
+    # untaintすれば引数にできる
+    File.exist? args.untaint
+    return args + "b"
+  end
+  def test2(args)
+    File.exist? "./test"
+    return args + "b"
+  end
+  def return_value
+    File.exist? "./test"
+    return UntaintClass.new
+  end
+end
+class UntaintClass2
+  def test(args)
+    raise IOError.new
+  end
+end
+class UntaintClass3
+  def test(args)
+    raise LoadError.new
+  end
+end
+
+module PermitterTestTaint
+end

data/test/all_tests.rb

@@ -0,0 +1,24 @@
+#!/usr/bin/ruby
+
+$: << "../lib"
+
+require 'rubygems'
+require 'runit/testcase'
+require 'runit/cui/testrunner'
+
+require 'test_ProcessManager'
+require 'test_BackTestCollector'
+require 'test_Collector'
+require 'test_Configuration'
+require 'test_Operator'
+require 'test_Output'
+require 'test_Output_registry'
+require 'test_OutputManager'
+require 'test_Process'
+
+require 'agent/agent_tests'
+require 'dao/dao_tests'
+require 'migration/migration_tests'
+require 'plugin/plugin_tests'
+require 'shared/shared_tests'
+require 'util/util_tests'

data/test/dao/dao_tests.rb

@@ -0,0 +1,9 @@
+#!/usr/bin/ruby
+
+require 'rubygems'
+require 'runit/testcase'
+require 'runit/cui/testrunner'
+
+require 'dao/test_FileSystemDao'
+require 'dao/test_RateDao'
+require 'dao/test_TradeResultDao'