def test_continuous_event_without_error(self): begin_event_timestamp = 1623596860.1202102 scylla_bench_event = ScyllaBenchEvent(node="node", stress_cmd="stress_cmd", log_file_name="log_file_name", publish_event=False) scylla_bench_event.event_id = "14f35b64-2fcc-4b6e-a09d-4aeaf4faa543" begin_event = scylla_bench_event.begin_event() begin_event.event_timestamp = begin_event.begin_timestamp = begin_event_timestamp self.assertEqual( str(begin_event), "(ScyllaBenchEvent Severity.NORMAL) period_type=begin " "event_id=14f35b64-2fcc-4b6e-a09d-4aeaf4faa543: node=node\nstress_cmd=stress_cmd" ) self.assertEqual(begin_event.event_timestamp, begin_event_timestamp) self.assertEqual(begin_event.timestamp, begin_event_timestamp) self.assertEqual(begin_event, pickle.loads(pickle.dumps(begin_event))) scylla_bench_event.end_event() scylla_bench_event.event_timestamp = scylla_bench_event.end_timestamp = 1623596861.1202102 self.assertEqual( str(scylla_bench_event), '(ScyllaBenchEvent Severity.NORMAL) period_type=end ' 'event_id=14f35b64-2fcc-4b6e-a09d-4aeaf4faa543 duration=1s: ' 'node=node\nstress_cmd=stress_cmd') scylla_bench_event.log_file_name = "log_file_name" self.assertEqual(scylla_bench_event, pickle.loads(pickle.dumps(scylla_bench_event)))
def test_continuous_event_with_error(self): begin_event_timestamp = 1623596860.1202102 scylla_bench_event = ScyllaBenchEvent(node="node", stress_cmd="stress_cmd", log_file_name="log_file_name", publish_event=False) scylla_bench_event.event_id = "14f35b64-2fcc-4b6e-a09d-4aeaf4faa543" begin_event = scylla_bench_event.begin_event() begin_event.timestamp = begin_event.begin_timestamp = begin_event_timestamp self.assertEqual( str(begin_event), "(ScyllaBenchEvent Severity.NORMAL) period_type=begin " "event_id=14f35b64-2fcc-4b6e-a09d-4aeaf4faa543: node=node\nstress_cmd=stress_cmd" ) self.assertEqual(begin_event.timestamp, begin_event_timestamp) self.assertEqual(begin_event, pickle.loads(pickle.dumps(begin_event))) try: raise ValueError('Stress command completed with bad status 1') except Exception as exc: # pylint: disable=broad-except scylla_bench_event.severity = Severity.ERROR scylla_bench_event.add_error([str(exc)]) scylla_bench_event.end_event() scylla_bench_event.end_timestamp = scylla_bench_event.timestamp = 1623596960.1202102 self.assertEqual( str(scylla_bench_event), '(ScyllaBenchEvent Severity.ERROR) period_type=end ' 'event_id=14f35b64-2fcc-4b6e-a09d-4aeaf4faa543 duration=1m40s: ' 'node=node\nstress_cmd=stress_cmd\nerrors:' '\n\nStress command completed with bad status 1') scylla_bench_event.log_file_name = "log_file_name" self.assertEqual(scylla_bench_event, pickle.loads(pickle.dumps(scylla_bench_event)))