Chromium Code Reviews| Index: chrome/browser/history/visit_tracker_unittest.cc |
| diff --git a/chrome/browser/history/visit_tracker_unittest.cc b/chrome/browser/history/visit_tracker_unittest.cc |
| index 5c84d28521b072b5248cc8fc425cde39cba3b819..21052a05ac208499fe80098ea632930daf19a0ae 100644 |
| --- a/chrome/browser/history/visit_tracker_unittest.cc |
| +++ b/chrome/browser/history/visit_tracker_unittest.cc |
| @@ -11,9 +11,8 @@ using history::VisitTracker; |
| namespace { |
| struct VisitToTest { |
| - // Identifies the host, we'll cast this to a pointer when querying (the |
| - // tracker isn't allowed to dereference this pointer). |
| - int host; |
| + // Identifies the context. |
| + ContextID context_id; |
| int32 page_id; |
| // Used when adding this to the tracker |
| @@ -30,25 +29,25 @@ struct VisitToTest { |
| // The tracker uses RenderProcessHost pointers for scoping but never |
| // dereferences them. We use ints because it's easier. This function converts |
| // between the two. |
| -void* MakeFakeHost(int id) { |
| - void* host = 0; |
| - memcpy(&host, &id, sizeof(int)); |
| - return host; |
| +static void* MakeFakeContextID(int id) { |
| + ContextID context_id = 0; |
|
sdefresne
2014/06/10 15:51:54
nit: same here, I think a reinterpret_cast would b
|
| + memcpy(&context_id, &id, sizeof(id)); |
| + return context_id; |
| } |
| void RunTest(VisitTracker* tracker, VisitToTest* test, int test_count) { |
| for (int i = 0; i < test_count; i++) { |
| // Our host pointer is actually just an int, convert it (it will not get |
| // dereferenced). |
| - void* host = MakeFakeHost(test[i].host); |
| + ContextID context_id = test[i].context_id; |
| // Check the referrer for this visit. |
| history::VisitID ref_visit = tracker->GetLastVisit( |
| - host, test[i].page_id, GURL(test[i].referrer)); |
| + context_id, test[i].page_id, GURL(test[i].referrer)); |
| EXPECT_EQ(test[i].referring_visit_id, ref_visit); |
| // Now add this visit. |
| - tracker->AddVisit(host, test[i].page_id, GURL(test[i].url), |
| + tracker->AddVisit(context_id, test[i].page_id, GURL(test[i].url), |
| test[i].visit_id); |
| } |
| } |
| @@ -60,11 +59,14 @@ void RunTest(VisitTracker* tracker, VisitToTest* test, int test_count) { |
| TEST(VisitTracker, SimpleTransitions) { |
| VisitToTest test_simple[] = { |
| // Started here: |
| - {1, 1, "http://www.google.com/", 1, "", 0}, |
| + {MakeFakeContextID(1), 1, "http://www.google.com/", 1, |
| + "", 0}, |
| // Clicked a link: |
| - {1, 2, "http://images.google.com/", 2, "http://www.google.com/", 1}, |
| + {MakeFakeContextID(1), 2, "http://images.google.com/", 2, |
| + "http://www.google.com/", 1}, |
| // Went back, then clicked a link: |
| - {1, 3, "http://video.google.com/", 3, "http://www.google.com/", 1}, |
| + {MakeFakeContextID(1), 3, "http://video.google.com/", 3, |
| + "http://www.google.com/", 1}, |
| }; |
| VisitTracker tracker; |
| @@ -76,17 +78,23 @@ TEST(VisitTracker, SimpleTransitions) { |
| TEST(VisitTracker, Frames) { |
| VisitToTest test_frames[] = { |
| // Started here: |
| - {1, 1, "http://foo.com/", 1, "", 0}, |
| + {MakeFakeContextID(1), 1, "http://foo.com/", 1, |
| + "", 0}, |
| // Which had an auto-loaded subframe: |
| - {1, 1, "http://foo.com/ad.html", 2, "http://foo.com/", 1}, |
| + {MakeFakeContextID(1), 1, "http://foo.com/ad.html", 2, |
| + "http://foo.com/", 1}, |
| // ...and another auto-loaded subframe: |
| - {1, 1, "http://foo.com/ad2.html", 3, "http://foo.com/", 1}, |
| + {MakeFakeContextID(1), 1, "http://foo.com/ad2.html", 3, |
| + "http://foo.com/", 1}, |
| // ...and the user navigated the first subframe to somwhere else |
| - {1, 2, "http://bar.com/", 4, "http://foo.com/ad.html", 2}, |
| + {MakeFakeContextID(1), 2, "http://bar.com/", 4, |
| + "http://foo.com/ad.html", 2}, |
| // ...and then the second subframe somewhere else |
| - {1, 3, "http://fud.com/", 5, "http://foo.com/ad2.html", 3}, |
| + {MakeFakeContextID(1), 3, "http://fud.com/", 5, |
| + "http://foo.com/ad2.html", 3}, |
| // ...and then the main frame somewhere else. |
| - {1, 4, "http://www.google.com/", 6, "http://foo.com/", 1}, |
| + {MakeFakeContextID(1), 4, "http://www.google.com/", 6, |
| + "http://foo.com/", 1}, |
| }; |
| VisitTracker tracker; |
| @@ -119,20 +127,23 @@ TEST(VisitTracker, MultiProcess) { |
| TEST(VisitTracker, ProcessRemove) { |
| // Simple navigation from one process. |
| VisitToTest part1[] = { |
| - {1, 1, "http://www.google.com/", 1, "", 0}, |
| - {1, 2, "http://images.google.com/", 2, "http://www.google.com/", 1}, |
| + {MakeFakeContextID(1), 1, "http://www.google.com/", 1, |
| + "", 0}, |
| + {MakeFakeContextID(1), 2, "http://images.google.com/", 2, |
| + "http://www.google.com/", 1}, |
| }; |
| VisitTracker tracker; |
| RunTest(&tracker, part1, arraysize(part1)); |
| - // Say that process has been destroyed. |
| - tracker.NotifyRenderProcessHostDestruction(MakeFakeHost(1)); |
| + // Say that context has been destroyed. |
| + tracker.ClearCachedDataForContextID(MakeFakeContextID(1)); |
| // Simple navigation from a new process with the same ID, it should not find |
| // a referrer. |
| VisitToTest part2[] = { |
| - {1, 1, "http://images.google.com/", 2, "http://www.google.com/", 0}, |
| + {MakeFakeContextID(1), 1, "http://images.google.com/", 2, |
| + "http://www.google.com/", 0}, |
| }; |
| RunTest(&tracker, part2, arraysize(part2)); |
| } |