diff --git a/lib/uri.rb b/lib/uri.rb index 394c156ac5..59a7c4ad28 100644 --- a/lib/uri.rb +++ b/lib/uri.rb @@ -101,3 +101,4 @@ require_relative 'uri/ldap' require_relative 'uri/ldaps' require_relative 'uri/mailto' require_relative 'uri/ws' +require_relative 'uri/wss' diff --git a/test/uri/test_common.rb b/test/uri/test_common.rb index 5e30cda41d..0fa7e8ac70 100644 --- a/test/uri/test_common.rb +++ b/test/uri/test_common.rb @@ -42,17 +42,17 @@ class TestCommon < Test::Unit::TestCase end def test_register_scheme - assert_equal(["FILE", "FTP", "HTTP", "HTTPS", "LDAP", "LDAPS", "MAILTO", "WS"].sort, URI.scheme_list.keys.sort) + assert_equal(["FILE", "FTP", "HTTP", "HTTPS", "LDAP", "LDAPS", "MAILTO", "WS", "WSS"].sort, URI.scheme_list.keys.sort) foobar = Class.new(URI::Generic) URI.register_scheme 'FOOBAR', foobar begin - assert_equal(["FILE", "FTP", "HTTP", "HTTPS", "LDAP", "LDAPS", "MAILTO", "WS", "FOOBAR"].sort, URI.scheme_list.keys.sort) + assert_equal(["FILE", "FTP", "HTTP", "HTTPS", "LDAP", "LDAPS", "MAILTO", "WS", "WSS", "FOOBAR"].sort, URI.scheme_list.keys.sort) ensure URI.const_get(:Schemes).send(:remove_const, :FOOBAR) end - assert_equal(["FILE", "FTP", "HTTP", "HTTPS", "LDAP", "LDAPS", "MAILTO", "WS"].sort, URI.scheme_list.keys.sort) + assert_equal(["FILE", "FTP", "HTTP", "HTTPS", "LDAP", "LDAPS", "MAILTO", "WS", "WSS"].sort, URI.scheme_list.keys.sort) end def test_regexp diff --git a/test/uri/test_wss.rb b/test/uri/test_wss.rb new file mode 100644 index 0000000000..fd49d9e126 --- /dev/null +++ b/test/uri/test_wss.rb @@ -0,0 +1,71 @@ +# frozen_string_literal: false +require 'test/unit' +require 'uri/http' +require 'uri/wss' + +module URI + + +class TestWSS < Test::Unit::TestCase + def setup + end + + def teardown + end + + def uri_to_ary(uri) + uri.class.component.collect {|c| uri.send(c)} + end + + def test_build + u = URI::WSS.build(host: 'www.example.com', path: '/foo/bar') + assert_kind_of(URI::WSS, u) + end + + def test_parse + u = URI.parse('wss://a') + assert_kind_of(URI::WSS, u) + assert_equal(['wss', + nil, 'a', URI::HTTPS.default_port, + '', nil], uri_to_ary(u)) + end + + def test_normalize + host = 'aBcD' + u1 = URI.parse('wss://' + host + '/eFg?HiJ') + u2 = URI.parse('wss://' + host.downcase + '/eFg?HiJ') + assert(u1.normalize.host == 'abcd') + assert(u1.normalize.path == u1.path) + assert(u1.normalize == u2.normalize) + assert(!u1.normalize.host.equal?(u1.host)) + assert( u2.normalize.host.equal?(u2.host)) + + assert_equal('wss://abc/', URI.parse('wss://abc').normalize.to_s) + end + + def test_equal + assert(URI.parse('wss://abc') == URI.parse('wss://ABC')) + assert(URI.parse('wss://abc/def') == URI.parse('wss://ABC/def')) + assert(URI.parse('wss://abc/def') != URI.parse('wss://ABC/DEF')) + end + + def test_request_uri + assert_equal('/', URI.parse('wss://a.b.c/').request_uri) + assert_equal('/?abc=def', URI.parse('wss://a.b.c/?abc=def').request_uri) + assert_equal('/', URI.parse('wss://a.b.c').request_uri) + assert_equal('/?abc=def', URI.parse('wss://a.b.c?abc=def').request_uri) + assert_equal(nil, URI.parse('wss:foo').request_uri) + end + + def test_select + assert_equal(['wss', 'a.b.c', 443], URI.parse('wss://a.b.c/').select(:scheme, :host, :port)) + u = URI.parse('wss://a.b.c/') + assert_equal(uri_to_ary(u), u.select(*u.component)) + assert_raise(ArgumentError) do + u.select(:scheme, :host, :not_exist, :port) + end + end +end + + +end