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::ContextID;
10using history::VisitTracker;
11
12namespace {
13
14struct VisitToTest {
15  // Identifies the context.
16  int context_id_int;
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
30void RunTest(VisitTracker* tracker, VisitToTest* test, int test_count) {
31  for (int i = 0; i < test_count; i++) {
32    // Our host pointer is actually just an int, convert it (it will not get
33    // dereferenced).
34    ContextID context_id = reinterpret_cast<ContextID>(test[i].context_id_int);
35
36    // Check the referrer for this visit.
37    history::VisitID ref_visit = tracker->GetLastVisit(
38        context_id, test[i].page_id, GURL(test[i].referrer));
39    EXPECT_EQ(test[i].referring_visit_id, ref_visit);
40
41    // Now add this visit.
42    tracker->AddVisit(
43        context_id, test[i].page_id, GURL(test[i].url), test[i].visit_id);
44  }
45}
46
47}  // namespace
48
49// A simple test that makes sure we transition between main pages in the
50// presence of back/forward.
51TEST(VisitTracker, SimpleTransitions) {
52  VisitToTest test_simple[] = {
53      // Started here:
54      {1, 1, "http://www.google.com/", 1, "", 0},
55      // Clicked a link:
56      {1, 2, "http://images.google.com/", 2, "http://www.google.com/", 1},
57      // Went back, then clicked a link:
58      {1, 3, "http://video.google.com/", 3, "http://www.google.com/", 1},
59  };
60
61  VisitTracker tracker;
62  RunTest(&tracker, test_simple, arraysize(test_simple));
63}
64
65// Test that referrer is properly computed when there are different frame
66// navigations happening.
67TEST(VisitTracker, Frames) {
68  VisitToTest test_frames[] = {
69      // Started here:
70      {1, 1, "http://foo.com/", 1, "", 0},
71      // Which had an auto-loaded subframe:
72      {1, 1, "http://foo.com/ad.html", 2, "http://foo.com/", 1},
73      // ...and another auto-loaded subframe:
74      {1, 1, "http://foo.com/ad2.html", 3, "http://foo.com/", 1},
75      // ...and the user navigated the first subframe to somwhere else
76      {1, 2, "http://bar.com/", 4, "http://foo.com/ad.html", 2},
77      // ...and then the second subframe somewhere else
78      {1, 3, "http://fud.com/", 5, "http://foo.com/ad2.html", 3},
79      // ...and then the main frame somewhere else.
80      {1, 4, "http://www.google.com/", 6, "http://foo.com/", 1},
81  };
82
83  VisitTracker tracker;
84  RunTest(&tracker, test_frames, arraysize(test_frames));
85}
86
87// Test frame navigation to make sure that the referrer is properly computed
88// when there are multiple processes navigating the same pages.
89TEST(VisitTracker, MultiProcess) {
90  VisitToTest test_processes[] = {
91    // Process 1 and 2 start here:
92    {1, 1, "http://foo.com/",           1, "",                       0},
93    {2, 1, "http://foo.com/",           2, "",                       0},
94    // They have some subframes:
95    {1, 1, "http://foo.com/ad.html",    3, "http://foo.com/",        1},
96    {2, 1, "http://foo.com/ad.html",    4, "http://foo.com/",        2},
97    // Subframes are navigated:
98    {1, 2, "http://bar.com/",           5, "http://foo.com/ad.html", 3},
99    {2, 2, "http://bar.com/",           6, "http://foo.com/ad.html", 4},
100    // Main frame is navigated:
101    {1, 3, "http://www.google.com/",    7, "http://foo.com/",        1},
102    {2, 3, "http://www.google.com/",    8, "http://foo.com/",        2},
103  };
104
105  VisitTracker tracker;
106  RunTest(&tracker, test_processes, arraysize(test_processes));
107}
108
109// Test that processes get removed properly.
110TEST(VisitTracker, ProcessRemove) {
111  // Simple navigation from one process.
112  VisitToTest part1[] = {
113      {1, 1, "http://www.google.com/", 1, "", 0},
114      {1, 2, "http://images.google.com/", 2, "http://www.google.com/", 1},
115  };
116
117  VisitTracker tracker;
118  RunTest(&tracker, part1, arraysize(part1));
119
120  // Say that context has been invalidated.
121  tracker.ClearCachedDataForContextID(reinterpret_cast<ContextID>(1));
122
123  // Simple navigation from a new process with the same ID, it should not find
124  // a referrer.
125  VisitToTest part2[] = {
126      {1, 1, "http://images.google.com/", 2, "http://www.google.com/", 0},
127  };
128  RunTest(&tracker, part2, arraysize(part2));
129}
130