def test_frames_with_extra_message(self):
# SETUP
expected_report = """\
i = 1 | i = 2
| extra message"""
# EXEC
builder = ReportBuilder()
frame1 = builder.start_frame(1, 2)
frame1.assign(name='i', value=1, line_number=1)
frame2 = builder.start_frame(1, 2)
frame2.assign(name='i', value=2, line_number=1)
builder.add_extra_message('extra message', 2)
report = builder.report()
# VERIFY
self.assertEqual(expected_report.splitlines(), report.splitlines())
def test_frames(self):
# SETUP
expected_report = """\
i = 1 | i = 2
n = 2 | """
# EXEC
builder = ReportBuilder()
frame1 = builder.start_frame(1, 2)
frame1.assign(name='i', value=1, line_number=1)
frame2 = builder.start_frame(1, 2)
frame2.assign(name='i', value=2, line_number=1)
frame1.assign(name='n', value=2, line_number=2)
report = builder.report()
# VERIFY
self.assertEqual(expected_report.splitlines(), report.splitlines())
def test_frames_with_extra_message(self):
# SETUP
expected_report = """\
i = 1 | i = 2
| extra message"""
# EXEC
builder = ReportBuilder()
frame1 = builder.start_frame(1, 2)
frame1.assign(name='i', value=1, line_number=1)
frame2 = builder.start_frame(1, 2)
frame2.assign(name='i', value=2, line_number=1)
builder.add_extra_message('extra message', 2)
report = builder.report()
# VERIFY
self.assertReportEqual(expected_report, report)
def test_frames(self):
# SETUP
expected_report = """\
i = 1 | i = 2
n = 2 | """
# EXEC
builder = ReportBuilder()
frame1 = builder.start_frame(1, 2)
frame1.assign(name='i', value=1, line_number=1)
frame2 = builder.start_frame(1, 2)
frame2.assign(name='i', value=2, line_number=1)
frame1.assign(name='n', value=2, line_number=2)
report = builder.report()
# VERIFY
self.assertReportEqual(expected_report, report)
def test_message_limit_with_empty_blocks(self):
# SETUP
expected_report = """\
"""
# EXEC
builder = ReportBuilder(message_limit=2)
builder.start_frame(1, 5)
builder.start_block(1, 3)
builder.start_block(1, 3)
report = builder.report()
# VERIFY
self.assertReportEqual(expected_report, report)
def test_count_all_messages(self):
# SETUP
# EXEC
builder = ReportBuilder()
frame = builder.start_frame(1, 2)
frame.add_message('a', 1)
frame.add_message('b', 2)
# VERIFY
self.assertEqual(0, builder.message_count)
self.assertEqual(2, builder.count_all_messages())
