mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-01-19 04:21:17 +00:00
227 lines
6.6 KiB
Python
227 lines
6.6 KiB
Python
# Copyright 2013-2015 ARM Limited
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
|
|
# pylint: disable=W0231,W0613,E0611,W0603,R0201
|
|
from unittest import TestCase
|
|
|
|
from nose.tools import assert_equal, raises, assert_true, assert_false
|
|
|
|
from wlauto import Instrument
|
|
from wlauto.core import signal, instrumentation
|
|
from wlauto.instrumentation import instrument_is_installed, clear_instrumentation
|
|
|
|
|
|
class MockInstrument(Instrument):
|
|
|
|
name = 'mock'
|
|
|
|
def __init__(self):
|
|
Instrument.__init__(self, None)
|
|
self.before = 0
|
|
self.after = 0
|
|
|
|
def before_workload_execution(self, context):
|
|
self.before += 1
|
|
|
|
def after_workload_execution(self, context):
|
|
self.after += 1
|
|
|
|
|
|
class MockInstrument2(Instrument):
|
|
|
|
name = 'mock_2'
|
|
|
|
def __init__(self):
|
|
Instrument.__init__(self, None)
|
|
self.before = 0
|
|
self.after = 0
|
|
self.result = 0
|
|
|
|
def before_workload_execution(self, context):
|
|
self.before += 1
|
|
|
|
def after_workload_execution(self, context):
|
|
self.after += 1
|
|
|
|
def after_workload_result_update(self, context):
|
|
self.result += 1
|
|
|
|
|
|
class MockInstrument3(Instrument):
|
|
|
|
name = 'mock_3'
|
|
|
|
def __init__(self):
|
|
Instrument.__init__(self, None)
|
|
|
|
def slow_before_workload_execution(self, context):
|
|
global counter
|
|
counter += 1
|
|
|
|
|
|
class MockInstrument4(Instrument):
|
|
|
|
name = 'mock_4'
|
|
|
|
def __init__(self):
|
|
Instrument.__init__(self, None)
|
|
|
|
def slow_before_first_iteration_boot(self, context):
|
|
global counter
|
|
counter = 4
|
|
|
|
|
|
class MockInstrument5(Instrument):
|
|
|
|
name = 'mock_5'
|
|
|
|
def __init__(self):
|
|
Instrument.__init__(self, None)
|
|
|
|
def fast_before_first_iteration_boot(self, context):
|
|
global counter
|
|
counter += 2
|
|
|
|
|
|
class MockInstrument6(Instrument):
|
|
|
|
name = 'mock_6'
|
|
|
|
def __init__(self):
|
|
Instrument.__init__(self, None)
|
|
|
|
def before_first_iteration_boot(self, context):
|
|
global counter
|
|
counter *= 10
|
|
|
|
|
|
class BadInstrument(Instrument):
|
|
|
|
name = 'bad'
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
# Not specifying the context argument.
|
|
def teardown(self):
|
|
pass
|
|
|
|
|
|
counter = 0
|
|
|
|
|
|
class InstrumentationTest(TestCase):
|
|
|
|
def tearDown(self):
|
|
clear_instrumentation()
|
|
|
|
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_enable_disable(self):
|
|
instrument = _instantiate(MockInstrument)
|
|
instrument2 = _instantiate(MockInstrument2)
|
|
instrumentation.install(instrument)
|
|
instrumentation.install(instrument2)
|
|
|
|
instrumentation.disable_all()
|
|
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, 0)
|
|
assert_equal(instrument.after, 0)
|
|
assert_equal(instrument2.before, 0)
|
|
assert_equal(instrument2.after, 0)
|
|
assert_equal(instrument2.result, 0)
|
|
|
|
instrumentation.enable(instrument)
|
|
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, 0)
|
|
assert_equal(instrument2.after, 0)
|
|
assert_equal(instrument2.result, 0)
|
|
|
|
instrumentation.enable_all()
|
|
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, 2)
|
|
assert_equal(instrument.after, 2)
|
|
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 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)
|
|
|
|
@raises(ValueError)
|
|
def test_bad_argspec(self):
|
|
instrument = _instantiate(BadInstrument)
|
|
instrumentation.install(instrument)
|
|
|
|
def test_check_installed(self):
|
|
instrumentation.install(_instantiate(MockInstrument))
|
|
assert_true(instrument_is_installed('mock'))
|
|
assert_true(instrument_is_installed(MockInstrument))
|
|
assert_false(instrument_is_installed(MockInstrument2))
|
|
|
|
def install_local_instrument(self):
|
|
instrument = _instantiate(MockInstrument3)
|
|
instrumentation.install(instrument)
|
|
|
|
@raises(ValueError)
|
|
def test_duplicate_install(self):
|
|
instrument = _instantiate(MockInstrument)
|
|
instrument2 = _instantiate(MockInstrument)
|
|
instrumentation.install(instrument)
|
|
instrumentation.install(instrument2)
|
|
|
|
|
|
def _instantiate(cls):
|
|
# Needed to get around Extension's __init__ checks
|
|
return cls()
|
|
|