Index: tools/telemetry/telemetry/page/page_unittest.py |
diff --git a/tools/telemetry/telemetry/page/page_unittest.py b/tools/telemetry/telemetry/page/page_unittest.py |
index c60c59df5fb742be0d1fe8751b66d4c15951b04e..b392e55b457d90a402787dc4cfa3ef992fe6af59 100644 |
--- a/tools/telemetry/telemetry/page/page_unittest.py |
+++ b/tools/telemetry/telemetry/page/page_unittest.py |
@@ -137,3 +137,19 @@ class TestPage(unittest.TestCase): |
'url': 'http://example.com/', |
'name': 'Example' |
}, named_dict) |
+ |
+ def testTransferToPageSet(self): |
+ page_set_a = page_set.PageSet() |
+ page_set_b = page_set.PageSet() |
+ page_foo = page.Page('http://foo.com', page_set_a) |
+ page_bar = page.Page('http://bar.com', page_set_a) |
+ page_baz = page.Page('http://baz.com', page_set_a) |
+ |
+ page_set_a.AddPage(page_foo) |
+ page_set_a.AddPage(page_bar) |
+ page_set_a.AddPage(page_baz) |
+ |
+ page_bar.TransferToPageSet(page_set_b) |
+ self.assertEqual([page_foo, page_baz], page_set_a.pages) |
+ self.assertEqual([page_bar], page_set_b.pages) |
+ self.assertIs(page_set_b, page_bar.page_set) |