unageanu-jiji 0.1.0

data/test/test_AgentRegistory.rb

@@ -0,0 +1,566 @@
+#!/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 "fileutils"
+
+# AgentRegistory のテスト
+class AgentRegistryTest <  RUNIT::TestCase
+  include JIJI::AgentUtil
+  def setup
+    @dir = File.dirname(__FILE__) + "/agents"
+    FileUtils.mkdir_p @dir
+
+    @registry = JIJI::AgentRegistry.new( "#{@dir}/agent", "#{@dir}/sharedlib" )
+    @registry.conf = JIJI::Configuration.new
+  end
+
+  def teardown
+    FileUtils.rm_rf @dir
+  end
+
+  # 基本動作のテスト
+  def test_basic
+
+    # 最初は空
+    assert_equals @registry.collect, []
+    assert_equals @registry.files, []
+    assert_equals @registry.files(:shared_lib), []
+
+    # 追加
+    foo_body = "class Foo; include JIJI::Agent; end"
+    var_body = <<-BODY
+    class Var
+      include JIJI::Agent
+      def test
+        [JIJI::Agent::Shared::VALUE, JIJI::Agent::Shared::SharedLib.new, JIJI::Agent::Shared.m] # 共有ライブラリのAPIを使用。
+      end
+    end
+BODY
+    
+    assert_equals @registry.add_file( "foo.rb", foo_body ), :success
+    assert_equals @registry.add_file( "var.rb", var_body ), :success
+
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb", "Var@var.rb"]
+    assert_agent_files @registry.files, ["foo.rb","var.rb"]
+    assert_agent_files @registry.files(:shared_lib), []
+
+
+    # 取得
+    cl = @registry.get "Foo@foo.rb"
+    assert_not_nil cl
+    cl.new # インスタンスを生成できるのを確認
+
+    cl = @registry.get "Var@var.rb"
+    assert_not_nil cl
+    obj = safe(4) {cl.new}
+    begin
+      obj.test
+      fail
+    rescue
+    end
+
+    # 共有ライブラリを追加。
+    shared_lib_body = <<-BODY
+    VALUE="val"
+    class SharedLib
+      def foo
+        "fooo" # 共有ライブラリのAPIを使用。
+      end
+    end
+    def self.m
+      "method called."
+    end
+BODY
+    assert_equals @registry.add_file( "shared_lib.rb", shared_lib_body, :shared_lib ), :success
+
+    # 一覧
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb", "Var@var.rb"]
+    assert_agent_files @registry.files, ["foo.rb","var.rb"]
+    assert_agent_files @registry.files(:shared_lib), ["shared_lib.rb"]
+
+    # 共有ライブラリを呼び出すAPIが利用可能になる。
+    assert_equals( safe(4) { obj.test[0] }, "val")
+    assert_equals( safe(4) { obj.test[1].foo}, "fooo" )
+    assert_equals( safe(4) { obj.test[2] }, "method called.")
+
+    begin
+      @registry.get("Not_Found@notfound.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @registry.get("Not_Found@foo.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @registry.get("Not::Not_Found@foo.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+
+    #ファイル取得
+    assert_equals @registry.get_file("foo.rb"), foo_body + "\n"
+    assert_equals @registry.get_file("var.rb"), var_body
+    assert_equals @registry.get_file("shared_lib.rb", :shared_lib), shared_lib_body
+    begin
+      @registry.get_file("Not_Found")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @registry.get_file("Not_Found", :shared_lib)
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @registry.get_file("foo.rb", :shared_lib)
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    
+    # 上書き更新
+     var_body = <<-BODY
+    class Var2
+      include JIJI::Agent
+      def test
+        [JIJI::Agent::Shared::VALUE, JIJI::Agent::Shared::SharedLib.new, "xx"] # 共有ライブラリのAPIを使用。
+      end
+    end
+BODY
+    assert_equals @registry.add_file( "var.rb", var_body ), :success
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb", "Var2@var.rb"]
+    assert_agent_files @registry.files, ["foo.rb","var.rb"]
+
+    assert_equals @registry.get_file("foo.rb"), foo_body + "\n"
+    assert_equals @registry.get_file("var.rb"), var_body
+    assert_equals @registry.get_file("shared_lib.rb", :shared_lib), shared_lib_body
+
+    cl = @registry.get "Var2@var.rb"
+    assert_not_nil cl
+    obj = @registry.create("Var2@var.rb",{})
+
+    assert_equals( safe(4) { obj.test[0] }, "val")
+    assert_equals( safe(4) { obj.test[1].foo}, "fooo" )
+    assert_equals( safe(4) { obj.test[2] }, "xx")
+    
+    # 古いクラスは利用不可
+    begin
+      @registry.get("Var@var.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+
+    # レジストリを再作成
+    @registry = JIJI::AgentRegistry.new( "#{@dir}/agent", "#{@dir}/sharedlib" )
+    @registry.conf = JIJI::Configuration.new
+    assert_equals @registry.collect, []
+    assert_agent_files @registry.files, ["foo.rb","var.rb"]
+    assert_agent_files @registry.files(:shared_lib), ["shared_lib.rb"]
+    
+    # 再読み込み
+    @registry.load_all
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb", "Var2@var.rb"]
+    assert_agent_files @registry.files, ["foo.rb","var.rb"]
+    assert_agent_files @registry.files(:shared_lib), ["shared_lib.rb"]
+    
+    # 共有ライブラリを削除
+    #ライブラリ内のクラスはレジストリ自体を再作成するまで利用可能
+    @registry.remove_file( "shared_lib.rb", :shared_lib )
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb", "Var2@var.rb"]
+    assert_agent_files @registry.files, ["foo.rb","var.rb"]
+    assert_agent_files @registry.files(:shared_lib), []
+    
+    cl = @registry.get "Var2@var.rb"
+    assert_not_nil cl
+    obj = @registry.create("Var2@var.rb",{})
+    assert_equals( safe(4) { obj.test[0] }, "val")
+    assert_equals( safe(4) { obj.test[1].foo}, "fooo" )
+    assert_equals( safe(4) { obj.test[2] }, "xx")
+    
+    begin
+      @registry.remove_file( "NotFound", :shared_lib )
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    
+    # 削除
+    @registry.remove_file( "var.rb" )
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb"]
+    assert_agent_files @registry.files, ["foo.rb"]
+    assert_agent_files @registry.files(:shared_lib), []
+    
+    begin
+      @registry.remove_file( "NotFound" )
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb"]
+    assert_agent_files @registry.files, ["foo.rb"]
+
+    # 削除したクラスは利用不可
+    cl = @registry.get "Foo@foo.rb"
+    assert_not_nil cl
+    begin
+      @registry.get("Var2@var.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+
+    # レジストリを再作成
+    @registry = JIJI::AgentRegistry.new( "#{@dir}/agent", "#{@dir}/sharedlib" )
+    @registry.conf = JIJI::Configuration.new
+    @registry.load_all
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb"]
+    assert_agent_files @registry.files, ["foo.rb"]
+    assert_agent_files @registry.files(:shared_lib), []
+ 
+     cl = @registry.get "Foo@foo.rb"
+    assert_not_nil cl
+ 
+    # 再登録
+    str = <<-CLASS_DEF
+    class Var; include JIJI::Agent; end
+    module Xx
+      class Var2_1; include JIJI::Agent; end
+      class Var2_2; include JIJI::Agent; end
+      module Yy
+        class Var2_3; include JIJI::Agent; end
+      end
+      class X; end
+      Y = "foo"
+    end
+    class Var3; include JIJI::Agent; end
+    class A; end
+    class S
+      include JIJI::Agent
+      def test
+        JIJI::Agent::Shared::VALUE
+      end
+    end
+CLASS_DEF
+    @registry.add_file( "var.rb", str )
+    assert_equals @registry.collect.sort!, ["Foo@foo.rb",  "S@var.rb", "Var3@var.rb", "Var@var.rb", "Xx::Var2_1@var.rb", "Xx::Var2_2@var.rb","Xx::Yy::Var2_3@var.rb"]
+    assert_agent_files @registry.files, ["foo.rb","var.rb"]
+
+    # 取得
+    begin
+      assert_nil @registry.get("A@var.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @registry.get("Xx@var.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @registry.get("Xx::Y@var.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @registry.get("Xx::X@var.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    begin
+      @registry.get("Xx::Yy@var.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+    
+  end
+
+  # エージェントファイル一覧を確認する。
+  def assert_agent_files( files,  expects )
+    assert_equals files.length, expects.length
+    files.each_index { |i|
+      # 名前が取得できることを確認。最終更新日時は設定されていることのみ確認
+      assert_equals expects[i], files[i]["name"]
+      assert_not_nil files[i]["update"]
+    }
+  end
+
+  # プロパティ一覧のテスト
+  def test_properties
+    str =<<-AGENT_STR
+      class Foo
+        include JIJI::Agent
+        def initialize
+          @a = 1
+          @b = "foo"
+          @c = 0.1
+        end
+        def property_infos
+          [
+            Property.new( "a", "aaa", 1 ),
+            Property.new( "b", "bbb", "foo" ),
+            Property.new( "c", "ccc", 0.1 )
+          ]
+        end
+        def description
+          "テスト"
+        end
+        attr :a, true
+        attr :b, true
+        attr :c, true
+      end
+    AGENT_STR
+
+    @registry.add_file( "foo.rb", str )
+
+    ps = @registry.get_property_infos "Foo@foo.rb"
+    assert_equals ps, [
+      JIJI::Agent::Property.new( "a", "aaa", 1 ),
+      JIJI::Agent::Property.new( "b", "bbb", "foo" ),
+      JIJI::Agent::Property.new( "c", "ccc", 0.1 )
+    ]
+
+    agent = @registry.create( "Foo@foo.rb" )
+    assert_equals [1,"foo", 0.1], safe(4){
+      # エージェントはデフォルトではsafeレベル4で実行されるため、汚染されている。
+	    # safeレベル0では呼び出せない。
+      [agent.a, agent.b, agent.c]
+    }
+    agent = @registry.create( "Foo@foo.rb", {
+      "a" => 100, "b" => "test", "c" => 0.34
+    })
+    assert_equals [100,"test", 0.34],safe(4){
+	    [agent.a, agent.b, agent.c]
+    }
+
+    # 説明の取得
+    assert_equals "テスト", @registry.get_description("Foo@foo.rb")
+  end
+
+  # クラス更新のテスト
+  def test_update
+    str = <<-CLASS_DEF
+  module Test
+    class Foo
+      include JIJI::Agent
+      def test
+       "test"
+      end
+    end
+  end
+CLASS_DEF
+    @registry.add_file( "var.rb", str )
+
+    agent = @registry.create "Test::Foo@var.rb"
+    assert_equals "test", safe(4){
+      agent.test
+    }
+
+    # クラスを更新
+    str = <<-CLASS_DEF
+  module Test
+    class Foo
+      include JIJI::Agent
+      def test
+       "test2"
+      end
+    end
+  end
+CLASS_DEF
+    @registry.add_file( "var.rb", str )
+
+    agent2 = @registry.create "Test::Foo@var.rb"
+    assert_equals ["test","test2"], safe(4){
+      [agent.test, agent2.test] # 古いクラスも再作成されるまでは有効
+    }
+    
+    # コンパイルエラーになる場合
+    str = <<-CLASS_DEF
+  module Test
+    class Foo
+      include JIJI::Agent
+      def test
+       "test"
+      end
+    end
+CLASS_DEF
+    result = @registry.add_file( "var.rb", str )
+    puts result
+    assert_not_nil result
+    begin
+      @registry.get("Test::Foo@var.rb")
+      fail
+    rescue JIJI::UserError
+      assert_equals JIJI::ERROR_NOT_FOUND, $!.code
+    end
+  end
+  
+  # 共有ライブラリの更新
+  def test_update_sharedlib
+    
+    # 共有ライブラリを使うクラス
+    @registry.add_file "foo.rb", <<-BODY
+    class Var
+      include JIJI::Agent
+      def init
+        @s = JIJI::Agent::Shared::SharedLib.new
+      end
+      def test
+        [JIJI::Agent::Shared::VALUE, JIJI::Agent::Shared::SharedLib.new, JIJI::Agent::Shared.m] # 共有ライブラリのAPIを使用。
+      end
+      def test2
+        [JIJI::Agent::Shared::VALUE2, JIJI::Agent::Shared::SharedLib2.new, JIJI::Agent::Shared.m2] # 共有ライブラリのAPIを使用。
+      end
+      attr_reader :s
+    end
+BODY
+    
+    obj = @registry.create("Var@foo.rb",{})
+    
+    # まだ共有ライブラリがないのでエラー
+    begin 
+      safe(4) { obj.test } 
+      fail
+    rescue
+    end
+    begin 
+      safe(4) { obj.init } 
+      fail
+    rescue
+  end
+  
+    # 共有ライブラリを定義
+    shared_lib_body = <<-BODY
+    VALUE='val'
+    class SharedLib
+      def foo
+        "fooo"
+      end
+    end
+    def self.m
+      "method called."
+    end
+BODY
+    assert_equals @registry.add_file( "shared_lib.rb", shared_lib_body, :shared_lib ), :success
+    
+    assert_equals safe(4){obj.test[0]},  'val'
+    assert_equals safe(4){obj.test[1].foo},  'fooo'
+    assert_equals safe(4){obj.test[2]},  'method called.'
+    
+    safe(4){ obj.init }
+    assert_equals safe(4){obj.s.foo},  'fooo'
+    
+    obj2 = @registry.create("Var@foo.rb", {})
+    assert_equals safe(4){obj2.test[0]},  'val'
+    assert_equals safe(4){obj2.test[1].foo},  'fooo'
+    assert_equals safe(4){obj2.test[2]},  'method called.'
+    safe(4){ obj2.init }
+    assert_equals safe(4){obj2.s.foo},  'fooo'
+    
+    # テスト2はまだ呼び出せない
+    begin 
+      safe(4) { obj.test2 } 
+      fail
+    rescue; end
+    begin 
+      safe(4) { obj2.test2 } 
+      fail
+    rescue; end
+    
+    # 共有ライブラリ2を追加
+    shared_lib_body = <<-BODY
+    VALUE2=JIJI::Agent::Shared::VALUE + ":" +JIJI::Agent::Shared.m
+    class SharedLib2 < JIJI::Agent::Shared::SharedLib
+      def foo2
+        JIJI::Agent::Shared.m + ":" + JIJI::Agent::Shared::SharedLib.new.foo
+      end
+    end
+    def self.m2
+      JIJI::Agent::Shared.m
+    end
+BODY
+    assert_equals @registry.add_file( "shared_lib2.rb", shared_lib_body, :shared_lib ), :success
+    
+    assert_equals safe(4){obj.test[0]},  'val'
+    assert_equals safe(4){obj.test[1].foo},  'fooo'
+    assert_equals safe(4){obj.test[2]},  'method called.'
+    assert_equals safe(4){obj2.test[0]},  'val'
+    assert_equals safe(4){obj2.test[1].foo},  'fooo'
+    assert_equals safe(4){obj2.test[2]},  'method called.'
+    assert_equals safe(4){obj.s.foo},  'fooo'
+    assert_equals safe(4){obj2.s.foo},  'fooo'
+    
+    assert_equals safe(4){obj.test2[0]},  'val:method called.'
+    assert_equals safe(4){obj.test2[1].foo},  'fooo'
+    assert_equals safe(4){obj.test2[1].foo2},  'method called.:fooo'
+    assert_equals safe(4){obj.test2[2]},  'method called.'
+    assert_equals safe(4){obj2.test2[0]},  'val:method called.'
+    assert_equals safe(4){obj2.test2[1].foo},  'fooo'
+    assert_equals safe(4){obj2.test2[1].foo2},  'method called.:fooo'
+    assert_equals safe(4){obj2.test2[2]},  'method called.'
+    
+    # 共有ライブラリを上書き更新
+    shared_lib_body = <<-BODY
+    VALUE='val2'
+    class SharedLib
+      def foo
+        "fooo2"
+      end
+    end
+    def self.m
+      "method called2."
+    end
+BODY
+    assert_equals @registry.add_file( "shared_lib.rb", shared_lib_body, :shared_lib ), :success
+    
+    assert_equals safe(4){obj.test[0]},  'val2'
+    assert_equals safe(4){obj.test[1].foo},  'fooo2'
+    assert_equals safe(4){obj.test[2]},  'method called2.'
+    assert_equals safe(4){obj2.test[0]},  'val2'
+    assert_equals safe(4){obj2.test[1].foo},  'fooo2'
+    assert_equals safe(4){obj2.test[2]},  'method called2.'
+    # すでに作成済みのインスタンスの動作は変わらない
+    assert_equals safe(4){obj.s.foo},  'fooo'
+    assert_equals safe(4){obj2.s.foo},  'fooo'
+    # 再作成すると変わる
+    safe(4){
+      obj.init
+      obj2.init
+    }
+    assert_equals safe(4){obj.s.foo},  'fooo2'
+    assert_equals safe(4){obj2.s.foo},  'fooo2'
+    
+    # 依存モジュール内からの呼び出し結果も変わる
+    assert_equals safe(4){obj.test2[0]},  'val:method called.' # 定数は定義の段階で評価されているので変わらない
+    assert_equals safe(4){obj.test2[1].foo},  'fooo'
+    assert_equals safe(4){obj.test2[1].foo2},  'method called2.:fooo2'
+    assert_equals safe(4){obj.test2[2]},  'method called2.'
+    assert_equals safe(4){obj2.test2[0]},  'val:method called.'
+    assert_equals safe(4){obj2.test2[1].foo},  'fooo'
+    assert_equals safe(4){obj2.test2[1].foo2},  'method called2.:fooo2'
+    assert_equals safe(4){obj2.test2[2]},  'method called2.'
+  end
+
+end
+