def test_priority_prefix_instrument(self): global counter counter = 0 instrument1 = _instantiate(MockInstrument4) instrument2 = _instantiate(MockInstrument5) instrument3 = _instantiate(MockInstrument6) instrumentation.install(instrument1) instrumentation.install(instrument2) instrumentation.install(instrument3) signal.send(signal.BEFORE_FIRST_ITERATION_BOOT, self, context=None) assert_equal(counter, 42)
def test_install(self): instrument = _instantiate(MockInstrument) instrument2 = _instantiate(MockInstrument2) instrumentation.install(instrument) instrumentation.install(instrument2) signal.send(signal.BEFORE_WORKLOAD_EXECUTION, self, context=None) signal.send(signal.AFTER_WORKLOAD_EXECUTION, self, context=None) signal.send(signal.AFTER_WORKLOAD_RESULT_UPDATE, self, context=None) assert_equal(instrument.before, 1) assert_equal(instrument.after, 1) assert_equal(instrument2.before, 1) assert_equal(instrument2.after, 1) assert_equal(instrument2.result, 1)
def test_local_instrument(self): global counter counter = 0 self.install_local_instrument() signal.send(signal.BEFORE_WORKLOAD_EXECUTION, self, context=None) assert_equal(counter, 1)
def _send(self, s): signal.send(s, self, self.context)
def emit(self, record): if record.levelno == logging.ERROR: signal.send(signal.ERROR_LOGGED, self) elif record.levelno == logging.WARNING: signal.send(signal.WARNING_LOGGED, self)