| 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 <algorithm> | 5 #include <algorithm> |
| 6 | 6 |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/scoped_temp_dir.h" | 8 #include "base/scoped_temp_dir.h" |
| 9 #include "base/string16.h" | 9 #include "base/string16.h" |
| 10 #include "base/utf_string_conversions.h" | 10 #include "base/utf_string_conversions.h" |
| 11 #include "chrome/browser/intents/web_intent_data.h" | 11 #include "chrome/browser/intents/web_intent_service_data.h" |
| 12 #include "chrome/browser/webdata/web_database.h" | 12 #include "chrome/browser/webdata/web_database.h" |
| 13 #include "chrome/browser/webdata/web_intents_table.h" | 13 #include "chrome/browser/webdata/web_intents_table.h" |
| 14 #include "chrome/common/chrome_paths.h" | 14 #include "chrome/common/chrome_paths.h" |
| 15 #include "googleurl/src/gurl.h" | 15 #include "googleurl/src/gurl.h" |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 17 | 17 |
| 18 namespace { | 18 namespace { |
| 19 | 19 |
| 20 GURL test_url("http://google.com/"); | 20 GURL test_url("http://google.com/"); |
| 21 string16 test_action = ASCIIToUTF16("http://webintents.org/intents/share"); | 21 string16 test_action = ASCIIToUTF16("http://webintents.org/intents/share"); |
| 22 string16 test_action_2 = ASCIIToUTF16("http://webintents.org/intents/view"); | 22 string16 test_action_2 = ASCIIToUTF16("http://webintents.org/intents/view"); |
| 23 string16 test_title = ASCIIToUTF16("Test WebIntent"); | 23 string16 test_title = ASCIIToUTF16("Test WebIntent"); |
| 24 string16 test_title_2 = ASCIIToUTF16("Test WebIntent #2"); | 24 string16 test_title_2 = ASCIIToUTF16("Test WebIntent #2"); |
| 25 string16 mime_image = ASCIIToUTF16("image/*"); | 25 string16 mime_image = ASCIIToUTF16("image/*"); |
| 26 string16 mime_video = ASCIIToUTF16("video/*"); | 26 string16 mime_video = ASCIIToUTF16("video/*"); |
| 27 | 27 |
| 28 WebIntentData MakeIntent(const GURL& url, const string16& action, | 28 WebIntentServiceData MakeIntent(const GURL& url, const string16& action, |
| 29 const string16& type, const string16& title) { | 29 const string16& type, const string16& title) { |
| 30 WebIntentData intent; | 30 WebIntentServiceData service; |
| 31 intent.service_url = url; | 31 service.service_url = url; |
| 32 intent.action = action; | 32 service.action = action; |
| 33 intent.type = type; | 33 service.type = type; |
| 34 intent.title = title; | 34 service.title = title; |
| 35 intent.disposition = WebIntentData::DISPOSITION_INLINE; | 35 service.disposition = WebIntentServiceData::DISPOSITION_INLINE; |
| 36 return intent; | 36 return service; |
| 37 } | 37 } |
| 38 | 38 |
| 39 class WebIntentsTableTest : public testing::Test { | 39 class WebIntentsTableTest : public testing::Test { |
| 40 protected: | 40 protected: |
| 41 virtual void SetUp() { | 41 virtual void SetUp() { |
| 42 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 42 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 43 ASSERT_EQ(sql::INIT_OK, | 43 ASSERT_EQ(sql::INIT_OK, |
| 44 db_.Init(temp_dir_.path().AppendASCII("TestWebDatabase.db"))); | 44 db_.Init(temp_dir_.path().AppendASCII("TestWebDatabase.db"))); |
| 45 } | 45 } |
| 46 | 46 |
| 47 WebIntentsTable* IntentsTable() { | 47 WebIntentsTable* IntentsTable() { |
| 48 return db_.GetWebIntentsTable(); | 48 return db_.GetWebIntentsTable(); |
| 49 } | 49 } |
| 50 | 50 |
| 51 WebDatabase db_; | 51 WebDatabase db_; |
| 52 ScopedTempDir temp_dir_; | 52 ScopedTempDir temp_dir_; |
| 53 }; | 53 }; |
| 54 | 54 |
| 55 // Test we can add, retrieve, and remove intents from the database. | 55 // Test we can add, retrieve, and remove intents from the database. |
| 56 TEST_F(WebIntentsTableTest, SetGetDeleteIntent) { | 56 TEST_F(WebIntentsTableTest, SetGetDeleteIntent) { |
| 57 std::vector<WebIntentData> intents; | 57 std::vector<WebIntentServiceData> intents; |
| 58 | 58 |
| 59 // By default, no intents exist. | 59 // By default, no intents exist. |
| 60 EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents)); | 60 EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents)); |
| 61 EXPECT_EQ(0U, intents.size()); | 61 EXPECT_EQ(0U, intents.size()); |
| 62 | 62 |
| 63 // Now adding one. | 63 // Now adding one. |
| 64 WebIntentData intent = MakeIntent(test_url, test_action, mime_image, | 64 WebIntentServiceData intent = MakeIntent(test_url, test_action, mime_image, |
| 65 test_title); | 65 test_title); |
| 66 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); | 66 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); |
| 67 | 67 |
| 68 // Make sure that intent can now be fetched | 68 // Make sure that intent can now be fetched |
| 69 EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents)); | 69 EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents)); |
| 70 ASSERT_EQ(1U, intents.size()); | 70 ASSERT_EQ(1U, intents.size()); |
| 71 EXPECT_EQ(intent, intents[0]); | 71 EXPECT_EQ(intent, intents[0]); |
| 72 | 72 |
| 73 // Remove the intent. | 73 // Remove the intent. |
| 74 EXPECT_TRUE(IntentsTable()->RemoveWebIntent(intent)); | 74 EXPECT_TRUE(IntentsTable()->RemoveWebIntent(intent)); |
| 75 | 75 |
| 76 // Intent should now be gone. | 76 // Intent should now be gone. |
| 77 intents.clear(); | 77 intents.clear(); |
| 78 EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents)); | 78 EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents)); |
| 79 EXPECT_EQ(0U, intents.size()); | 79 EXPECT_EQ(0U, intents.size()); |
| 80 } | 80 } |
| 81 | 81 |
| 82 // Test we support multiple intents for the same MIME type | 82 // Test we support multiple intents for the same MIME type |
| 83 TEST_F(WebIntentsTableTest, SetMultipleIntents) { | 83 TEST_F(WebIntentsTableTest, SetMultipleIntents) { |
| 84 std::vector<WebIntentData> intents; | 84 std::vector<WebIntentServiceData> intents; |
| 85 | 85 |
| 86 WebIntentData intent = MakeIntent(test_url, test_action, mime_image, | 86 WebIntentServiceData intent = MakeIntent(test_url, test_action, mime_image, |
| 87 test_title); | 87 test_title); |
| 88 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); | 88 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); |
| 89 | 89 |
| 90 intent.type = mime_video; | 90 intent.type = mime_video; |
| 91 intent.title = test_title_2; | 91 intent.title = test_title_2; |
| 92 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); | 92 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); |
| 93 | 93 |
| 94 // Recover stored intents from DB. | 94 // Recover stored intents from DB. |
| 95 EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents)); | 95 EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents)); |
| 96 ASSERT_EQ(2U, intents.size()); | 96 ASSERT_EQ(2U, intents.size()); |
| 97 | 97 |
| 98 // WebIntentsTable does not guarantee order, so ensure order here. | 98 // WebIntentsTable does not guarantee order, so ensure order here. |
| 99 if (intents[0].type == mime_video) | 99 if (intents[0].type == mime_video) |
| 100 std::swap(intents[0], intents[1]); | 100 std::swap(intents[0], intents[1]); |
| 101 | 101 |
| 102 EXPECT_EQ(intent, intents[1]); | 102 EXPECT_EQ(intent, intents[1]); |
| 103 | 103 |
| 104 intent.type = mime_image; | 104 intent.type = mime_image; |
| 105 intent.title = test_title; | 105 intent.title = test_title; |
| 106 EXPECT_EQ(intent, intents[0]); | 106 EXPECT_EQ(intent, intents[0]); |
| 107 } | 107 } |
| 108 | 108 |
| 109 // Test we support getting all intents independent of action. | 109 // Test we support getting all intents independent of action. |
| 110 TEST_F(WebIntentsTableTest, GetAllIntents) { | 110 TEST_F(WebIntentsTableTest, GetAllIntents) { |
| 111 std::vector<WebIntentData> intents; | 111 std::vector<WebIntentServiceData> intents; |
| 112 | 112 |
| 113 WebIntentData intent = MakeIntent(test_url, test_action, mime_image, | 113 WebIntentServiceData intent = MakeIntent(test_url, test_action, mime_image, |
| 114 test_title); | 114 test_title); |
| 115 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); | 115 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); |
| 116 | 116 |
| 117 intent.action = test_action_2; | 117 intent.action = test_action_2; |
| 118 intent.title = test_title_2; | 118 intent.title = test_title_2; |
| 119 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); | 119 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); |
| 120 | 120 |
| 121 // Recover stored intents from DB. | 121 // Recover stored intents from DB. |
| 122 EXPECT_TRUE(IntentsTable()->GetAllWebIntents(&intents)); | 122 EXPECT_TRUE(IntentsTable()->GetAllWebIntents(&intents)); |
| 123 ASSERT_EQ(2U, intents.size()); | 123 ASSERT_EQ(2U, intents.size()); |
| 124 | 124 |
| 125 // WebIntentsTable does not guarantee order, so ensure order here. | 125 // WebIntentsTable does not guarantee order, so ensure order here. |
| 126 if (intents[0].type == test_action_2) | 126 if (intents[0].type == test_action_2) |
| 127 std::swap(intents[0], intents[1]); | 127 std::swap(intents[0], intents[1]); |
| 128 | 128 |
| 129 EXPECT_EQ(intent, intents[1]); | 129 EXPECT_EQ(intent, intents[1]); |
| 130 | 130 |
| 131 intent.action = test_action; | 131 intent.action = test_action; |
| 132 intent.title = test_title; | 132 intent.title = test_title; |
| 133 EXPECT_EQ(intent, intents[0]); | 133 EXPECT_EQ(intent, intents[0]); |
| 134 } | 134 } |
| 135 | 135 |
| 136 TEST_F(WebIntentsTableTest, DispositionToStringMapping) { | 136 TEST_F(WebIntentsTableTest, DispositionToStringMapping) { |
| 137 WebIntentData intent = MakeIntent(test_url, test_action, mime_image, | 137 WebIntentServiceData intent = MakeIntent(test_url, test_action, mime_image, |
| 138 test_title); | 138 test_title); |
| 139 intent.disposition = WebIntentData::DISPOSITION_WINDOW; | 139 intent.disposition = WebIntentServiceData::DISPOSITION_WINDOW; |
| 140 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); | 140 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); |
| 141 | 141 |
| 142 intent = MakeIntent(test_url, test_action, mime_video, | 142 intent = MakeIntent(test_url, test_action, mime_video, |
| 143 test_title); | 143 test_title); |
| 144 intent.disposition = WebIntentData::DISPOSITION_INLINE; | 144 intent.disposition = WebIntentServiceData::DISPOSITION_INLINE; |
| 145 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); | 145 EXPECT_TRUE(IntentsTable()->SetWebIntent(intent)); |
| 146 | 146 |
| 147 std::vector<WebIntentData> intents; | 147 std::vector<WebIntentServiceData> intents; |
| 148 EXPECT_TRUE(IntentsTable()->GetAllWebIntents(&intents)); | 148 EXPECT_TRUE(IntentsTable()->GetAllWebIntents(&intents)); |
| 149 ASSERT_EQ(2U, intents.size()); | 149 ASSERT_EQ(2U, intents.size()); |
| 150 | 150 |
| 151 if (intents[0].disposition == WebIntentData::DISPOSITION_WINDOW) | 151 if (intents[0].disposition == WebIntentServiceData::DISPOSITION_WINDOW) |
| 152 std::swap(intents[0], intents[1]); | 152 std::swap(intents[0], intents[1]); |
| 153 | 153 |
| 154 EXPECT_EQ(WebIntentData::DISPOSITION_INLINE, intents[0].disposition); | 154 EXPECT_EQ(WebIntentServiceData::DISPOSITION_INLINE, intents[0].disposition); |
| 155 EXPECT_EQ(WebIntentData::DISPOSITION_WINDOW, intents[1].disposition); | 155 EXPECT_EQ(WebIntentServiceData::DISPOSITION_WINDOW, intents[1].disposition); |
| 156 } | 156 } |
| 157 } // namespace | 157 } // namespace |
| OLD | NEW |