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 |