extensions/browser/api/sockets_tcp/sockets_tcp_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/memory/scoped_ptr.h"
   6 #include "base/values.h"
   7 #include "content/public/test/test_browser_context.h"
   8 #include "extensions/browser/api/api_resource_manager.h"
   9 #include "extensions/browser/api/socket/socket.h"
  10 #include "extensions/browser/api/socket/tcp_socket.h"
  11 #include "extensions/browser/api/sockets_tcp/sockets_tcp_api.h"
  12 #include "extensions/browser/api_unittest.h"
  13 #include "testing/gmock/include/gmock/gmock.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15
  16 namespace extensions {
  17 namespace core_api {
  18
  19 static scoped_ptr<KeyedService> ApiResourceManagerTestFactory(
  20     content::BrowserContext* context) {
  21   content::BrowserThread::ID id;
  22   CHECK(content::BrowserThread::GetCurrentThreadIdentifier(&id));
  23   return ApiResourceManager<
  24       ResumableTCPSocket>::CreateApiResourceManagerForTest(context, id);
  25 }
  26
  27 class SocketsTcpUnitTest : public ApiUnitTest {
  28  public:
  29   void SetUp() override {
  30     ApiUnitTest::SetUp();
  31
  32     ApiResourceManager<ResumableTCPSocket>::GetFactoryInstance()
  33         ->SetTestingFactoryAndUse(browser_context(),
  34                                   ApiResourceManagerTestFactory);
  35   }
  36 };
  37
  38 TEST_F(SocketsTcpUnitTest, Create) {
  39   // Get BrowserThread
  40   content::BrowserThread::ID id;
  41   CHECK(content::BrowserThread::GetCurrentThreadIdentifier(&id));
  42
  43   // Create SocketCreateFunction and put it on BrowserThread
  44   SocketsTcpCreateFunction* function = new SocketsTcpCreateFunction();
  45   function->set_work_thread_id(id);
  46
  47   // Run tests
  48   scoped_ptr<base::DictionaryValue> result(RunFunctionAndReturnDictionary(
  49       function, "[{\"persistent\": true, \"name\": \"foo\"}]"));
  50   ASSERT_TRUE(result.get());
  51 }
  52
  53 }  // namespace core_api
  54 }  // namespace extensions