# -*- 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, "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:("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:("IMPS")', domain='example.com') ujid = jid.unescape() self.assertEqual(ujid.local, 'here\'s_a_wild_&_/cr%zy/_address_for:("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)