libs/openid/test/test_xri.py

   1 from unittest import TestCase
   2 from openid.yadis import xri
   3
   4 class XriDiscoveryTestCase(TestCase):
   5     def test_isXRI(self):
   6         i = xri.identifierScheme
   7         self.failUnlessEqual(i('=john.smith'), 'XRI')
   8         self.failUnlessEqual(i('@smiths/john'), 'XRI')
   9         self.failUnlessEqual(i('smoker.myopenid.com'), 'URI')
  10         self.failUnlessEqual(i('xri://=john'), 'XRI')
  11         self.failUnlessEqual(i(''), 'URI')
  12
  13
  14 class XriEscapingTestCase(TestCase):
  15     def test_escaping_percents(self):
  16         self.failUnlessEqual(xri.escapeForIRI('@example/abc%2Fd/ef'),
  17                              '@example/abc%252Fd/ef')
  18
  19
  20     def test_escaping_xref(self):
  21         # no escapes
  22         esc = xri.escapeForIRI
  23         self.failUnlessEqual('@example/foo/(@bar)', esc('@example/foo/(@bar)'))
  24         # escape slashes
  25         self.failUnlessEqual('@example/foo/(@bar%2Fbaz)',
  26                              esc('@example/foo/(@bar/baz)'))
  27         self.failUnlessEqual('@example/foo/(@bar%2Fbaz)/(+a%2Fb)',
  28                              esc('@example/foo/(@bar/baz)/(+a/b)'))
  29         # escape query ? and fragment #
  30         self.failUnlessEqual('@example/foo/(@baz%3Fp=q%23r)?i=j#k',
  31                              esc('@example/foo/(@baz?p=q#r)?i=j#k'))
  32
  33
  34
  35 class XriTransformationTestCase(TestCase):
  36     def test_to_iri_normal(self):
  37         self.failUnlessEqual(xri.toIRINormal('@example'), 'xri://@example')
  38
  39     try:
  40         unichr(0x10000)
  41     except ValueError:
  42         # bleh narrow python build
  43         def test_iri_to_url(self):
  44             s = u'l\xa1m'
  45             expected = 'l%C2%A1m'
  46             self.failUnlessEqual(xri.iriToURI(s), expected)
  47     else:
  48         def test_iri_to_url(self):
  49             s = u'l\xa1m\U00101010n'
  50             expected = 'l%C2%A1m%F4%81%80%90n'
  51             self.failUnlessEqual(xri.iriToURI(s), expected)
  52
  53
  54
  55 class CanonicalIDTest(TestCase):
  56     def mkTest(providerID, canonicalID, isAuthoritative):
  57         def test(self):
  58             result = xri.providerIsAuthoritative(providerID, canonicalID)
  59             format = "%s providing %s, expected %s"
  60             message = format % (providerID, canonicalID, isAuthoritative)
  61             self.failUnlessEqual(isAuthoritative, result, message)
  62
  63         return test
  64
  65     test_equals = mkTest('=', '=!698.74D1.A1F2.86C7', True)
  66     test_atOne = mkTest('@!1234', '@!1234!ABCD', True)
  67     test_atTwo = mkTest('@!1234!5678', '@!1234!5678!ABCD', True)
  68
  69     test_atEqualsFails = mkTest('@!1234', '=!1234!ABCD', False)
  70     test_tooDeepFails = mkTest('@!1234', '@!1234!ABCD!9765', False)
  71     test_atEqualsAndTooDeepFails = mkTest('@!1234!ABCD', '=!1234', False)
  72     test_differentBeginningFails = mkTest('=!BABE', '=!D00D', False)
  73
  74 class TestGetRootAuthority(TestCase):
  75     def mkTest(the_xri, expected_root):
  76         def test(self):
  77             actual_root = xri.rootAuthority(the_xri)
  78             self.failUnlessEqual(actual_root, xri.XRI(expected_root))
  79         return test
  80
  81     test_at = mkTest("@foo", "@")
  82     test_atStar = mkTest("@foo*bar", "@")
  83     test_atStarStar = mkTest("@*foo*bar", "@")
  84     test_atWithPath = mkTest("@foo/bar", "@")
  85     test_bangBang = mkTest("!!990!991", "!")
  86     test_bang = mkTest("!1001!02", "!")
  87     test_equalsStar = mkTest("=foo*bar", "=")
  88     test_xrefPath = mkTest("(example.com)/foo", "(example.com)")
  89     test_xrefStar = mkTest("(example.com)*bar/foo", "(example.com)")
  90     test_uriAuth = mkTest("baz.example.com/foo", "baz.example.com")
  91     test_uriAuthPort = mkTest("baz.example.com:8080/foo",
  92                               "baz.example.com:8080")
  93
  94     # Looking at the ABNF in XRI Syntax 2.0, I don't think you can
  95     # have example.com*bar.  You can do (example.com)*bar, but that
  96     # would mean something else.
  97     ##("example.com*bar/(=baz)", "example.com*bar"),
  98     ##("baz.example.com!01/foo", "baz.example.com!01"),
  99
 100 if __name__ == '__main__':
 101     import unittest
 102     unittest.main()