| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <set> | 5 #include <set> |
| 6 #include <vector> | 6 #include <vector> |
| 7 | 7 |
| 8 #include "base/command_line.h" | 8 #include "base/command_line.h" |
| 9 #include "base/file_path.h" | 9 #include "base/file_path.h" |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| (...skipping 682 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 693 history::SOURCE_SYNCED); | 693 history::SOURCE_SYNCED); |
| 694 | 694 |
| 695 // Fetch the row information about the url from history db. | 695 // Fetch the row information about the url from history db. |
| 696 VisitVector visits; | 696 VisitVector visits; |
| 697 URLID row_id = backend_->db_->GetRowForURL(url, NULL); | 697 URLID row_id = backend_->db_->GetRowForURL(url, NULL); |
| 698 backend_->db_->GetVisitsForURL(row_id, &visits); | 698 backend_->db_->GetVisitsForURL(row_id, &visits); |
| 699 | 699 |
| 700 // Check if all the visits to the url are stored in database. | 700 // Check if all the visits to the url are stored in database. |
| 701 ASSERT_EQ(3U, visits.size()); | 701 ASSERT_EQ(3U, visits.size()); |
| 702 VisitSourceMap visit_sources; | 702 VisitSourceMap visit_sources; |
| 703 backend_->db_->GetVisitsSource(visits, &visit_sources); | 703 ASSERT_TRUE(backend_->GetVisitsSource(visits, &visit_sources)); |
| 704 ASSERT_EQ(3U, visit_sources.size()); | 704 ASSERT_EQ(3U, visit_sources.size()); |
| 705 int sources = 0; | 705 int sources = 0; |
| 706 for (int i = 0; i < 3; i++) { | 706 for (int i = 0; i < 3; i++) { |
| 707 switch (visit_sources[visits[i].visit_id]) { | 707 switch (visit_sources[visits[i].visit_id]) { |
| 708 case history::SOURCE_EXTENSION: | 708 case history::SOURCE_EXTENSION: |
| 709 sources |= 0x1; | 709 sources |= 0x1; |
| 710 break; | 710 break; |
| 711 case history::SOURCE_FIREFOX_IMPORTED: | 711 case history::SOURCE_FIREFOX_IMPORTED: |
| 712 sources |= 0x2; | 712 sources |= 0x2; |
| 713 break; | 713 break; |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 747 history::SOURCE_BROWSED, false)); | 747 history::SOURCE_BROWSED, false)); |
| 748 backend_->AddPage(request3); | 748 backend_->AddPage(request3); |
| 749 | 749 |
| 750 // Three visits should be added with proper sources. | 750 // Three visits should be added with proper sources. |
| 751 VisitVector visits; | 751 VisitVector visits; |
| 752 URLRow row; | 752 URLRow row; |
| 753 URLID id = backend_->db()->GetRowForURL(url, &row); | 753 URLID id = backend_->db()->GetRowForURL(url, &row); |
| 754 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); | 754 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); |
| 755 ASSERT_EQ(3U, visits.size()); | 755 ASSERT_EQ(3U, visits.size()); |
| 756 VisitSourceMap visit_sources; | 756 VisitSourceMap visit_sources; |
| 757 backend_->db_->GetVisitsSource(visits, &visit_sources); | 757 ASSERT_TRUE(backend_->GetVisitsSource(visits, &visit_sources)); |
| 758 ASSERT_EQ(1U, visit_sources.size()); | 758 ASSERT_EQ(1U, visit_sources.size()); |
| 759 EXPECT_EQ(history::SOURCE_SYNCED, visit_sources.begin()->second); | 759 EXPECT_EQ(history::SOURCE_SYNCED, visit_sources.begin()->second); |
| 760 } | 760 } |
| 761 | 761 |
| 762 TEST_F(HistoryBackendTest, AddVisitsSource) { | 762 TEST_F(HistoryBackendTest, AddVisitsSource) { |
| 763 ASSERT_TRUE(backend_.get()); | 763 ASSERT_TRUE(backend_.get()); |
| 764 | 764 |
| 765 GURL url1("http://www.cnn.com"); | 765 GURL url1("http://www.cnn.com"); |
| 766 std::vector<VisitInfo> visits1, visits2; | 766 std::vector<VisitInfo> visits1, visits2; |
| 767 visits1.push_back(VisitInfo( | 767 visits1.push_back(VisitInfo( |
| (...skipping 18 matching lines...) Expand all Loading... |
| 786 backend_->AddVisits(url1, visits1, history::SOURCE_IE_IMPORTED); | 786 backend_->AddVisits(url1, visits1, history::SOURCE_IE_IMPORTED); |
| 787 backend_->AddVisits(url2, visits2, history::SOURCE_SYNCED); | 787 backend_->AddVisits(url2, visits2, history::SOURCE_SYNCED); |
| 788 | 788 |
| 789 // Verify the visits were added with their sources. | 789 // Verify the visits were added with their sources. |
| 790 VisitVector visits; | 790 VisitVector visits; |
| 791 URLRow row; | 791 URLRow row; |
| 792 URLID id = backend_->db()->GetRowForURL(url1, &row); | 792 URLID id = backend_->db()->GetRowForURL(url1, &row); |
| 793 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); | 793 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); |
| 794 ASSERT_EQ(3U, visits.size()); | 794 ASSERT_EQ(3U, visits.size()); |
| 795 VisitSourceMap visit_sources; | 795 VisitSourceMap visit_sources; |
| 796 backend_->db_->GetVisitsSource(visits, &visit_sources); | 796 ASSERT_TRUE(backend_->GetVisitsSource(visits, &visit_sources)); |
| 797 ASSERT_EQ(3U, visit_sources.size()); | 797 ASSERT_EQ(3U, visit_sources.size()); |
| 798 for (int i = 0; i < 3; i++) | 798 for (int i = 0; i < 3; i++) |
| 799 EXPECT_EQ(history::SOURCE_IE_IMPORTED, visit_sources[visits[i].visit_id]); | 799 EXPECT_EQ(history::SOURCE_IE_IMPORTED, visit_sources[visits[i].visit_id]); |
| 800 id = backend_->db()->GetRowForURL(url2, &row); | 800 id = backend_->db()->GetRowForURL(url2, &row); |
| 801 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); | 801 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); |
| 802 ASSERT_EQ(2U, visits.size()); | 802 ASSERT_EQ(2U, visits.size()); |
| 803 backend_->db_->GetVisitsSource(visits, &visit_sources); | 803 ASSERT_TRUE(backend_->GetVisitsSource(visits, &visit_sources)); |
| 804 ASSERT_EQ(2U, visit_sources.size()); | 804 ASSERT_EQ(2U, visit_sources.size()); |
| 805 for (int i = 0; i < 2; i++) | 805 for (int i = 0; i < 2; i++) |
| 806 EXPECT_EQ(history::SOURCE_SYNCED, visit_sources[visits[i].visit_id]); | 806 EXPECT_EQ(history::SOURCE_SYNCED, visit_sources[visits[i].visit_id]); |
| 807 } | 807 } |
| 808 | 808 |
| 809 TEST_F(HistoryBackendTest, GetMostRecentVisits) { | 809 TEST_F(HistoryBackendTest, GetMostRecentVisits) { |
| 810 ASSERT_TRUE(backend_.get()); | 810 ASSERT_TRUE(backend_.get()); |
| 811 | 811 |
| 812 GURL url1("http://www.cnn.com"); | 812 GURL url1("http://www.cnn.com"); |
| 813 std::vector<VisitInfo> visits1; | 813 std::vector<VisitInfo> visits1; |
| (...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 918 VisitVector visits; | 918 VisitVector visits; |
| 919 URLRow row; | 919 URLRow row; |
| 920 URLID id = backend_->db()->GetRowForURL(url1, &row); | 920 URLID id = backend_->db()->GetRowForURL(url1, &row); |
| 921 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); | 921 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); |
| 922 ASSERT_EQ(2U, visits.size()); | 922 ASSERT_EQ(2U, visits.size()); |
| 923 // Remove these visits. | 923 // Remove these visits. |
| 924 ASSERT_TRUE(backend_->RemoveVisits(visits)); | 924 ASSERT_TRUE(backend_->RemoveVisits(visits)); |
| 925 | 925 |
| 926 // Now check only url2's source in visit_source table. | 926 // Now check only url2's source in visit_source table. |
| 927 VisitSourceMap visit_sources; | 927 VisitSourceMap visit_sources; |
| 928 backend_->db_->GetVisitsSource(visits, &visit_sources); | 928 ASSERT_TRUE(backend_->GetVisitsSource(visits, &visit_sources)); |
| 929 ASSERT_EQ(0U, visit_sources.size()); | 929 ASSERT_EQ(0U, visit_sources.size()); |
| 930 id = backend_->db()->GetRowForURL(url2, &row); | 930 id = backend_->db()->GetRowForURL(url2, &row); |
| 931 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); | 931 ASSERT_TRUE(backend_->db()->GetVisitsForURL(id, &visits)); |
| 932 ASSERT_EQ(2U, visits.size()); | 932 ASSERT_EQ(2U, visits.size()); |
| 933 backend_->db_->GetVisitsSource(visits, &visit_sources); | 933 ASSERT_TRUE(backend_->GetVisitsSource(visits, &visit_sources)); |
| 934 ASSERT_EQ(2U, visit_sources.size()); | 934 ASSERT_EQ(2U, visit_sources.size()); |
| 935 for (int i = 0; i < 2; i++) | 935 for (int i = 0; i < 2; i++) |
| 936 EXPECT_EQ(history::SOURCE_SYNCED, visit_sources[visits[i].visit_id]); | 936 EXPECT_EQ(history::SOURCE_SYNCED, visit_sources[visits[i].visit_id]); |
| 937 } | 937 } |
| 938 | 938 |
| 939 // Test for migration of adding visit_source table. | 939 // Test for migration of adding visit_source table. |
| 940 TEST_F(HistoryBackendTest, MigrationVisitSource) { | 940 TEST_F(HistoryBackendTest, MigrationVisitSource) { |
| 941 ASSERT_TRUE(backend_.get()); | 941 ASSERT_TRUE(backend_.get()); |
| 942 backend_->Closing(); | 942 backend_->Closing(); |
| 943 backend_ = NULL; | 943 backend_ = NULL; |
| (...skipping 184 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1128 ASSERT_TRUE(backend_->GetFaviconFromDB(url, TOUCH_ICON, &favicon)); | 1128 ASSERT_TRUE(backend_->GetFaviconFromDB(url, TOUCH_ICON, &favicon)); |
| 1129 std::string touchicon_data( | 1129 std::string touchicon_data( |
| 1130 favicon.image_data->front(), | 1130 favicon.image_data->front(), |
| 1131 favicon.image_data->front() + favicon.image_data->size()); | 1131 favicon.image_data->front() + favicon.image_data->size()); |
| 1132 | 1132 |
| 1133 EXPECT_EQ(TOUCH_ICON, favicon.icon_type); | 1133 EXPECT_EQ(TOUCH_ICON, favicon.icon_type); |
| 1134 EXPECT_EQ(icon_url, favicon.icon_url); | 1134 EXPECT_EQ(icon_url, favicon.icon_url); |
| 1135 EXPECT_EQ(blob_data, touchicon_data); | 1135 EXPECT_EQ(blob_data, touchicon_data); |
| 1136 } | 1136 } |
| 1137 } // namespace history | 1137 } // namespace history |
| OLD | NEW |