296 lines
11 KiB
Python
296 lines
11 KiB
Python
# -*- encoding: utf8 -*-
|
||
from __future__ import unicode_literals
|
||
import unittest
|
||
from slixmpp.test import SlixTest
|
||
from slixmpp import JID, InvalidJID
|
||
from slixmpp.jid import nodeprep
|
||
|
||
|
||
class TestJIDClass(SlixTest):
|
||
|
||
"""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, "Different JIDs are considered equal")
|
||
self.assertTrue(jid1 != jid2, "Different JIDs are considered equal")
|
||
|
||
def testZeroLengthDomain(self):
|
||
jid1 = JID('')
|
||
jid2 = JID()
|
||
self.assertTrue(jid1 == jid2, "Empty JIDs are not considered equal")
|
||
self.assertTrue(jid1.domain == '', "Empty JID’s domain part not empty")
|
||
self.assertTrue(jid1.full == '', "Empty JID’s full part not empty")
|
||
|
||
self.assertRaises(InvalidJID, JID, 'user@')
|
||
self.assertRaises(InvalidJID, JID, '/resource')
|
||
self.assertRaises(InvalidJID, JID, 'user@/resource')
|
||
|
||
def testZeroLengthLocalPart(self):
|
||
self.assertRaises(InvalidJID, JID, '@test.com')
|
||
self.assertRaises(InvalidJID, JID, '@test.com/resource')
|
||
|
||
def testZeroLengthNodeDomain(self):
|
||
self.assertRaises(InvalidJID, JID, '@/test.com')
|
||
|
||
def testZeroLengthResource(self):
|
||
self.assertRaises(InvalidJID, JID, 'test.com/')
|
||
self.assertRaises(InvalidJID, JID, 'user@test.com/')
|
||
|
||
def test1023LengthDomain(self):
|
||
domain = ('a.' * 509) + 'a.com'
|
||
jid = JID('user@%s/resource' % domain)
|
||
|
||
def test1023LengthLocalPart(self):
|
||
local = 'a' * 1023
|
||
jid = JID('%s@test.com' % local)
|
||
|
||
def test1023LengthResource(self):
|
||
resource = 'r' * 1023
|
||
jid = JID('test.com/%s' % resource)
|
||
|
||
def test1024LengthDomain(self):
|
||
domain = ('a.' * 509) + 'aa.com'
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, '%s/resource' % domain)
|
||
self.assertRaises(InvalidJID, JID, domain)
|
||
|
||
def test1024LengthLocalPart(self):
|
||
local = 'a' * 1024
|
||
self.assertRaises(InvalidJID, JID, '%s@test.com' % local)
|
||
self.assertRaises(InvalidJID, JID, '%s@test.com/resource' % local)
|
||
|
||
def test1024LengthResource(self):
|
||
resource = 'r' * 1024
|
||
self.assertRaises(InvalidJID, JID, 'test.com/%s' % resource)
|
||
self.assertRaises(InvalidJID, JID, 'user@test.com/%s' % resource)
|
||
|
||
def testTooLongDomainLabel(self):
|
||
domain = ('a' * 64) + '.com'
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
|
||
def testDomainEmptyLabel(self):
|
||
domain = 'aaa..bbb.com'
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
|
||
def testDomainIPv4(self):
|
||
domain = '127.0.0.1'
|
||
|
||
jid1 = JID('%s' % domain)
|
||
jid2 = JID('user@%s' % domain)
|
||
jid3 = JID('%s/resource' % domain)
|
||
jid4 = JID('user@%s/resource' % domain)
|
||
|
||
def testDomainIPv6(self):
|
||
domain = '[::1]'
|
||
|
||
jid1 = JID('%s' % domain)
|
||
jid2 = JID('user@%s' % domain)
|
||
jid3 = JID('%s/resource' % domain)
|
||
jid4 = JID('user@%s/resource' % domain)
|
||
|
||
def testDomainInvalidIPv6NoBrackets(self):
|
||
domain = '::1'
|
||
|
||
self.assertRaises(InvalidJID, JID, '%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, '%s/resource' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
|
||
def testDomainInvalidIPv6MissingBracket(self):
|
||
domain = '[::1'
|
||
|
||
self.assertRaises(InvalidJID, JID, '%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, '%s/resource' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
|
||
def testDomainInvalidIPv6WrongBracket(self):
|
||
domain = '[::]1]'
|
||
|
||
self.assertRaises(InvalidJID, JID, '%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, '%s/resource' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
|
||
def testDomainWithPort(self):
|
||
domain = 'example.com:5555'
|
||
|
||
self.assertRaises(InvalidJID, JID, '%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s' % domain)
|
||
self.assertRaises(InvalidJID, JID, '%s/resource' % domain)
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
|
||
def testDomainWithTrailingDot(self):
|
||
domain = 'example.com.'
|
||
jid = JID('user@%s/resource' % domain)
|
||
|
||
self.assertEqual(jid.domain, 'example.com')
|
||
|
||
def testDomainWithDashes(self):
|
||
domain = 'example.com-'
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
|
||
domain = '-example.com'
|
||
self.assertRaises(InvalidJID, JID, 'user@%s/resource' % domain)
|
||
|
||
def testACEDomain(self):
|
||
domain = 'xn--bcher-kva.ch'
|
||
jid = JID('user@%s/resource' % domain)
|
||
|
||
self.assertEqual(jid.domain.encode('utf-8'), b'b\xc3\xbccher.ch')
|
||
|
||
def testJIDUnescape(self):
|
||
jid = JID('here\\27s_a_wild_\\26_\\2fcr%zy\\2f_\\40ddress\\20for\\3a\\3cwv\\3e(\\22IMPS\\22)\\5c@example.com')
|
||
ujid = jid.unescape()
|
||
self.assertEqual(ujid.local, 'here\'s_a_wild_&_/cr%zy/_@ddress for:<wv>("imps")\\')
|
||
|
||
jid = JID('blah\\5cfoo\\5c20bar@example.com')
|
||
ujid = jid.unescape()
|
||
self.assertEqual(ujid.local, 'blah\\foo\\20bar')
|
||
|
||
def testStartOrEndWithEscapedSpaces(self):
|
||
local = ' foo'
|
||
self.assertRaises(InvalidJID, JID, '%s@example.com' % local)
|
||
|
||
local = 'bar '
|
||
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)
|