1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
from __future__ import print_function
import unittest
import os
import tempfile
import glob
debug = False
class BaseCVsTest(unittest.TestCase):
def remove_file_and_directories(self, filename):
os.remove(filename)
filename = os.path.dirname(filename)
while glob.glob(os.path.join(filename, "*")) == []:
os.rmdir(filename)
filename = os.path.dirname(filename)
def setUp(self, *args, **kwargs):
# --------------
# Create tmpfile
# --------------
self.tmpfile = tempfile.mkstemp()[1]
if debug:
print("TEMP:", self.tmpfile)
self.delete_files = []
def tearDown(self):
if debug:
print("would be unlinking:", self.tmpfile)
return
os.unlink(self.tmpfile)
for filename in self.delete_files:
self.remove_file_and_directories(filename)
self.delete_files.remove(filename)
def assertCV(self, text_to_find, line_trigger='Error:', number_of_lines_to_scan=1):
line_to_scan = ""
if debug:
print("LINE TRIGGER:", line_trigger)
with open(self.tmpfile) as f:
lines = f.readlines()
for i, line in enumerate(lines):
scan = "".join(lines[i:i+number_of_lines_to_scan]).strip()
if debug: print("LINE:", i, __file__, scan)
if line_trigger in line:
line_to_scan = scan
break
if debug:
print("SCANNED:", line_to_scan)
self.assertIn(text_to_find, line_to_scan)
|
