libs/openid/test/test_accept.py

   1 import unittest
   2 import os.path
   3 from openid.yadis import accept
   4
   5 def getTestData():
   6     """Read the test data off of disk
   7
   8     () -> [(int, str)]
   9     """
  10     filename = os.path.join(os.path.dirname(__file__), 'data', 'accept.txt')
  11     i = 1
  12     lines = []
  13     for line in file(filename):
  14         lines.append((i, line))
  15         i += 1
  16     return lines
  17
  18 def chunk(lines):
  19     """Return groups of lines separated by whitespace or comments
  20
  21     [(int, str)] -> [[(int, str)]]
  22     """
  23     chunks = []
  24     chunk = []
  25     for lineno, line in lines:
  26         stripped = line.strip()
  27         if not stripped or stripped[0] == '#':
  28             if chunk:
  29                 chunks.append(chunk)
  30                 chunk = []
  31         else:
  32             chunk.append((lineno, stripped))
  33
  34     if chunk:
  35         chunks.append(chunk)
  36
  37     return chunks
  38
  39 def parseLines(chunk):
  40     """Take the given chunk of lines and turn it into a test data dictionary
  41
  42     [(int, str)] -> {str:(int, str)}
  43     """
  44     items = {}
  45     for (lineno, line) in chunk:
  46         header, data = line.split(':', 1)
  47         header = header.lower()
  48         items[header] = (lineno, data.strip())
  49
  50     return items
  51
  52 def parseAvailable(available_text):
  53     """Parse an Available: line's data
  54
  55     str -> [str]
  56     """
  57     return [s.strip() for s in available_text.split(',')]
  58
  59 def parseExpected(expected_text):
  60     """Parse an Expected: line's data
  61
  62     str -> [(str, float)]
  63     """
  64     expected = []
  65     if expected_text:
  66         for chunk in expected_text.split(','):
  67             chunk = chunk.strip()
  68             mtype, qstuff = chunk.split(';')
  69             mtype = mtype.strip()
  70             assert '/' in mtype
  71             qstuff = qstuff.strip()
  72             q, qstr = qstuff.split('=')
  73             assert q == 'q'
  74             qval = float(qstr)
  75             expected.append((mtype, qval))
  76
  77     return expected
  78
  79 class MatchAcceptTest(unittest.TestCase):
  80     def __init__(self, descr, accept_header, available, expected):
  81         unittest.TestCase.__init__(self)
  82         self.accept_header = accept_header
  83         self.available = available
  84         self.expected = expected
  85         self.descr = descr
  86
  87     def shortDescription(self):
  88         return self.descr
  89
  90     def runTest(self):
  91         accepted = accept.parseAcceptHeader(self.accept_header)
  92         actual = accept.matchTypes(accepted, self.available)
  93         self.failUnlessEqual(self.expected, actual)
  94
  95 def pyUnitTests():
  96     lines = getTestData()
  97     chunks = chunk(lines)
  98     data_sets = map(parseLines, chunks)
  99     cases = []
 100     for data in data_sets:
 101         lnos = []
 102         lno, header = data['accept']
 103         lnos.append(lno)
 104         lno, avail_data = data['available']
 105         lnos.append(lno)
 106         try:
 107             available = parseAvailable(avail_data)
 108         except:
 109             print 'On line', lno
 110             raise
 111
 112         lno, exp_data = data['expected']
 113         lnos.append(lno)
 114         try:
 115             expected = parseExpected(exp_data)
 116         except:
 117             print 'On line', lno
 118             raise
 119
 120         descr = 'MatchAcceptTest for lines %r' % (lnos,)
 121         case = MatchAcceptTest(descr, header, available, expected)
 122         cases.append(case)
 123     return unittest.TestSuite(cases)
 124
 125 if __name__ == '__main__':
 126     runner = unittest.TextTestRunner()
 127     runner.run(loadTests())