| #!/usr/bin/env vpython |
| # Copyright 2018 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Tests for phased_orderfile.py.""" |
| |
| import collections |
| import unittest |
| |
| import phased_orderfile |
| import process_profiles |
| |
| from test_utils import (SimpleTestSymbol, |
| TestSymbolOffsetProcessor, |
| TestProfileManager) |
| |
| |
| class Mod10Processor(object): |
| """A restricted mock for a SymbolOffsetProcessor. |
| |
| This only implements GetReachedOffsetsFromDump, and works by mapping a dump |
| offset to offset - (offset % 10). If the dump offset is negative, it is marked |
| as not found. |
| """ |
| def GetReachedOffsetsFromDump(self, dump): |
| return [x - (x % 10) for x in dump if x >= 0] |
| |
| |
| class PhasedOrderfileTestCase(unittest.TestCase): |
| |
| def setUp(self): |
| self._file_counter = 0 |
| |
| def File(self, timestamp_sec, phase): |
| self._file_counter += 1 |
| return 'file-{}-{}.txt_{}'.format( |
| self._file_counter, timestamp_sec * 1000 * 1000 * 1000, phase) |
| |
| def testProfileStability(self): |
| symbols = [SimpleTestSymbol(str(i), i, 10) |
| for i in xrange(20)] |
| phaser = phased_orderfile.PhasedAnalyzer( |
| None, TestSymbolOffsetProcessor(symbols)) |
| opo = lambda s, c, i: phased_orderfile.OrderfilePhaseOffsets( |
| startup=s, common=c, interaction=i) |
| phaser._phase_offsets = [opo(range(5), range(6, 10), range(11,15)), |
| opo(range(4), range(6, 10), range(18, 20))] |
| self.assertEquals((1.25, 1, None), phaser.ComputeStability()) |
| |
| def testIsStable(self): |
| symbols = [SimpleTestSymbol(str(i), i, 10) |
| for i in xrange(20)] |
| phaser = phased_orderfile.PhasedAnalyzer( |
| None, TestSymbolOffsetProcessor(symbols)) |
| opo = lambda s, c, i: phased_orderfile.OrderfilePhaseOffsets( |
| startup=s, common=c, interaction=i) |
| phaser._phase_offsets = [opo(range(5), range(6, 10), range(11,15)), |
| opo(range(4), range(6, 10), range(18, 20))] |
| phaser.STARTUP_STABILITY_THRESHOLD = 1.1 |
| self.assertFalse(phaser.IsStableProfile()) |
| phaser.STARTUP_STABILITY_THRESHOLD = 1.5 |
| self.assertTrue(phaser.IsStableProfile()) |
| |
| def testGetOrderfilePhaseOffsets(self): |
| mgr = TestProfileManager({ |
| self.File(0, 0): [12, 21, -1, 33], |
| self.File(0, 1): [31, 49, 52], |
| self.File(100, 0): [113, 128], |
| self.File(200, 1): [132, 146], |
| self.File(300, 0): [19, 20, 32], |
| self.File(300, 1): [24, 39]}) |
| phaser = phased_orderfile.PhasedAnalyzer(mgr, Mod10Processor()) |
| opo = lambda s, c, i: phased_orderfile.OrderfilePhaseOffsets( |
| startup=s, common=c, interaction=i) |
| self.assertListEqual([opo([10, 20], [30], [40, 50]), |
| opo([110, 120], [], []), |
| opo([], [], [130, 140]), |
| opo([10], [20, 30], [])], |
| phaser._GetOrderfilePhaseOffsets()) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |
