diff options
Diffstat (limited to 'test/unit.py')
-rw-r--r-- | test/unit.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/test/unit.py b/test/unit.py index dfd8f4fc..ae6154aa 100644 --- a/test/unit.py +++ b/test/unit.py @@ -2,6 +2,7 @@ import os import re import sys import json +import mmap import time import shutil import socket @@ -363,9 +364,14 @@ class TestUnitApplicationProto(TestUnitControl): return time.mktime(time.strptime(date, '%a, %d %b %Y %H:%M:%S %Z')) def search_in_log(self, pattern): - with open(self.testdir + '/unit.log', 'r') as f: + with open(self.testdir + '/unit.log', 'r', errors='ignore') as f: return re.search(pattern, f.read()) + def find_in_log(self, pattern): + with open(self.testdir + '/unit.log', 'rb', 0) as f, \ + mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ) as s: + return s.find(pattern) != -1 + class TestUnitApplicationPython(TestUnitApplicationProto): def load(self, script, name=None): if name is None: |