| Index: extensions/browser/api/serial/serial_apitest.cc
|
| diff --git a/extensions/browser/api/serial/serial_apitest.cc b/extensions/browser/api/serial/serial_apitest.cc
|
| index a756b897e67b4ec3aeb89dd54eb80eada083181e..5306508075c15f2e52aa51eae0cca2629465883a 100644
|
| --- a/extensions/browser/api/serial/serial_apitest.cc
|
| +++ b/extensions/browser/api/serial/serial_apitest.cc
|
| @@ -11,11 +11,11 @@
|
| #include "device/serial/test_serial_io_handler.h"
|
| #include "extensions/browser/api/serial/serial_api.h"
|
| #include "extensions/browser/api/serial/serial_connection.h"
|
| +#include "extensions/browser/api/serial/serial_service_factory.h"
|
| #include "extensions/browser/extension_function.h"
|
| #include "extensions/common/api/serial.h"
|
| #include "extensions/common/switches.h"
|
| #include "extensions/test/result_catcher.h"
|
| -#include "extensions/test/test_service_registration_manager.h"
|
| #include "testing/gmock/include/gmock/gmock.h"
|
|
|
| using testing::_;
|
| @@ -109,12 +109,16 @@ class SerialApiTest : public ExtensionApiTest,
|
| ExtensionApiTest::SetUpCommandLine(command_line);
|
| if (GetParam())
|
| command_line->AppendSwitch(switches::kEnableMojoSerialService);
|
| - test_service_registration_manager_.reset(
|
| - new TestServiceRegistrationManager);
|
| + }
|
| +
|
| + void TearDownOnMainThread() override {
|
| + SetSerialServiceFactoryForTest(nullptr);
|
| + ExtensionApiTest::TearDownOnMainThread();
|
| }
|
|
|
| protected:
|
| - scoped_ptr<TestServiceRegistrationManager> test_service_registration_manager_;
|
| + base::Callback<void(mojo::InterfaceRequest<device::serial::SerialService>)>
|
| + serial_service_factory_;
|
| };
|
|
|
| ExtensionFunction* FakeSerialGetDevicesFunctionFactory() {
|
| @@ -175,8 +179,8 @@ IN_PROC_BROWSER_TEST_P(SerialApiTest, SerialFakeHardware) {
|
|
|
| #if SIMULATE_SERIAL_PORTS
|
| if (GetParam()) {
|
| - test_service_registration_manager_->OverrideServiceFactoryForTest(
|
| - base::Bind(&CreateTestSerialService));
|
| + serial_service_factory_ = base::Bind(&CreateTestSerialService);
|
| + SetSerialServiceFactoryForTest(&serial_service_factory_);
|
| } else {
|
| ASSERT_TRUE(ExtensionFunctionDispatcher::OverrideFunction(
|
| "serial.getDevices", FakeSerialGetDevicesFunctionFactory));
|
|
|