1// Copyright (c) 2006-2008 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#include "chrome/browser/history/visit_tracker.h"
6#include "base/basictypes.h"
7#include "testing/gtest/include/gtest/gtest.h"
8
9using history::VisitTracker;
10
11namespace {
12
13struct VisitToTest {
14  // Identifies the host, we'll cast this to a pointer when querying (the
15  // tracker isn't allowed to dereference this pointer).
16  int host;
17  int32 page_id;
18
19  // Used when adding this to the tracker
20  const char* url;
21  const history::VisitID visit_id;
22
23  // Used when finding the referrer
24  const char* referrer;
25
26  // the correct referring visit ID to compare to the computed one
27  history::VisitID referring_visit_id;
28};
29
30// The tracker uses RenderProcessHost pointers for scoping but never
31// dereferences them. We use ints because it's easier. This function converts
32// between the two.
33void* MakeFakeHost(int id) {
34  void* host = 0;
35  memcpy(&host, &id, sizeof(int));
36  return host;
37}
38
39void RunTest(VisitTracker* tracker, VisitToTest* test, int test_count) {
40  for (int i = 0; i < test_count; i++) {
41    // Our host pointer is actually just an int, convert it (it will not get
42    // dereferenced).
43    void* host = MakeFakeHost(test[i].host);
44
45    // Check the referrer for this visit.
46    history::VisitID ref_visit = tracker->GetLastVisit(
47        host, test[i].page_id, GURL(test[i].referrer));
48    EXPECT_EQ(test[i].referring_visit_id, ref_visit);
49
50    // Now add this visit.
51    tracker->AddVisit(host, test[i].page_id, GURL(test[i].url),
52                      test[i].visit_id);
53  }
54}
55
56}  // namespace
57
58// A simple test that makes sure we transition between main pages in the
59// presence of back/forward.
60TEST(VisitTracker, SimpleTransitions) {
61  VisitToTest test_simple[] = {
62    // Started here:
63    {1, 1, "http://www.google.com/",    1, "",                       0},
64    // Clicked a link:
65    {1, 2, "http://images.google.com/", 2, "http://www.google.com/", 1},
66    // Went back, then clicked a link:
67    {1, 3, "http://video.google.com/",  3, "http://www.google.com/", 1},
68  };
69
70  VisitTracker tracker;
71  RunTest(&tracker, test_simple, arraysize(test_simple));
72}
73
74// Test that referrer is properly computed when there are different frame
75// navigations happening.
76TEST(VisitTracker, Frames) {
77  VisitToTest test_frames[] = {
78    // Started here:
79    {1, 1, "http://foo.com/",           1, "",                        0},
80    // Which had an auto-loaded subframe:
81    {1, 1, "http://foo.com/ad.html",    2, "http://foo.com/",         1},
82    // ...and another auto-loaded subframe:
83    {1, 1, "http://foo.com/ad2.html",   3, "http://foo.com/",         1},
84    // ...and the user navigated the first subframe to somwhere else
85    {1, 2, "http://bar.com/",           4, "http://foo.com/ad.html",  2},
86    // ...and then the second subframe somewhere else
87    {1, 3, "http://fud.com/",           5, "http://foo.com/ad2.html", 3},
88    // ...and then the main frame somewhere else.
89    {1, 4, "http://www.google.com/",    6, "http://foo.com/",         1},
90  };
91
92  VisitTracker tracker;
93  RunTest(&tracker, test_frames, arraysize(test_frames));
94}
95
96// Test frame navigation to make sure that the referrer is properly computed
97// when there are multiple processes navigating the same pages.
98TEST(VisitTracker, MultiProcess) {
99  VisitToTest test_processes[] = {
100    // Process 1 and 2 start here:
101    {1, 1, "http://foo.com/",           1, "",                       0},
102    {2, 1, "http://foo.com/",           2, "",                       0},
103    // They have some subframes:
104    {1, 1, "http://foo.com/ad.html",    3, "http://foo.com/",        1},
105    {2, 1, "http://foo.com/ad.html",    4, "http://foo.com/",        2},
106    // Subframes are navigated:
107    {1, 2, "http://bar.com/",           5, "http://foo.com/ad.html", 3},
108    {2, 2, "http://bar.com/",           6, "http://foo.com/ad.html", 4},
109    // Main frame is navigated:
110    {1, 3, "http://www.google.com/",    7, "http://foo.com/",        1},
111    {2, 3, "http://www.google.com/",    8, "http://foo.com/",        2},
112  };
113
114  VisitTracker tracker;
115  RunTest(&tracker, test_processes, arraysize(test_processes));
116}
117
118// Test that processes get removed properly.
119TEST(VisitTracker, ProcessRemove) {
120  // Simple navigation from one process.
121  VisitToTest part1[] = {
122    {1, 1, "http://www.google.com/",    1, "",                       0},
123    {1, 2, "http://images.google.com/", 2, "http://www.google.com/", 1},
124  };
125
126  VisitTracker tracker;
127  RunTest(&tracker, part1, arraysize(part1));
128
129  // Say that process has been destroyed.
130  tracker.NotifyRenderProcessHostDestruction(MakeFakeHost(1));
131
132  // Simple navigation from a new process with the same ID, it should not find
133  // a referrer.
134  VisitToTest part2[] = {
135    {1, 1, "http://images.google.com/", 2, "http://www.google.com/", 0},
136  };
137  RunTest(&tracker, part2, arraysize(part2));
138}
139