#!/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