289 lines
11 KiB
Python
289 lines
11 KiB
Python
# -*- encoding: utf8 -*-
|
|
from __future__ import unicode_literals
|
|
from sleekxmpp.test import *
|
|
from sleekxmpp import JID, InvalidJID
|
|
from sleekxmpp.jid import nodeprep
|
|
|
|
|
|
class TestJIDClass(SleekTest):
|
|
|
|
"""Verify that the JID class can parse and manipulate JIDs."""
|
|
|
|
def testJIDFromFull(self):
|
|
"""Test using JID of the form 'user@server/resource/with/slashes'."""
|
|
self.check_jid(JID('user@someserver/some/resource'),
|
|
'user',
|
|
'someserver',
|
|
'some/resource',
|
|
'user@someserver',
|
|
'user@someserver/some/resource',
|
|
'user@someserver/some/resource')
|
|
|
|
def testJIDchange(self):
|
|
"""Test changing JID of the form 'user@server/resource/with/slashes'"""
|
|
j = JID('user1@someserver1/some1/resource1')
|
|
j.user = 'user'
|
|
j.domain = 'someserver'
|
|
j.resource = 'some/resource'
|
|
self.check_jid(j,
|
|
'user',
|
|
'someserver',
|
|
'some/resource',
|
|
'user@someserver',
|
|
'user@someserver/some/resource',
|
|
'user@someserver/some/resource')
|
|
|
|
def testJIDaliases(self):
|
|
"""Test changing JID using aliases for domain."""
|
|
j = JID('user@someserver/resource')
|
|
j.server = 'anotherserver'
|
|
self.check_jid(j, domain='anotherserver')
|
|
j.host = 'yetanother'
|
|
self.check_jid(j, domain='yetanother')
|
|
|
|
def testJIDSetFullWithUser(self):
|
|
"""Test setting the full JID with a user portion."""
|
|
j = JID('user@domain/resource')
|
|
j.full = 'otheruser@otherdomain/otherresource'
|
|
self.check_jid(j,
|
|
'otheruser',
|
|
'otherdomain',
|
|
'otherresource',
|
|
'otheruser@otherdomain',
|
|
'otheruser@otherdomain/otherresource',
|
|
'otheruser@otherdomain/otherresource')
|
|
|
|
def testJIDFullNoUserWithResource(self):
|
|
"""
|
|
Test setting the full JID without a user
|
|
portion and with a resource.
|
|
"""
|
|
j = JID('user@domain/resource')
|
|
j.full = 'otherdomain/otherresource'
|
|
self.check_jid(j,
|
|
'',
|
|
'otherdomain',
|
|
'otherresource',
|
|
'otherdomain',
|
|
'otherdomain/otherresource',
|
|
'otherdomain/otherresource')
|
|
|
|
def testJIDFullNoUserNoResource(self):
|
|
"""
|
|
Test setting the full JID without a user
|
|
portion and without a resource.
|
|
"""
|
|
j = JID('user@domain/resource')
|
|
j.full = 'otherdomain'
|
|
self.check_jid(j,
|
|
'',
|
|
'otherdomain',
|
|
'',
|
|
'otherdomain',
|
|
'otherdomain',
|
|
'otherdomain')
|
|
|
|
def testJIDBareUser(self):
|
|
"""Test setting the bare JID with a user."""
|
|
j = JID('user@domain/resource')
|
|
j.bare = 'otheruser@otherdomain'
|
|
self.check_jid(j,
|
|
'otheruser',
|
|
'otherdomain',
|
|
'resource',
|
|
'otheruser@otherdomain',
|
|
'otheruser@otherdomain/resource',
|
|
'otheruser@otherdomain/resource')
|
|
|
|
def testJIDBareNoUser(self):
|
|
"""Test setting the bare JID without a user."""
|
|
j = JID('user@domain/resource')
|
|
j.bare = 'otherdomain'
|
|
self.check_jid(j,
|
|
'',
|
|
'otherdomain',
|
|
'resource',
|
|
'otherdomain',
|
|
'otherdomain/resource',
|
|
'otherdomain/resource')
|
|
|
|
def testJIDNoResource(self):
|
|
"""Test using JID of the form 'user@domain'."""
|
|
self.check_jid(JID('user@someserver'),
|
|
'user',
|
|
'someserver',
|
|
'',
|
|
'user@someserver',
|
|
'user@someserver',
|
|
'user@someserver')
|
|
|
|
def testJIDNoUser(self):
|
|
"""Test JID of the form 'component.domain.tld'."""
|
|
self.check_jid(JID('component.someserver'),
|
|
'',
|
|
'component.someserver',
|
|
'',
|
|
'component.someserver',
|
|
'component.someserver',
|
|
'component.someserver')
|
|
|
|
def testJIDEquality(self):
|
|
"""Test that JIDs with the same content are equal."""
|
|
jid1 = JID('user@domain/resource')
|
|
jid2 = JID('user@domain/resource')
|
|
self.assertTrue(jid1 == jid2, "Same JIDs are not considered equal")
|
|
self.assertFalse(jid1 != jid2, "Same JIDs are considered not equal")
|
|
|
|
def testJIDInequality(self):
|
|
jid1 = JID('user@domain/resource')
|
|
jid2 = JID('otheruser@domain/resource')
|
|
self.assertFalse(jid1 == jid2, "Same JIDs are not considered equal")
|
|
self.assertTrue(jid1 != jid2, "Same JIDs are considered not equal")
|
|
|
|
def testZeroLengthDomain(self):
|
|
self.assertRaises(InvalidJID, JID, domain='')
|
|
self.assertRaises(InvalidJID, JID, 'user@/resource')
|
|
|
|
def testZeroLengthLocalPart(self):
|
|
self.assertRaises(InvalidJID, JID, local='', domain='test.com')
|
|
self.assertRaises(InvalidJID, JID, '@/test.com')
|
|
|
|
def testZeroLengthResource(self):
|
|
self.assertRaises(InvalidJID, JID, domain='test.com', resource='')
|
|
self.assertRaises(InvalidJID, JID, 'test.com/')
|
|
|
|
def test1023LengthDomain(self):
|
|
domain = ('a.' * 509) + 'a.com'
|
|
jid1 = JID(domain=domain)
|
|
jid2 = JID('user@%s/resource' % domain)
|
|
|
|
def test1023LengthLocalPart(self):
|
|
local = 'a' * 1023
|
|
jid1 = JID(local=local, domain='test.com')
|
|
jid2 = JID('%s@test.com' % local)
|
|
|
|
def test1023LengthResource(self):
|
|
resource = 'r' * 1023
|
|
jid1 = JID(domain='test.com', resource=resource)
|
|
jid2 = JID('test.com/%s' % resource)
|
|
|
|
def test1024LengthDomain(self):
|
|
domain = ('a.' * 509) + 'aa.com'
|
|
self.assertRaises(InvalidJID, JID, domain=domain)
|
|
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
|
|
|
def test1024LengthLocalPart(self):
|
|
local = 'a' * 1024
|
|
self.assertRaises(InvalidJID, JID, local=local, domain='test.com')
|
|
self.assertRaises(InvalidJID, JID, '%s@/test.com' % local)
|
|
|
|
def test1024LengthResource(self):
|
|
resource = 'r' * 1024
|
|
self.assertRaises(InvalidJID, JID, domain='test.com', resource=resource)
|
|
self.assertRaises(InvalidJID, JID, 'test.com/%s' % resource)
|
|
|
|
def testTooLongDomainLabel(self):
|
|
domain = ('a' * 64) + '.com'
|
|
self.assertRaises(InvalidJID, JID, domain=domain)
|
|
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
|
|
|
def testDomainEmptyLabel(self):
|
|
domain = 'aaa..bbb.com'
|
|
self.assertRaises(InvalidJID, JID, domain=domain)
|
|
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
|
|
|
def testDomainIPv4(self):
|
|
domain = '127.0.0.1'
|
|
jid1 = JID(domain=domain)
|
|
jid2 = JID('user@%s/resource' % domain)
|
|
|
|
def testDomainIPv6(self):
|
|
domain = '[::1]'
|
|
jid1 = JID(domain=domain)
|
|
jid2 = JID('user@%s/resource' % domain)
|
|
|
|
def testDomainInvalidIPv6NoBrackets(self):
|
|
domain = '::1'
|
|
jid1 = JID(domain=domain)
|
|
jid2 = JID('user@%s/resource' % domain)
|
|
|
|
self.assertEqual(jid1.domain, '[::1]')
|
|
self.assertEqual(jid2.domain, '[::1]')
|
|
|
|
def testDomainInvalidIPv6MissingBracket(self):
|
|
domain = '[::1'
|
|
jid1 = JID(domain=domain)
|
|
jid2 = JID('user@%s/resource' % domain)
|
|
|
|
self.assertEqual(jid1.domain, '[::1]')
|
|
self.assertEqual(jid2.domain, '[::1]')
|
|
|
|
def testDomainWithPort(self):
|
|
domain = 'example.com:5555'
|
|
self.assertRaises(InvalidJID, JID, domain=domain)
|
|
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
|
|
|
def testDomainWithTrailingDot(self):
|
|
domain = 'example.com.'
|
|
jid1 = JID(domain=domain)
|
|
jid2 = JID('user@%s/resource' % domain)
|
|
|
|
self.assertEqual(jid1.domain, 'example.com')
|
|
self.assertEqual(jid2.domain, 'example.com')
|
|
|
|
def testDomainWithDashes(self):
|
|
domain = 'example.com-'
|
|
self.assertRaises(InvalidJID, JID, domain=domain)
|
|
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
|
|
|
domain = '-example.com'
|
|
self.assertRaises(InvalidJID, JID, domain=domain)
|
|
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
|
|
|
def testACEDomain(self):
|
|
domain = 'xn--bcher-kva.ch'
|
|
jid1 = JID(domain=domain)
|
|
jid2 = JID('user@%s/resource' % domain)
|
|
|
|
self.assertEqual(jid1.domain.encode('utf-8'), b'b\xc3\xbccher.ch')
|
|
self.assertEqual(jid2.domain.encode('utf-8'), b'b\xc3\xbccher.ch')
|
|
|
|
def testJIDEscapeExistingSequences(self):
|
|
jid = JID(local='blah\\foo\\20bar', domain='example.com')
|
|
self.assertEqual(jid.local, 'blah\\foo\\5c20bar')
|
|
|
|
def testJIDEscape(self):
|
|
jid = JID(local='here\'s_a_wild_&_/cr%zy/_address_for:<wv>("IMPS")',
|
|
domain='example.com')
|
|
self.assertEqual(jid.local, r'here\27s_a_wild_\26_\2fcr%zy\2f_address_for\3a\3cwv\3e(\22IMPS\22)')
|
|
|
|
def testJIDUnescape(self):
|
|
jid = JID(local='here\'s_a_wild_&_/cr%zy/_address_for:<wv>("IMPS")',
|
|
domain='example.com')
|
|
ujid = jid.unescape()
|
|
self.assertEqual(ujid.local, 'here\'s_a_wild_&_/cr%zy/_address_for:<wv>("IMPS")')
|
|
|
|
jid = JID(local='blah\\foo\\20bar', domain='example.com')
|
|
ujid = jid.unescape()
|
|
self.assertEqual(ujid.local, 'blah\\foo\\20bar')
|
|
|
|
def testStartOrEndWithEscapedSpaces(self):
|
|
local = ' foo'
|
|
self.assertRaises(InvalidJID, JID, local=local, domain='example.com')
|
|
self.assertRaises(InvalidJID, JID, '%s@example.com' % local)
|
|
|
|
local = 'bar '
|
|
self.assertRaises(InvalidJID, JID, local=local, domain='example.com')
|
|
self.assertRaises(InvalidJID, JID, '%s@example.com' % local)
|
|
|
|
# Need more input for these cases. A JID starting with \20 *is* valid
|
|
# according to RFC 6122, but is not according to XEP-0106.
|
|
#self.assertRaises(InvalidJID, JID, '%s@example.com' % '\\20foo2')
|
|
#self.assertRaises(InvalidJID, JID, '%s@example.com' % 'bar2\\20')
|
|
|
|
def testNodePrepIdemptotent(self):
|
|
node = 'ᴹᴵᴷᴬᴱᴸ'
|
|
self.assertEqual(nodeprep(node), nodeprep(nodeprep(node)))
|
|
|
|
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(TestJIDClass)
|