Get web application list using Drive V2 API.

chrome/browser/chromeos/gdata/drive_webapps_registry_unittest.cc

 // Copyright (c) 2012 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 "base/file_path.h"
 #include "base/file_util.h"
 #include "base/json/json_file_value_serializer.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/string16.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
+#include "chrome/browser/chromeos/gdata/drive_api_parser.h"
 #include "chrome/browser/chromeos/gdata/drive_webapps_registry.h"
 #include "chrome/browser/chromeos/gdata/gdata_wapi_parser.h"
 #include "chrome/common/chrome_paths.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::Value;
 using base::DictionaryValue;
 using base::ListValue;
 
-#define IF_EXPECT_EQ(arg1, arg2) \
-  EXPECT_EQ(arg1, arg2); \
-  if (arg1 == arg2)
-
 namespace gdata {
 
 class DriveWebAppsRegistryTest : public testing::Test {
  protected:
   DriveWebAppsRegistryTest()
       : ui_thread_(content::BrowserThread::UI, &message_loop_){
   }
 
-  static Value* LoadJSONFile(const std::string& filename) {
+  static Value* LoadJSONFile(const std::string& prefix,
+                             const std::string& filename) {
     FilePath path;
     std::string error;
     // Test files for this unit test are located in
     // src/chrome/test/data/chromeos/gdata/*
     PathService::Get(chrome::DIR_TEST_DATA, &path);
-    path = path.AppendASCII("chromeos")
-        .AppendASCII("gdata")
-        .AppendASCII(filename.c_str());
+    path = path.AppendASCII("chromeos").AppendASCII(prefix)
+           .AppendASCII(filename);
     EXPECT_TRUE(file_util::PathExists(path)) <<
         "Couldn't find " << path.value();
 
     JSONFileValueSerializer serializer(path);
     Value* value = serializer.Deserialize(NULL, &error);
     EXPECT_TRUE(value) <<
         "Parse error " << path.value() << ": " << error;
     return value;
   }
 
@@ skipping 34 matching lines @@
               "Drive App 2", "Drive App Object 2",
               is_primary);
   }
 
  private:
   MessageLoopForUI message_loop_;
   content::TestBrowserThread ui_thread_;
 };
 
 TEST_F(DriveWebAppsRegistryTest, LoadAndFindWebApps) {
-  scoped_ptr<Value> document(LoadJSONFile("account_metadata.json"));
+  scoped_ptr<Value> document(LoadJSONFile("gdata", "account_metadata.json"));
   ASSERT_TRUE(document.get());
   ASSERT_TRUE(document->GetType() == Value::TYPE_DICTIONARY);
   DictionaryValue* entry_value;
   ASSERT_TRUE(reinterpret_cast<DictionaryValue*>(document.get())->GetDictionary(
       std::string("entry"), &entry_value));
   ASSERT_TRUE(entry_value);
 
   // Load feed.
   scoped_ptr<AccountMetadataFeed> feed(
       AccountMetadataFeed::CreateFrom(*document));
   ASSERT_TRUE(feed.get());
   scoped_ptr<DriveWebAppsRegistry> web_apps(new DriveWebAppsRegistry);
-  web_apps->UpdateFromFeed(feed.get());
+  web_apps->UpdateFromFeed(*feed.get());
 
   // Find by extension 'ext_1'.
   ScopedVector<DriveWebAppInfo> ext_1_results;
   FilePath ext1_file(FILE_PATH_LITERAL("gdata/SampleFile.ext_1"));
   web_apps->GetWebAppsForFile(ext1_file, std::string(), &ext_1_results);
-  IF_EXPECT_EQ(1U, ext_1_results.size()) {
-    EXPECT_TRUE(VerifyApp1(ext_1_results, true));
-  }
+  ASSERT_EQ(1U, ext_1_results.size());
+  EXPECT_TRUE(VerifyApp1(ext_1_results, true));
 
   // Find by extension 'ext_3'.
   ScopedVector<DriveWebAppInfo> ext_3_results;
   FilePath ext3_file(FILE_PATH_LITERAL("gdata/AnotherFile.ext_3"));
   web_apps->GetWebAppsForFile(ext3_file, std::string(), &ext_3_results);
-  IF_EXPECT_EQ(2U, ext_3_results.size()) {
-    EXPECT_TRUE(VerifyApp1(ext_3_results, false));
-    EXPECT_TRUE(VerifyApp2(ext_3_results, true));
-  }
+  ASSERT_EQ(2U, ext_3_results.size());
+  EXPECT_TRUE(VerifyApp1(ext_3_results, false));
+  EXPECT_TRUE(VerifyApp2(ext_3_results, true));
 
   // Find by mimetype 'ext_3'.
   ScopedVector<DriveWebAppInfo> mime_results;
   web_apps->GetWebAppsForFile(FilePath(), "application/test_type_2",
                               &mime_results);
-  IF_EXPECT_EQ(1U, mime_results.size()) {
-    EXPECT_TRUE(VerifyApp2(mime_results, true));
-  }
+  ASSERT_EQ(1U, mime_results.size());
+  EXPECT_TRUE(VerifyApp2(mime_results, true));
 
   // Find by extension and mimetype.
   ScopedVector<DriveWebAppInfo> mime_ext_results;
   FilePath mime_file(FILE_PATH_LITERAL("gdata/MimeFile.ext_2"));
   web_apps->GetWebAppsForFile(mime_file, "application/test_type_2",
                               &mime_ext_results);
-  IF_EXPECT_EQ(2U, mime_ext_results.size()) {
-    EXPECT_TRUE(VerifyApp1(mime_ext_results, true));
-    EXPECT_TRUE(VerifyApp2(mime_ext_results, true));
-  }
+  ASSERT_EQ(2U, mime_ext_results.size());
+  EXPECT_TRUE(VerifyApp1(mime_ext_results, true));
+  EXPECT_TRUE(VerifyApp2(mime_ext_results, true));
+}
+
+TEST_F(DriveWebAppsRegistryTest, LoadAndFindDriveWebApps) {
+  scoped_ptr<Value> document(LoadJSONFile("drive", "applist.json"));
+  ASSERT_TRUE(document.get());
+  ASSERT_TRUE(document->GetType() == Value::TYPE_DICTIONARY);
+
+  // Load feed.
+  scoped_ptr<AppList> app_list(AppList::CreateFrom(*document));
+  ASSERT_TRUE(app_list.get());
+  scoped_ptr<DriveWebAppsRegistry> web_apps(new DriveWebAppsRegistry);
+  web_apps->UpdateFromApplicationList(*app_list.get());
+
+  // Find by primary extension 'exe'.
+  ScopedVector<DriveWebAppInfo> ext_results;
+  FilePath ext_file(FILE_PATH_LITERAL("drive/file.exe"));
+  web_apps->GetWebAppsForFile(ext_file, std::string(), &ext_results);
+  ASSERT_EQ(1U, ext_results.size());
+  VerifyApp(ext_results, "abcdefghabcdefghabcdefghabcdefgh", "123456788192",
+            "Drive app 1", "", true);
+
+  // Find by primary MIME type.
+  ScopedVector<DriveWebAppInfo> primary_app;
+  web_apps->GetWebAppsForFile(FilePath(),
+      "application/vnd.google-apps.drive-sdk.123456788192", &primary_app);
+  ASSERT_EQ(1U, primary_app.size());
+  VerifyApp(primary_app, "abcdefghabcdefghabcdefghabcdefgh", "123456788192",
+            "Drive app 1", "", true);
+
+  // Find by secondary MIME type.
+  ScopedVector<DriveWebAppInfo> secondary_app;
+  web_apps->GetWebAppsForFile(FilePath(), "text/html", &secondary_app);
+  ASSERT_EQ(1U, secondary_app.size());
+  VerifyApp(secondary_app, "abcdefghabcdefghabcdefghabcdefgh", "123456788192",
+            "Drive app 1", "", false);
 }
 
 }  // namespace gdata