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)); |
} |