Tests: added flags to search functions in proto.py.
Also removed unnesessary re.compile() calls.
This commit is contained in:
@@ -16,18 +16,18 @@ class TestApplicationProto(TestControl):
|
||||
def date_to_sec_epoch(self, date, template='%a, %d %b %Y %X %Z'):
|
||||
return time.mktime(time.strptime(date, template))
|
||||
|
||||
def findall(self, pattern, name='unit.log'):
|
||||
def findall(self, pattern, name='unit.log', flags=re.M):
|
||||
with Log.open(name) as f:
|
||||
return re.findall(pattern, f.read())
|
||||
return re.findall(pattern, f.read(), flags)
|
||||
|
||||
def search_in_log(self, pattern, name='unit.log'):
|
||||
def search_in_log(self, pattern, name='unit.log', flags=re.M):
|
||||
with Log.open(name) as f:
|
||||
return re.search(pattern, f.read())
|
||||
return re.search(pattern, f.read(), flags)
|
||||
|
||||
def wait_for_record(self, pattern, name='unit.log', wait=150):
|
||||
def wait_for_record(self, pattern, name='unit.log', wait=150, flags=re.M):
|
||||
with Log.open(name) as f:
|
||||
for i in range(wait):
|
||||
found = re.search(pattern, f.read())
|
||||
found = re.search(pattern, f.read(), flags)
|
||||
|
||||
if found is not None:
|
||||
break
|
||||
|
||||
@@ -209,9 +209,7 @@ class TestHTTP:
|
||||
return {}
|
||||
|
||||
headers_text, body = m.group(1), m.group(2)
|
||||
|
||||
p = re.compile('(.*?)\x0d\x0a?', re.M | re.S)
|
||||
headers_lines = p.findall(headers_text)
|
||||
headers_lines = re.findall('(.*?)\x0d\x0a?', headers_text, re.M | re.S)
|
||||
|
||||
status = re.search(
|
||||
r'^HTTP\/\d\.\d\s(\d+)|$', headers_lines.pop(0)
|
||||
|
||||
Reference in New Issue
Block a user