Chromium Code Reviews| Index: components/printing/service/service_main.cc |
| diff --git a/components/printing/service/service_main.cc b/components/printing/service/service_main.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..87324e751a735d2cf33a1ac3130af99d1c28c62c |
| --- /dev/null |
| +++ b/components/printing/service/service_main.cc |
| @@ -0,0 +1,31 @@ |
| +// Copyright 2017 The Chromium Authors. All rights reserved. |
|
Ken Rockot(use gerrit already)
2017/07/25 05:02:06
nit: maybe test_service_main.cc would make more se
Wei Li
2017/07/26 05:08:50
More meaningful, done.
|
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "base/test/test_discardable_memory_allocator.h" |
| +#include "components/printing/service/pdf_compositor_service.h" |
| +#include "services/service_manager/public/c/main.h" |
| +#include "services/service_manager/public/cpp/service_runner.h" |
| + |
| +// In order to test PdfCompositorService, this class overrides and |
| +// uses a test discardable memory allocator. |
| +class PdfCompositorTestService : public printing::PdfCompositorService { |
| + public: |
| + explicit PdfCompositorTestService(const std::string& creator) |
| + : PdfCompositorService(creator) {} |
| + ~PdfCompositorTestService() override {} |
| + |
| + // PdfCompositorService: |
| + void PrepareToStart() override; |
| +}; |
| + |
| +void PdfCompositorTestService::PrepareToStart() { |
| + base::TestDiscardableMemoryAllocator mem_allocator; |
| + base::DiscardableMemoryAllocator::SetInstance(&mem_allocator); |
| +} |
| + |
| +MojoResult ServiceMain(MojoHandle service_request_handle) { |
| + service_manager::ServiceRunner runner( |
| + new PdfCompositorTestService("pdf_compositor_service_unittest")); |
| + return runner.Run(service_request_handle); |
| +} |