Index: tools/android/loading/test_utils.py |
diff --git a/tools/android/loading/test_utils.py b/tools/android/loading/test_utils.py |
index a221385a000835ee5e2e76a8a70e5e81983bf416..8e0782650b1613765b16e057175d11b750b5024a 100644 |
--- a/tools/android/loading/test_utils.py |
+++ b/tools/android/loading/test_utils.py |
@@ -4,6 +4,7 @@ |
"""Common utilities used in unit tests, within this directory.""" |
+import dependency_graph |
import devtools_monitor |
import loading_model |
import loading_trace |
@@ -166,6 +167,13 @@ class TestResourceGraph(loading_model.ResourceGraph): |
return cls(LoadingTraceFromEvents(requests, page_events, trace_events)) |
+class TestDependencyGraph(dependency_graph.RequestDependencyGraph): |
+ """A dependency graph created from requests using a simple lens.""" |
+ def __init__(self, requests): |
+ lens = SimpleLens(LoadingTraceFromEvents(requests)) |
+ super(TestDependencyGraph, self).__init__(requests, lens) |
+ |
+ |
class MockConnection(object): |
"""Mock out connection for testing. |