vcs 0.1 → 0.2.148

data/ruby_ex/session/test/sub_server_test.rb

@@ -0,0 +1,188 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: polrop $
+# $Id: sub_server_test.rb 68 2004-12-07 10:51:28Z polrop $
+
+
+#
+# Client/Server basic test suite
+#
+if defined? TEST_MODE or $0 == __FILE__
+
+if RUBY_VERSION >= '1.9.0'
+
+
+require 'test/unit/ui/yaml/testrunner'
+require 'session/server'
+require 'session/administrable'
+require 'session/client'
+
+
+module Session
+
+  module Test
+
+    class SubServerTest < ::Test::Unit::TestCase
+
+      class MyServer < Session::Server
+        include Administrable
+
+        #       protected
+        #       def request_testme(sid, profile, test_obj)
+        #         ret = test_obj.run
+        #         return test_obj, ret
+        #       end
+
+      end
+
+      def setup
+        unless @srv_pid = fork # son
+          $stderr = File.open('/dev/null')
+          @opts = MyServer::OPTS
+          @opts[:usrdb] = MyServer::USRDB
+          @opts[:max_usr] = 2
+          @opts[:wipeout_delay] = 2
+          @srv = MyServer.new(@opts)
+          DRb.start_service('druby://localhost:42000', @srv)
+          DRb.thread.join
+          exit(0)
+        end
+      end
+
+      def teardown
+        Process.kill('TERM', @srv_pid)
+        Process.waitpid(@srv_pid)
+      end
+
+      #
+      # Tests
+      #
+      def test_detach
+        clt = nil
+        assert_nothing_raised do
+          clt = Client.new('druby://localhost:42000',
+                           'admin',
+                           'admin'.crypt(Const::SALT_KEY))
+        end
+        assert_not_nil(clt)
+        assert(! clt.online?, 'clt online 1')
+        assert_equal('data', clt.ping('data'))
+        assert(clt.logout)
+      end
+
+      def test_sub_server
+        clt = nil
+        assert_nothing_raised do
+          clt = Client.new('druby://localhost:42000',
+                           'admin',
+                           'admin'.crypt(Const::SALT_KEY), true)
+        end
+        assert_not_nil(clt)
+        uri = clt.start_sub_server(nil, MyServer) { true }
+        assert_not_equal(clt.local_uri, uri)
+        assert_not_equal(clt.uri, uri)
+        assert_equal(1, clt.sessions.size)
+        assert_not_nil(clt.sessions[clt.sid])
+        clients = clt.profile[:dispatcher].select_all
+        assert_equal(1, clients.size)
+        client = clients.first
+        assert_equal(uri, client.uri)
+        assert_equal(1, client.request(:sessions).size)
+        assert_not_equal(client.request(:sessions), clt.sessions)
+        assert_not_nil(client.request(:sessions)[client.sid])
+
+        sub_clt = nil
+        assert_nothing_raised do
+          sub_clt = Client.new(uri,
+                               'admin',
+                               'admin'.crypt(Const::SALT_KEY))
+        end
+        assert_not_nil(sub_clt)
+        assert_equal(uri, sub_clt.uri)
+        assert_equal(2, sub_clt.sessions.size)
+        assert_equal(2, client.request(:sessions).size)
+        assert_not_nil(client.request(:sessions)[sub_clt.sid])
+        assert_not_nil(sub_clt.sessions[client.sid])
+
+        assert(clt.logout)
+      end
+
+      def test_kill_and_sub_server
+        clt = nil
+        assert_nothing_raised do
+          clt = Client.new('druby://localhost:42000',
+                           'admin',
+                           'admin'.crypt(Const::SALT_KEY))
+        end
+        assert_not_nil(clt)
+        uri = clt.start_sub_server(nil) {}
+        pid = clt.pid
+        sub_pid = clt.request_at(uri, :pid)
+        assert_not_equal($$, pid)
+        assert_not_equal(sub_pid, pid)
+        assert_not_equal($$, sub_pid)
+        assert_equal(@srv_pid, pid)
+
+        assert_raises(Errno::EADDRINUSE) do
+          clt.start_sub_server(uri)
+        end
+        stat = clt.stop_sub_server(uri)
+        assert_equal(sub_pid, stat.pid)
+        assert_equal(0, stat.exitstatus)
+        assert_nil(stat.termsig)
+        assert_raises(Errno::ESRCH) do
+          Process.kill('KILL', sub_pid)
+        end
+
+        new_uri = clt.start_sub_server(uri)
+        assert_equal(uri, new_uri)
+        new_sub_pid = clt.request_at(uri, :pid)
+        assert_not_equal(sub_pid, new_sub_pid)
+
+        assert_equal(1, Process.kill('KILL', new_sub_pid))
+
+        stat = clt.stop_sub_server(uri)
+        assert_equal(new_sub_pid, stat.pid)
+        assert_equal(nil, stat.exitstatus)
+        assert_equal(9, stat.termsig)
+
+        clt.start_sub_server(nil) {}
+        clt.start_sub_server(nil) {}
+        assert(clt.logout)
+      end
+
+      def test_request_at
+        clt = nil
+        assert_nothing_raised do
+          clt = Client.new('druby://localhost:42000',
+                           'admin',
+                           'admin'.crypt(Const::SALT_KEY))
+        end
+        assert_not_nil(clt)
+        uri = clt.start_sub_server(nil) {}
+        sub_pid = clt.request_at(uri, :pid)
+        assert_raises(Session::RequestError) do
+          clt.request_at(uri, :toto)
+        end
+        assert_equal(sub_pid, clt.stop_sub_server(uri).pid)
+        assert(clt.logout)
+      end
+
+      #
+      # Utilities
+      #
+
+    end # class SubServerTest
+
+  end # module Test
+
+end # module Session
+
+else
+  STDERR.puts('need ruby1.9')
+end
+
+
+end

data/ruby_ex/shuffle.rb

@@ -0,0 +1,30 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: shuffle.rb 220 2005-05-09 12:28:26Z ertai $
+
+class Array
+
+  def shuffle! ( generator=nil )
+    n = size
+    n.times do |i|
+      swap!(i, n.choose(generator))
+    end
+    self
+  end
+
+  def shuffle ( generator=nil )
+    dup.shuffle!(generator)
+  end
+
+  def swap! ( i, j )
+    self[i], self[j] = self[j], self[i]
+  end
+
+  def swap ( i, j )
+    dup.swap!(i, j)
+  end
+
+end # class Array
+
+

data/ruby_ex/spring.rb

@@ -0,0 +1,136 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: spring.rb 53 2004-12-02 22:24:03Z ertai $
+
+
+require 'thread'
+require 'observer'
+
+
+class Spring
+
+  include Observable
+
+  GET = :get
+  SIGN_UP = :sign_up
+
+  def initialize(*items)
+    @q = *items
+    @mutex = Mutex.new
+    @collecter = []
+  end
+
+  def get(collecter_id)
+    @mutex.synchronize do
+      if @collecter.include?(collecter_id)
+        item = @q.shift
+        changed(true)
+        notify_observers(self.object_id, GET, collecter_id)
+        return item
+      end
+    end
+  end
+
+  def size
+    @mutex.synchronize { return @q.size }
+  end
+  alias length size
+
+  def empty?
+    @mutex.synchronize { return @q.empty? }
+  end
+
+  def sign_up(collecter_id)
+    @mutex.synchronize do
+      unless @collecter.include?(collecter_id)
+        @collecter << collecter_id
+        changed(true)
+        notify_observers(self.object_id, SIGN_UP, collecter_id)
+        return nil
+      end
+    end
+  end
+
+  def num_collecter
+    @mutex.synchronize { @collecter.length }
+  end
+
+end # class Spring
+
+
+#
+# Unit test suite
+#
+if defined? TEST_MODE or $0 == __FILE__
+
+
+require 'test/unit/ui/yaml/testrunner'
+
+
+class SpringTest < Test::Unit::TestCase
+
+  class Obs
+
+    attr_reader :updated, :msg_seq
+
+    def initialize(s_id, msg_seq, t_id, spring_test_instance)
+      @s_id = s_id
+      @msg_seq = msg_seq
+      @t_id = t_id
+      @updated = false
+      @spring_test_instance = spring_test_instance
+    end
+
+    def update(s_id, msg, t_id)
+      @updated = true
+      @spring_test_instance.test_update(@s_id,
+                                        @msg_seq,
+                                        @t_id,
+                                        s_id,
+                                        msg,
+                                        t_id)
+    end
+
+  end
+
+  #
+  # tests
+  #
+  def test_simple
+    s = Spring.new('hello', 'world')
+    q = Queue.new
+    thread = Thread.new do
+      s.add_observer(q.get)
+      assert_equal(2, s.size)
+      assert_equal(2, s.length)
+      assert_nil(s.get(thread.object_id))
+      s.sign_up(thread.object_id)
+      assert_equal(1, s.num_collecter)
+      assert_equal('hello', s.get(thread.object_id))
+      assert_equal('world', s.get(thread.object_id))
+    end
+    o = Obs.new(s.object_id,
+                [Spring::SIGN_UP, Spring::GET, Spring::GET],
+                thread.object_id,
+                self)
+    q.put(o)
+    thread.join
+    assert_equal(true, o.updated, 'not visited at all')
+    assert_equal([], o.msg_seq)
+  end
+
+  # tools
+
+  def test_update(r_s_id, r_msg, r_t_id, s_id, msg, t_id)
+    assert_equal(r_t_id, t_id, 'bad collecter id')
+    assert_equal(r_s_id, s_id, 'bad spring id')
+    assert_equal(r_msg.shift, msg, 'bad message')
+  end
+
+end # class SpringTest
+
+
+end

data/ruby_ex/spring_set.rb

@@ -0,0 +1,137 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: spring_set.rb 186 2005-04-03 00:07:45Z ertai $
+
+
+require 'spring'
+
+
+class SpringSet
+
+  def initialize(*springs)
+    @springs = {}
+    springs.flatten.each do |s|
+      @springs[s.object_id] = {
+        :spring => s,
+        :num_collecter => s.num_collecter,
+        :size => s.size
+      }
+      s.add_observer(self)
+    end
+  end
+
+  def update(spring_id, msg, collecter_id)
+    case msg
+    when Spring::SIGN_UP
+      @springs[spring_id][:num_collecter] += 1
+    when Spring::GET
+      @springs[spring_id][:size] -= 1 if @springs[spring_id][:size] > 0
+    end
+  end
+
+  def get_most_free
+    calc_stat(:num_collecter) { |a, b| a < b }
+  end
+
+  def get_most_busy
+    calc_stat(:num_collecter) { |a, b| a > b }
+  end
+
+  def get_most_full
+    calc_stat(:size) { |a, b| a > b }
+  end
+
+  def get_most_empty
+    calc_stat(:size) { |a, b| a < b }
+  end
+
+  def size
+    @springs.size
+  end
+
+  alias length size
+
+  def num_collecter(spring_id)
+    @springs[spring_id][:num_collecter]
+  end
+
+  def spring_size(spring_id)
+    @springs[spring_id][:size]
+  end
+
+  alias spring_length spring_size
+
+  def each
+    @springs.values.each { |v| yield(v[:spring]) }
+  end
+
+  protected
+  def calc_stat(field, &cmp)
+    spring = nil
+    @springs.values.each do |s|
+      if spring.nil? or cmp[s[field], spring[field]]
+        spring = s
+      end
+    end
+    return spring[:spring]
+  end
+
+end # class SpringSet
+
+
+#
+# Unit test suite
+#
+if defined? TEST_MODE or $0 == __FILE__
+
+
+require 'test/unit/ui/yaml/testrunner'
+
+
+class SpringSetTest < Test::Unit::TestCase
+
+  #
+  # tests
+  #
+  def test_simple
+    s1 = Spring.new('hello', 'world')
+    s2 = Spring.new('bonjour', 'tout', 'monde')
+    s = SpringSet.new(s1, s2)
+    assert_equal(s1.num_collecter, s.num_collecter(s1.object_id))
+    assert_equal(s2.num_collecter, s.num_collecter(s2.object_id))
+    assert_equal(s1.size, s.spring_size(s1.object_id))
+    assert_equal(s2.length, s.spring_length(s2.object_id))
+    thread = Thread.new do
+      sleep(0.0001)
+      assert_equal(2, s.size)
+      assert_equal(2, s.length)
+      assert_equal(s2, s.get_most_full)
+      assert_equal(s1, s.get_most_empty)
+      s1.sign_up(thread.object_id)
+      assert_equal(s1.num_collecter, s.num_collecter(s1.object_id))
+      assert_equal(s1, s.get_most_busy)
+      assert_equal(s2, s.get_most_free)
+      s2.sign_up(thread.object_id)
+      assert_equal(s2.num_collecter, s.num_collecter(s2.object_id), 'bad num collecter')
+      assert_equal('bonjour', s2.get(thread.object_id))
+      assert_equal(s2.size, s.spring_size(s2.object_id), 'bad size')
+      assert_equal('tout', s2.get(thread.object_id))
+      assert_equal(s2.size, s.spring_size(s2.object_id))
+      assert_equal('monde', s2.get(thread.object_id))
+      assert_equal(s2.size, s.spring_size(s2.object_id))
+      assert_equal(nil, s2.get(thread.object_id))
+      assert_equal(s2.size, s.spring_size(s2.object_id))
+      assert_equal(0, s.spring_size(s2.object_id))
+      assert_equal(s1, s.get_most_full)
+      assert_equal(s2, s.get_most_empty)
+    end
+    thread.join
+  end
+
+end # class SpringSetTest
+
+
+end