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 |