def test__write_protocol_entry(self, write_mock):
"""
Should call _write with "test data"
"""
from protocol.logfileprotocol import LogFileProtocol
actual_data = "test data"
actual_logfileprotocol = LogFileProtocol(examiner="tst")
actual_logfileprotocol._write_protocol_entry(entrydata=actual_data,
entryheader=None)
expected_data = "test data\r\n"
self.assertEqual(expected_data, write_mock.call_args[0][0])
def test___init__with_own_protocol_log_file(self):
"""
self._procotollogfile should be set to the provided protocol log file name.
:return:
"""
from protocol.logfileprotocol import LogFileProtocol
protocol = LogFileProtocol(
own_protocol_filename=self.expected_log_file, examiner='tada')
try:
self.assertEqual(self.expected_log_file,
protocol._protocolfilename)
finally:
protocol.__del__()
def test_get_current_file_counter(self, placeholder_mock, glob_mock):
"""
Should return 4 when there are three log file protocols in the directory
"""
from protocol.logfileprotocol import LogFileProtocol
glob_mock.return_value = [
'00001_tst_2020-05-26.txt', '00002_tst_2020-05-26.txt',
'00003_tst_2020-05-26.txt'
]
actual_protocol = LogFileProtocol(examiner='tst',
filedate=self.actual_filedate,
own_protocol_filename='')
actual_counter = actual_protocol._get_current_file_counter(
actual_protocol._search_pattern)
expected_counter = 4
self.assertEqual(expected_counter, actual_counter)
def test_set_task_metadata(self, write_mock):
"""
Should call _write with "metadata1: data1\r\n" and "metadata2: data2\r\n"
"""
from protocol.logfileprotocol import LogFileProtocol
from businesslogic.data import CollectionMetaData
metadata = OrderedDict()
metadata["metadata1"] = "data1"
metadata["metadata2"] = "data2"
actual_metadata = CollectionMetaData(metadata)
actual_logfileprotocol = LogFileProtocol(examiner="tst")
expected_data1 = "metadata1: data1\r\n"
expected_data2 = "metadata2: data2\r\n"
actual_logfileprotocol.set_task_metadata(metadata=actual_metadata)
self.assertEqual(expected_data1, write_mock.call_args_list[0].args[0])
self.assertEqual(expected_data2, write_mock.call_args_list[1].args[0])
def test__get_header_seperator(self):
"""
Should return a string with as many '*' as the header is long
"""
from protocol.logfileprotocol import LogFileProtocol
actual_data = "This is a test header"
expected_seperator = '*' * len(actual_data)
actual_seperator = LogFileProtocol._get_header_seperator(actual_data)
self.assertEqual(expected_seperator, actual_seperator)
def test_protocol_filename_empty(self):
"""
Should return 00001_tst_2020-05-26.txt
"""
from protocol.logfileprotocol import LogFileProtocol
actual_examiner = "tst"
actual_log_file_protocol_writer = LogFileProtocol(
examiner=actual_examiner, filedate=self.actual_filedate)
expected_log_file_name = "00001_tst_2020-05-26.txt"
self.assertEqual(expected_log_file_name,
actual_log_file_protocol_writer.protocol_filename)
def test_protocol_filename_counter_2(self, placeholder_mock, counter_mock):
"""
Should return current log file name "00002_tst_2020-05-26.txt
"""
from protocol.logfileprotocol import LogFileProtocol
actual_tester = 'tst'
counter_mock.return_value = 2
actual_log_file_protocol_writer = LogFileProtocol(
examiner=actual_tester, filedate=self.actual_filedate)
expected_log_file_name = "00002_tst_2020-05-26.txt"
self.assertEqual(expected_log_file_name,
actual_log_file_protocol_writer.protocol_filename)
def get_collector(
cls,
instructionsfile: str,
examiner: str = '',
placeholderfile: str = Placeholder.get_globalplaceholderfile(),
protocollogfile: str = ''):
Placeholder.set_globalplaceholderfile(placeholderfile)
cls._logger.debug(f"Protocol log file: '{protocollogfile}'")
protocol = LogFileProtocol(examiner,
own_protocol_filename=protocollogfile)
xmlparser = XmlParser(instructionsfile, protocol)
collector = cls(parser=xmlparser, protocol=protocol)
return collector
def test_protocol_filename_with_artefactid(self):
"""
Should return 1.2.3.4_tst_2020-05-26.txt
:return:
"""
from protocol.logfileprotocol import LogFileProtocol
actual_examiner = "tst"
actual_log_file_protocol_writer = LogFileProtocol(
examiner=actual_examiner,
filedate=self.actual_filedate,
artifactid="1.2.3.4")
expected_log_file_name = "1.2.3.4_tst_2020-05-26.txt"
self.assertEqual(expected_log_file_name,
actual_log_file_protocol_writer.protocol_filename)
def test__prepare_protocol_entry_with_header(self):
"""
Should write entry into file looking like this:
***************************
This is another test header
***************************
"""
from protocol.logfileprotocol import LogFileProtocol
actual_header = "This is another test header"
expected_entr = "***************************\r\n" \
"This is another test header\r\n" \
"***************************\r\n"
actual_entry = LogFileProtocol._prepare_protocol_entry(
entryheader=actual_header, entrydata=None)
self.assertEqual(expected_entr, actual_entry)