| OLD | NEW |
| (Empty) |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import re | |
| 6 from StringIO import StringIO | |
| 7 import unittest | |
| 8 | |
| 9 import resource_sack | |
| 10 import resource_sack_display | |
| 11 from test_utils import (MakeRequest, | |
| 12 TestResourceGraph) | |
| 13 | |
| 14 | |
| 15 class ResourceSackDispayTestCase(unittest.TestCase): | |
| 16 def test_SimpleOutput(self): | |
| 17 g1 = TestResourceGraph.FromRequestList([ | |
| 18 MakeRequest(0, 'null'), | |
| 19 MakeRequest(1, 0), | |
| 20 MakeRequest(2, 0), | |
| 21 MakeRequest(3, 1)]) | |
| 22 g2 = TestResourceGraph.FromRequestList([ | |
| 23 MakeRequest(0, 'null'), | |
| 24 MakeRequest(1, 0), | |
| 25 MakeRequest(2, 0), | |
| 26 MakeRequest(4, 2)]) | |
| 27 sack = resource_sack.GraphSack() | |
| 28 sack.ConsumeGraph(g1) | |
| 29 sack.ConsumeGraph(g2) | |
| 30 buf = StringIO() | |
| 31 resource_sack_display.ToDot(sack, buf, | |
| 32 long_edge_msec=1000) | |
| 33 dot = buf.getvalue() | |
| 34 # Short edge. | |
| 35 self.assertTrue(re.search(r'0 -> 1[^]]+color=green \]', dot, re.MULTILINE)) | |
| 36 # Long edge. | |
| 37 self.assertTrue(re.search(r'0 -> 3[^]]+penwidth=8', dot)) | |
| 38 | |
| 39 | |
| 40 | |
| 41 if __name__ == '__main__': | |
| 42 unittest.main() | |
| OLD | NEW |