Esempio n. 1
0
    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)))