class TestFileReader(BZTestCase):
def setUp(self):
super(TestFileReader, self).setUp()
self.obj = MockFileReader()
def configure(self, file_name):
self.obj.name = file_name
def tearDown(self):
if self.obj and self.obj.fds:
self.obj.fds.close()
super(TestFileReader, self).tearDown()
def test_file_len(self):
self.configure(join(RESOURCES_DIR, 'jmeter', 'jtl', 'file.notfound'))
self.sniff_log(self.obj.log)
list(self.obj.get_lines(size=1))
self.assertIn('File not appeared yet', self.log_recorder.debug_buff.getvalue())
self.obj.name = join(RESOURCES_DIR, 'jmeter', 'jtl', 'unicode.jtl')
lines = list(self.obj.get_lines(size=1))
self.assertEqual(1, len(lines))
lines = list(self.obj.get_lines(last_pass=True))
self.assertEqual(13, len(lines))
self.assertTrue(all(l.endswith('\n') for l in lines))
def test_decode(self):
old_string = "Тест.Эхо"
fd, gen_file_name = tempfile.mkstemp()
os.close(fd)
mod_str = old_string + '\n'
if PY2:
mod_str = bytearray(mod_str).decode('utf-8') # convert to utf-8 on py2 for writing...
with open(gen_file_name, 'wb') as fd: # use target system encoding for writing
fd.write(mod_str.encode(self.obj.SYS_ENCODING)) # important on win where it's not 'utf-8'
try:
self.configure(gen_file_name)
self.assertEqual('utf-8', self.obj.cp)
lines = list(self.obj.get_lines(True))
self.assertEqual(self.obj.SYS_ENCODING, self.obj.cp) # on win self.obj.cp must be changed during of
self.assertEqual(1, len(lines)) # reading (see MockFileReader)
new_string = lines[0].rstrip()
if PY2:
new_string = new_string.encode('utf-8')
self.assertEqual(old_string, new_string)
finally:
if self.obj.fds:
self.obj.fds.close()
os.remove(gen_file_name)
def setUp(self):
super(TestFileReader, self).setUp()
self.obj = MockFileReader()
def setUp(self):
super(TestFileReader, self).setUp()
self.obj = MockFileReader()