chrome/common/extensions/extension_file_util_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/common/extensions/extension_file_util.h"
   6
   7 #include <set>
   8 #include <string>
   9
  10 #include "base/path_service.h"
  11 #include "chrome/common/chrome_paths.h"
  12 #include "extensions/common/extension.h"
  13 #include "extensions/common/file_util.h"
  14 #include "extensions/common/manifest.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 namespace extensions {
  18
  19 typedef testing::Test ExtensionFileUtilTest;
  20
  21 // Test that a browser action extension returns a path to an icon.
  22 TEST_F(ExtensionFileUtilTest, GetBrowserImagePaths) {
  23   base::FilePath install_dir;
  24   ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &install_dir));
  25   install_dir = install_dir.AppendASCII("extensions")
  26                     .AppendASCII("api_test")
  27                     .AppendASCII("browser_action")
  28                     .AppendASCII("basics");
  29
  30   std::string error;
  31   scoped_refptr<Extension> extension(file_util::LoadExtension(
  32       install_dir, Manifest::UNPACKED, Extension::NO_FLAGS, &error));
  33   ASSERT_TRUE(extension.get());
  34
  35   // The extension contains one icon.
  36   std::set<base::FilePath> paths =
  37       extension_file_util::GetBrowserImagePaths(extension.get());
  38   ASSERT_EQ(1u, paths.size());
  39   EXPECT_EQ("icon.png", paths.begin()->BaseName().AsUTF8Unsafe());
  40 }
  41
  42 // Test that extensions with zero-length action icons will not load.
  43 TEST_F(ExtensionFileUtilTest, CheckZeroLengthActionIconFiles) {
  44   base::FilePath install_dir;
  45   ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &install_dir));
  46
  47   // Try to install an extension with a zero-length browser action icon file.
  48   base::FilePath ext_dir = install_dir.AppendASCII("extensions")
  49                                .AppendASCII("bad")
  50                                .AppendASCII("Extensions")
  51                                .AppendASCII("gggggggggggggggggggggggggggggggg");
  52
  53   std::string error;
  54   scoped_refptr<Extension> extension2(file_util::LoadExtension(
  55       ext_dir, Manifest::UNPACKED, Extension::NO_FLAGS, &error));
  56   EXPECT_FALSE(extension2.get());
  57   EXPECT_STREQ("Could not load icon 'icon.png' for browser action.",
  58                error.c_str());
  59
  60   // Try to install an extension with a zero-length page action icon file.
  61   ext_dir = install_dir.AppendASCII("extensions")
  62                 .AppendASCII("bad")
  63                 .AppendASCII("Extensions")
  64                 .AppendASCII("hhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhh");
  65
  66   scoped_refptr<Extension> extension3(file_util::LoadExtension(
  67       ext_dir, Manifest::UNPACKED, Extension::NO_FLAGS, &error));
  68   EXPECT_FALSE(extension3.get());
  69   EXPECT_STREQ("Could not load icon 'icon.png' for page action.",
  70                error.c_str());
  71 }
  72
  73 }  // namespace extensions