torquebox-console 0.2.3 → 0.2.4

data/vendor/bundle/ruby/1.8/gems/stomp-1.2.8/test/test_connection1p.rb

@@ -0,0 +1,396 @@
+# -*- encoding: utf-8 -*-
+
+if Kernel.respond_to?(:require_relative)
+  require_relative("test_helper")
+else
+  $:.unshift(File.dirname(__FILE__))
+  require 'test_helper'
+end
+
+=begin
+
+  Main class for testing Stomp::Connection instances, protocol levels 1.1+.
+
+=end
+class TestConnection1P < Test::Unit::TestCase
+  include TestBase
+  
+  def setup
+    @conn = get_connection()
+  end
+
+  def teardown
+    @conn.disconnect if @conn.open? # allow tests to disconnect
+  end
+
+  # Test basic connection open.
+  def test_conn_1p_0000
+    assert @conn.open?
+  end
+
+  # Test missing connect headers.
+  def test_conn_1p_0010
+    @conn.disconnect
+    #
+    cha = {:host => "localhost"}
+    assert_raise Stomp::Error::ProtocolErrorConnect do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+    end    
+    #
+    chb = {"accept-version" => "1.0"}
+    assert_raise Stomp::Error::ProtocolErrorConnect do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, chb)
+    end    
+  end
+
+  # Test requesting only a 1.0 connection.
+  def test_conn_1p_0020
+    @conn.disconnect
+    #
+    cha = {:host => "localhost", "accept-version" => "1.0"}
+    cha[:host] = "/" if ENV['STOMP_RABBIT']
+    conn = nil
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+      conn.disconnect
+    end
+    assert_equal conn.protocol, Stomp::SPL_10
+  end
+
+  # Test requesting only a 1.1+ connection.
+  def test_conn_1p_0030
+    @conn.disconnect
+    #
+    cha = get_conn_headers()
+    conn = nil
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+      conn.disconnect
+    end
+    assert conn.protocol >= Stomp::SPL_11
+  end
+
+  # Test basic request for no heartbeats.
+  def test_conn_1p_0040
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "0,0" # No heartbeats
+    conn = nil
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+      conn.disconnect
+    end
+    assert conn.protocol >= Stomp::SPL_11
+  end
+
+  # Test malformed heartbeat header.
+  def test_conn_1p_0050
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "10,10,20" # Bad header Heartbeats
+    conn = nil
+    assert_raise Stomp::Error::InvalidHeartBeatHeaderError do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+    end
+  end
+
+  # Test malformed heartbeat header.
+  def test_conn_11h_0060
+    @conn.disconnect
+    #
+    cha = get_conn_headers()
+    cha["heart-beat"] = "a,10" # Bad header Heartbeats
+    conn = nil
+    assert_raise Stomp::Error::InvalidHeartBeatHeaderError do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+    end
+  end
+
+  # Test a valid heartbeat header.
+  def test_conn_1p_0070
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "500,1000" # Valid heart beat headers
+    conn = nil
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+      conn.disconnect
+    end
+    assert conn.hbsend_interval > 0
+    assert conn.hbrecv_interval > 0
+  end
+
+  # Test only sending heartbeats.
+  def test_conn_1p_0080
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "10000,0" # Valid heart beat headers, send only
+    conn = nil
+    logger = Tlogger.new
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+      conn.set_logger(logger)
+      sleep 65
+      conn.set_logger(nil)
+      conn.disconnect
+    end
+    hb_asserts_send(conn)
+  end if ENV['STOMP_HB11LONG']
+
+  # Test only receiving heartbeats.
+  def test_conn_1p_0090
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "0,6000" # Valid heart beat headers, receive only
+    conn = nil
+    logger = Tlogger.new
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+#      m = conn.receive # This will hang forever .....
+      conn.set_logger(logger)
+      sleep 65
+      conn.set_logger(nil)
+      conn.disconnect
+    end
+    hb_asserts_recv(conn)
+  end if ENV['STOMP_HB11LONG']
+
+  # Test sending and receiving heartbeats.
+  def test_conn_1p_0100
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "5000,10000" # Valid heart beat headers, send and receive
+    conn = nil
+    logger = Tlogger.new
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+#      m = conn.receive # This will hang forever .....
+      conn.set_logger(logger)
+      sleep 65
+      conn.set_logger(nil)
+      conn.disconnect
+    end
+    hb_asserts_both(conn)
+  end if ENV['STOMP_HB11LONG']
+
+  # Test valid UTF8 data.
+  def test_conn_1p_0110
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "0,0" # No heartbeats
+    conn = nil
+    conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+    good_data = [
+      "\x41\xc3\xb1\x42",
+      "\xc2\x80", # 2 byte characters
+      "\xc2\xbf",
+      "\xdf\x80",
+      "\xdf\xbf",
+      "\xe0\xa0\x80", # 3 byte characters
+      "\xe0\xbf\x80",
+      "\xe0\xa0\xbf",
+      "\xe0\xbf\xbf",
+      "\xf1\x80\x80\x80", # 4 byte characters
+      "\xf1\xbf\xbf\xbf",
+      "\xf2\x80\x80\x80",
+      "\xf2\xbf\xbf\xbf",
+      "\xf3\x80\x80\x80",
+      "\xf3\xbf\xbf\xbf",
+    ]
+    good_data.each do |string|
+      assert conn.valid_utf8?(string), "good unicode specs 01: #{string}"
+    end
+    conn.disconnect
+  end
+
+  # Test invalid UTF8 data.
+  def test_conn_1p_0120
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "0,0" # No heartbeats
+    conn = nil
+    conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+    bad_data = [
+      "\x41\xc2\xc3\xb1\x42",
+      "\xed\xa0\x80", # UTF-16 surrogate halves
+      "\xed\xad\xbf",
+      "\xed\xae\x80",
+      "\xed\xaf\xbf",
+      "\xed\xb0\x80",
+      "\xed\xbe\x80",
+      "\xed\xbf\xbf",
+      "\xc0", # Single bytes
+      "\xc1",
+      "\xf5","\xf6","\xf7","\xf8","\xf9","\xfa","\xfb","\xfc",
+      "\xfd","\xfe","\xff",
+      "\xc0\x80", # Not shortest representation
+      "\xc1\x80",
+      "\xc0\x30",
+      "\xc1\x30",
+      "\xe0\x80\x80",
+      "\xf0\x80\x80\x80",
+    ]
+    bad_data.each do |string|
+      assert !conn.valid_utf8?(string), "bad unicode specs 01: #{string}"
+    end
+    conn.disconnect
+  end
+
+  # Repeated headers test. Currently:
+  # - Apollo emits repeated headers for a 1.1 connection only
+  # - RabbitMQ does not emit repeated headers under any circumstances
+  # - AMQ 5.6 does not emit repeated headers under any circumstances
+  # Pure luck that this runs against AMQ at present.
+  def test_conn_1p_0124
+    dest = make_destination
+    msg = "payload: #{Time.now.to_f}"
+    shdrs = { "key1" => "val1", "key2" => "val2",
+      "key3" => ["kv3", "kv2", "kv1"] }
+    assert_nothing_raised {
+      @conn.publish dest, msg, shdrs
+    }
+    #
+    sid = @conn.uuid()
+    @conn.subscribe dest, :id => sid
+    #
+    received = @conn.receive
+    assert_equal msg, received.body
+    if @conn.protocol != Stomp::SPL_10
+      assert_equal shdrs["key3"], received.headers["key3"] unless ENV['STOMP_RABBIT'] || ENV['STOMP_AMQ11']
+    else
+      assert_equal "kv3", received.headers["key3"]
+    end
+    #
+    @conn.unsubscribe dest, :id => sid
+  end
+
+  # Test frozen headers.
+  def test_conn_1p_0127
+    dest = make_destination
+    sid = @conn.uuid()
+    sid.freeze
+    assert_nothing_raised {
+      @conn.subscribe dest, :id => sid
+    }
+  end
+
+  # Test heartbeats with send and receive.
+  def test_conn_1p_0130
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "10000,6000" # Valid heart beat headers, send and receive
+    conn = nil
+    logger = Tlogger.new
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+#      m = conn.receive # This will hang forever .....
+      conn.set_logger(logger)
+      sleep 65
+      conn.set_logger(nil)
+      conn.disconnect
+    end
+    hb_asserts_both(conn)
+  end if ENV['STOMP_HB11LONG']
+
+  # Test heartbeats with send and receive.
+  def test_conn_1p_0135
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "10000,1000" # Valid heart beat headers, send and receive
+    conn = nil
+    logger = Tlogger.new
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+#      m = conn.receive # This will hang forever .....
+      conn.set_logger(logger)
+      sleep 65
+      conn.set_logger(nil)
+      conn.disconnect
+    end
+    hb_asserts_both(conn)
+  end if ENV['STOMP_HB11LONG']
+
+  # Test heartbeats with send and receive.
+  def test_conn_1p_0140
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    cha["heart-beat"] = "1000,10000" # Valid heart beat headers, send and receive
+    conn = nil
+    logger = Tlogger.new
+    assert_nothing_raised do
+      conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+#      m = conn.receive # This will hang forever .....
+      conn.set_logger(logger)
+      sleep 65
+      conn.set_logger(nil)
+      conn.disconnect
+    end
+    hb_asserts_both(conn)
+  end if ENV['STOMP_HB11LONG']
+
+  # Test very encoding / decoding of headers
+  def test_conn_1p_0200
+    @conn.disconnect
+    #
+    cha = get_conn_headers()    
+    conn = Stomp::Connection.open(user, passcode, host, port, false, 5, cha)
+    msg = "payload: #{Time.now.to_f}"
+    dest = make_destination
+    shdrs = { "ab:cd" => "ef:gh", "a\nb" => "c\nd", "x\\y" => "z\\s" }
+    if conn.protocol >= Stomp::SPL_12
+      shdrs["bb\rcc"] = "dd\ree"
+    end
+    assert_nothing_raised {
+      conn.publish dest, msg, shdrs
+    }
+    #
+    sid = conn.uuid()
+    conn.subscribe dest, :id => sid
+    #
+    received = conn.receive
+    assert_equal msg, received.body
+    #
+    shdrs.each_pair {|k,v|
+      assert received.headers.has_key?(k), "Key not found: #{k}"
+      assert received.headers.has_value?(v), "Value not found: #{v}"
+      assert received.headers[k] == v, "Mismatch: #{k},#{v}"
+    }
+    conn.disconnect
+  end unless ENV['STOMP_RABBIT']
+
+private
+
+  def hb_asserts_both(conn)
+    assert conn.hbsend_interval > 0
+    assert conn.hbrecv_interval > 0
+    assert conn.hbsend_count > 0
+    assert conn.hbrecv_count > 0
+  end
+
+  def hb_asserts_send(conn)
+    assert conn.hbsend_interval > 0
+    assert conn.hbrecv_interval == 0
+    assert conn.hbsend_count > 0
+    assert conn.hbrecv_count == 0
+  end
+
+  def hb_asserts_recv(conn)
+    assert conn.hbsend_interval == 0
+    assert conn.hbrecv_interval > 0
+    assert conn.hbsend_count == 0
+    assert conn.hbrecv_count > 0
+  end
+
+end if ENV['STOMP_TEST11p']
+

data/vendor/bundle/ruby/1.8/gems/stomp-1.2.8/test/test_helper.rb

@@ -0,0 +1,168 @@
+# -*- encoding: utf-8 -*-
+
+require 'test/unit'
+require 'timeout'
+
+if Kernel.respond_to?(:require_relative)
+  require_relative("../lib/stomp")
+  require_relative("tlogger")
+else
+  $:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
+  require 'stomp'
+  require 'tlogger'
+end
+
+begin
+  dummy = RUBY_ENGINE
+rescue NameError => ne
+  RUBY_ENGINE = "unknown"
+end
+
+=begin
+
+  Test helper methods.
+
+=end
+module TestBase
+
+  # Get user
+  def user
+    ENV['STOMP_USER'] || "guest"
+  end
+
+  # Gete passcode
+  def passcode
+    ENV['STOMP_PASSCODE'] || "guest"
+  end
+
+  # Get host
+  def host
+    ENV['STOMP_HOST'] || "localhost"
+  end
+
+  # Get port
+  def port
+    (ENV['STOMP_PORT'] || 61613).to_i
+  end
+
+  # Get SSL port
+  def ssl_port
+    (ENV['STOMP_SSLPORT'] || 61612).to_i
+  end
+
+  # Helper for minitest on 1.9
+  def caller_method_name
+    parse_caller(caller(2).first).last
+  end
+
+  # Helper for minitest on 1.9
+  def parse_caller(at)
+    if /^(.+?):(\d+)(?::in `(.*)')?/ =~ at
+      file = Regexp.last_match[1]
+      line = Regexp.last_match[2].to_i
+      method = Regexp.last_match[3]
+      method.gsub!(" ","_")
+      [file, line, method]
+    end
+  end
+
+  # Get a Stomp Connection.
+  def get_connection()
+    ch = get_conn_headers()
+    hash = { :hosts => [ 
+      {:login => user, :passcode => passcode, :host => host, :port => port, :ssl => nil},
+      ],
+      :reliable => false,
+      :connect_headers => ch,
+      :stompconn => get_stomp_conn(),
+      :usecrlf => get_crlf(),
+    }
+    conn = Stomp::Connection.open(hash)
+    conn
+  end
+
+  # Get a Stomp SSL Connection.
+  def get_ssl_connection()
+    ch = get_conn_headers()
+    ssl_params = Stomp::SSLParams.new # S/B safe for all Ruby versions tested
+    hash = { :hosts => [ 
+      {:login => user, :passcode => passcode, :host => host, :port => ssl_port, :ssl => ssl_params},
+      ],
+      :connect_headers => ch,
+      :stompconn => get_stomp_conn(),
+      :usecrlf => get_crlf(),
+    }
+    conn = Stomp::Connection.new(hash)
+    conn
+  end
+
+  # Get a Stomp Client.
+  def get_client()
+    hash = { :hosts => [ 
+          {:login => user, :passcode => passcode, :host => host, :port => port},
+          ],
+          :connect_headers => get_conn_headers(),
+          :stompconn => get_stomp_conn(),
+          :usecrlf => get_crlf(),
+        }
+
+    client = Stomp::Client.new(hash)
+    client
+  end
+
+  # Get a connection headers hash.
+  def get_conn_headers()
+    ch = {}
+    if ENV['STOMP_TEST11p']
+      #
+      raise "Invalid 1.1 plus test protocol" if ENV['STOMP_TEST11p'] == Stomp::SPL_10
+      #
+      if Stomp::SUPPORTED.index(ENV['STOMP_TEST11p'])
+        ch['accept-version'] = ENV['STOMP_TEST11p']
+      else
+        ch['accept-version'] = Stomp::SPL_11 # Just use 1.1
+      end
+      #
+      ch['host'] = ENV['STOMP_RABBIT'] ? "/" : host
+    end
+    ch
+  end
+
+  # Determine if tests should use STOMP instead of CONNECT
+  def get_stomp_conn()
+    usc = false
+    usc = true if ENV['STOMP_TEST11p'] && Stomp::SUPPORTED.index(ENV['STOMP_TEST11p']) && ENV['STOMP_TEST11p'] >= Stomp::SPL_11 && ENV['STOMP_CONN']
+    usc
+  end
+
+  # Determine if tests should \r\n as line ends
+  def get_crlf()
+    ucr = false
+    ucr = true if ENV['STOMP_TEST11p'] && Stomp::SUPPORTED.index(ENV['STOMP_TEST11p']) && ENV['STOMP_TEST11p'] >= Stomp::SPL_12 && ENV['STOMP_CRLF']
+    ucr
+  end
+
+  # Subscribe to a destination.
+  def conn_subscribe(dest, headers = {})
+    if @conn.protocol >= Stomp::SPL_11
+      headers[:id] = @conn.uuid() unless headers[:id]
+    end
+    @conn.subscribe dest, headers
+  end
+
+  # Get a dynamic destination name.
+  def make_destination
+    name = caller_method_name unless name
+    qname = ENV['STOMP_DOTQUEUE'] ? "/queue/test.ruby.stomp." + name : "/queue/test/ruby/stomp/" + name
+  end
+
+  #
+  def checkEmsg(cc)
+    m = cc.poll
+    if m
+      assert m.command != Stomp::CMD_ERROR
+    end
+  end
+
+end
+