Source Code for Module openid.test.linkparse

  1  from openid.consumer.html_parse import parseLinkAttrs 
  2  import os.path 
  3  import codecs 
  4  import unittest 
  5   
  6 -def parseLink(line): 
  7      parts = line.split() 
  8      optional = parts[0] == 'Link*:' 
  9      assert optional or parts[0] == 'Link:' 
 10   
 11      attrs = {} 
 12      for attr in parts[1:]: 
 13          k, v = attr.split('=', 1) 
 14          if k[-1] == '*': 
 15              attr_optional = 1 
 16              k = k[:-1] 
 17          else: 
 18              attr_optional = 0 
 19   
 20          attrs[k] = (attr_optional, v) 
 21   
 22      return (optional, attrs) 
 23   
 24 -def parseCase(s): 
 25      header, markup = s.split('\n\n', 1) 
 26      lines = header.split('\n') 
 27      name = lines.pop(0) 
 28      assert name.startswith('Name: ') 
 29      desc = name[6:] 
 30      return desc, markup, map(parseLink, lines) 
 31   
 32 -def parseTests(s): 
 33      tests = [] 
 34   
 35      cases = s.split('\n\n\n') 
 36      header = cases.pop(0) 
 37      tests_line, _ = header.split('\n', 1) 
 38      k, v = tests_line.split(': ') 
 39      assert k == 'Num Tests' 
 40      num_tests = int(v) 
 41   
 42      for case in cases[:-1]: 
 43          desc, markup, links = parseCase(case) 
 44          tests.append((desc, markup, links, case)) 
 45   
 46      return num_tests, tests 
 47   
 48 -class _LinkTest(unittest.TestCase): 
 49 -    def __init__(self, desc, case, expected, raw): 
 50          unittest.TestCase.__init__(self) 
 51          self.desc = desc 
 52          self.case = case 
 53          self.expected = expected 
 54          self.raw = raw 
 55   
 56 -    def shortDescription(self): 
 57          return self.desc 
 58   
 59 -    def runTest(self): 
 60          actual = parseLinkAttrs(self.case) 
 61          i = 0 
 62          for optional, exp_link in self.expected: 
 63              if optional: 
 64                  if i >= len(actual): 
 65                      continue 
 66   
 67              act_link = actual[i] 
 68              for k, (o, v) in exp_link.items(): 
 69                  if o: 
 70                      act_v = act_link.get(k) 
 71                      if act_v is None: 
 72                          continue 
 73                  else: 
 74                      act_v = act_link[k] 
 75   
 76                  if optional and v != act_v: 
 77                      break 
 78   
 79                  self.assertEqual(v, act_v) 
 80              else: 
 81                  i += 1 
 82   
 83          assert i == len(actual) 
 84   
 85 -def pyUnitTests(): 
 86      here = os.path.dirname(os.path.abspath(__file__)) 
 87      test_data_file_name = os.path.join(here, 'linkparse.txt') 
 88      test_data_file = codecs.open(test_data_file_name, 'r', 'utf-8') 
 89      test_data = test_data_file.read() 
 90      test_data_file.close() 
 91   
 92      num_tests, test_cases = parseTests(test_data) 
 93   
 94      tests = [_LinkTest(*case) for case in test_cases] 
 95   
 96      def test_parseSucceeded(): 
 97          assert len(test_cases) == num_tests, (len(test_cases), num_tests) 
 98   
 99      check_desc = 'Check that we parsed the correct number of test cases' 
100      check = unittest.FunctionTestCase( 
101          test_parseSucceeded, description=check_desc) 
102      tests.insert(0, check) 
103   
104      return unittest.TestSuite(tests) 
105   
106  if __name__ == '__main__': 
107      suite = pyUnitTests() 
108      runner = unittest.TextTestRunner() 
109      runner.run(suite) 
110