src/extensions/browser/api_unittest.h

   1 // Copyright 2014 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 #ifndef EXTENSIONS_BROWSER_API_UNITTEST_H_
   6 #define EXTENSIONS_BROWSER_API_UNITTEST_H_
   7
   8 #include <string>
   9
  10 #include "base/memory/ref_counted.h"
  11 #include "base/memory/scoped_ptr.h"
  12 #include "base/prefs/testing_pref_service.h"
  13 #include "extensions/browser/extensions_test.h"
  14 #include "extensions/browser/mock_extension_system.h"
  15
  16 namespace base {
  17 class Value;
  18 class DictionaryValue;
  19 class ListValue;
  20 }
  21
  22 namespace content {
  23 class NotificationService;
  24 class TestBrowserThreadBundle;
  25 }
  26
  27 class UIThreadExtensionFunction;
  28
  29 namespace extensions {
  30
  31 // Use this class to enable calling API functions in a unittest.
  32 // By default, this class will create and load an empty unpacked |extension_|,
  33 // which will be used in all API function calls. This extension can be
  34 // overridden using set_extension().
  35 // When calling RunFunction[AndReturn*], |args| should be in JSON format,
  36 // wrapped in a list. See also RunFunction* in api_test_utils.h.
  37 class ApiUnitTest : public ExtensionsTest {
  38  public:
  39   ApiUnitTest();
  40   ~ApiUnitTest() override;
  41
  42   const Extension* extension() const { return extension_.get(); }
  43   scoped_refptr<Extension> extension_ref() { return extension_; }
  44   void set_extension(scoped_refptr<Extension> extension) {
  45     extension_ = extension;
  46   }
  47
  48  protected:
  49   // SetUp creates and loads an empty, unpacked Extension.
  50   void SetUp() override;
  51
  52   // Various ways of running an API function. These methods take ownership of
  53   // |function|. |args| should be in JSON format, wrapped in a list.
  54   // See also the RunFunction* methods in extension_function_test_utils.h.
  55
  56   // Return the function result as a base::Value.
  57   scoped_ptr<base::Value> RunFunctionAndReturnValue(
  58       UIThreadExtensionFunction* function,
  59       const std::string& args);
  60
  61   // Return the function result as a base::DictionaryValue, or NULL.
  62   // This will EXPECT-fail if the result is not a DictionaryValue.
  63   scoped_ptr<base::DictionaryValue> RunFunctionAndReturnDictionary(
  64       UIThreadExtensionFunction* function,
  65       const std::string& args);
  66
  67   // Return the function result as a base::ListValue, or NULL.
  68   // This will EXPECT-fail if the result is not a ListValue.
  69   scoped_ptr<base::ListValue> RunFunctionAndReturnList(
  70       UIThreadExtensionFunction* function,
  71       const std::string& args);
  72
  73   // Return an error thrown from the function, if one exists.
  74   // This will EXPECT-fail if any result is returned from the function.
  75   std::string RunFunctionAndReturnError(UIThreadExtensionFunction* function,
  76                                         const std::string& args);
  77
  78   // Run the function and ignore any result.
  79   void RunFunction(UIThreadExtensionFunction* function,
  80                    const std::string& args);
  81
  82  private:
  83   scoped_ptr<content::NotificationService> notification_service_;
  84
  85   scoped_ptr<content::TestBrowserThreadBundle> thread_bundle_;
  86   TestingPrefServiceSimple testing_pref_service_;
  87
  88   MockExtensionSystemFactory<MockExtensionSystem> extension_system_factory_;
  89
  90   // The Extension used when running API function calls.
  91   scoped_refptr<Extension> extension_;
  92 };
  93
  94 }  // namespace extensions
  95
  96 #endif  // EXTENSIONS_BROWSER_API_UNITTEST_H_