tools/httptester/BaseTest.py

   1 """The test case itsself, and associated stuff"""
   2
   3 import string
   4
   5 class BaseTest:
   6     def __init__(self):
   7         self.res = -1
   8         self.reason = None
   9
  10     def parse_config(self, config):
  11         self.files = []
  12
  13         for line in config.readlines():
  14             line = string.strip(line)
  15             self.files.append({'file': line, 'read': 0})
  16
  17     baseName = 'index'
  18
  19     def verify_request(self, req):
  20         """Check that the request is valid.
  21
  22         Also needs to update any internal 'read' stuff"""
  23
  24         ## XXXXX
  25         ## This needs to be done using exceptions, maybe
  26         ## XXXXX
  27
  28         for i in self.files:
  29             if i['file'] == req.fname:
  30                 if i['read'] == 1:
  31                     self.res = 0
  32                     self.reason = "File %s was read twice" % (req.fname)
  33                     return 0
  34                 i['read'] = 1
  35                 break
  36             elif i['read'] == 0:
  37                 self.res = 0
  38                 self.reason = "File %s requested, expected %s" % (req.fname, i['file'])
  39                 return 0
  40
  41         ### Simplistic for now...
  42         res = req.headers.getheader('Host')
  43
  44         return res
  45
  46     def result(self):
  47         if self.res == -1:
  48             for i in self.files:
  49                 if i['read'] == 0:
  50                     self.res = 0
  51                     self.reason = "%s not read" % (i['file'])
  52                     return self.res, self.reason
  53             self.res = 1
  54
  55         return self.res, self.reason
  56
  57 tester = BaseTest