src/chrome/browser/extensions/api/sockets_tcp_server/sockets_tcp_server_api_unittest.cc

   1 // Copyright 2013 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 "base/values.h"
   6 #include "chrome/browser/browser_process_impl.h"
   7 #include "chrome/browser/extensions/extension_api_unittest.h"
   8 #include "chrome/browser/extensions/test_extension_system.h"
   9 #include "chrome/browser/profiles/profile_manager.h"
  10 #include "chrome/test/base/testing_browser_process.h"
  11 #include "extensions/browser/api/api_resource_manager.h"
  12 #include "extensions/browser/api/socket/socket.h"
  13 #include "extensions/browser/api/socket/tcp_socket.h"
  14 #include "extensions/browser/api/sockets_tcp_server/sockets_tcp_server_api.h"
  15 #include "testing/gmock/include/gmock/gmock.h"
  16 #include "testing/gtest/include/gtest/gtest.h"
  17
  18 namespace extensions {
  19 namespace core_api {
  20
  21 static KeyedService* ApiResourceManagerTestFactory(
  22     content::BrowserContext* context) {
  23   content::BrowserThread::ID id;
  24   CHECK(content::BrowserThread::GetCurrentThreadIdentifier(&id));
  25   return ApiResourceManager<
  26       ResumableTCPSocket>::CreateApiResourceManagerForTest(context, id);
  27 }
  28
  29 static KeyedService* ApiResourceManagerTestServerFactory(
  30     content::BrowserContext* context) {
  31   content::BrowserThread::ID id;
  32   CHECK(content::BrowserThread::GetCurrentThreadIdentifier(&id));
  33   return ApiResourceManager<
  34       ResumableTCPServerSocket>::CreateApiResourceManagerForTest(context, id);
  35 }
  36
  37 class SocketsTcpServerUnitTest : public ExtensionApiUnittest {
  38  public:
  39   void SetUp() override {
  40     ExtensionApiUnittest::SetUp();
  41
  42     ApiResourceManager<ResumableTCPSocket>::GetFactoryInstance()
  43         ->SetTestingFactoryAndUse(browser()->profile(),
  44                                   ApiResourceManagerTestFactory);
  45
  46     ApiResourceManager<ResumableTCPServerSocket>::GetFactoryInstance()
  47         ->SetTestingFactoryAndUse(browser()->profile(),
  48                                   ApiResourceManagerTestServerFactory);
  49   }
  50 };
  51
  52 TEST_F(SocketsTcpServerUnitTest, Create) {
  53   // Get BrowserThread
  54   content::BrowserThread::ID id;
  55   CHECK(content::BrowserThread::GetCurrentThreadIdentifier(&id));
  56
  57   // Create SocketCreateFunction and put it on BrowserThread
  58   SocketsTcpServerCreateFunction* function =
  59       new SocketsTcpServerCreateFunction();
  60   function->set_work_thread_id(id);
  61
  62   // Run tests
  63   scoped_ptr<base::DictionaryValue> result(RunFunctionAndReturnDictionary(
  64       function, "[{\"persistent\": true, \"name\": \"foo\"}]"));
  65   ASSERT_TRUE(result.get());
  66 }
  67
  68 }  // namespace core_api
  69 }  // namespace extensions