First implementation of web intents table.

chrome/browser/webdata/web_intents_table_unittest.cc

Index: chrome/browser/webdata/web_intents_table_unittest.cc
diff --git a/chrome/browser/webdata/web_intents_table_unittest.cc b/chrome/browser/webdata/web_intents_table_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..d96513840ec20e5f4c111cb849421dd57c7c09c9
--- /dev/null
+++ b/chrome/browser/webdata/web_intents_table_unittest.cc
@@ -0,0 +1,94 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include <algorithm>
+
+#include "base/file_util.h"
+#include "base/scoped_temp_dir.h"
+#include "base/string16.h"
+#include "base/utf_string_conversions.h"
+#include "chrome/browser/webdata/web_database.h"
+#include "chrome/browser/webdata/web_intents_table.h"
+#include "chrome/common/chrome_paths.h"
+#include "googleurl/src/gurl.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::Time;
+
+namespace {
+
+GURL test_url("http://google.com/");
+string16 test_action = ASCIIToUTF16("http://webintents.org/intents/share");
+string16 mime_image = ASCIIToUTF16("image/*");
+string16 mime_video = ASCIIToUTF16("video/*");
+
+class WebIntentsTableTest : public testing::Test {
+ protected:
+  virtual void SetUp() {
+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+    ASSERT_EQ(sql::INIT_OK,
+              db_.Init(temp_dir_.path().AppendASCII("TestWebDatabase.db")));
+  }
+
+  WebIntentsTable* IntentsTable() {
+    return db_.GetWebIntentsTable();
+  }
+
+  WebDatabase db_;
+  ScopedTempDir temp_dir_;
+};
+
+// Test we can add, retrieve, and remove intents from the database.
+TEST_F(WebIntentsTableTest, SetGetDeleteIntent) {
+  std::vector<WebIntentData> intents;
+
+  // By default, no intents exist.
+  EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents));
+  EXPECT_EQ(0U, intents.size());
+
+  // Now adding one.
+  EXPECT_TRUE(IntentsTable()->SetWebIntent(test_action, mime_image, test_url));
+
+  // Make sure that intent can now be fetched
+  EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents));
+  ASSERT_EQ(1U, intents.size());
+
+  EXPECT_EQ(test_url.spec(), intents[0].service_url.spec());
+  EXPECT_EQ(test_action, intents[0].action);
+  EXPECT_EQ(mime_image, intents[0].type);
+
+  // Remove the intent.
+  EXPECT_TRUE(IntentsTable()->RemoveWebIntent(test_action, mime_image,
+      test_url));
+
+  // Intent should now be gone.
+  intents.clear();
+  EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents));
+  EXPECT_EQ(0U, intents.size());
+}
+
+// Test we support multiple intents for the same MIME type
+TEST_F(WebIntentsTableTest, SetMultipleIntents) {
+  std::vector<WebIntentData> intents;
+
+  EXPECT_TRUE(IntentsTable()->SetWebIntent(test_action, mime_image, test_url));
+  EXPECT_TRUE(IntentsTable()->SetWebIntent(test_action, mime_video, test_url));
+
+  // Recover stored intents from DB.
+  EXPECT_TRUE(IntentsTable()->GetWebIntents(test_action, &intents));
+  ASSERT_EQ(2U, intents.size());
+
+  // WebIntentsTable does not guarantee order, so ensure order here.
+  if (intents[0].type == mime_video)
+    std::swap(intents[0], intents[1]);
+
+  EXPECT_EQ(test_url, intents[0].service_url);
+  EXPECT_EQ(test_action, intents[0].action);
+  EXPECT_EQ(mime_image, intents[0].type);
+
+  EXPECT_EQ(test_url, intents[1].service_url);
+  EXPECT_EQ(test_action, intents[1].action);
+  EXPECT_EQ(mime_video, intents[1].type);
+}
+} // namespace