chromium/chrome/browser/extensions/api/printing/fake_print_job_controller.h

// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_EXTENSIONS_API_PRINTING_FAKE_PRINT_JOB_CONTROLLER_H_
#define CHROME_BROWSER_EXTENSIONS_API_PRINTING_FAKE_PRINT_JOB_CONTROLLER_H_

#include "base/memory/weak_ptr.h"
#include "chrome/browser/printing/print_job_controller.h"

namespace extensions {

// Fake print job controller which doesn't send print jobs to actual printing
// pipeline.
// It's used in unit and API integration tests.
class FakePrintJobController : public printing::PrintJobController {
 public:
  FakePrintJobController();
  ~FakePrintJobController() override;

  // If `fail_` is set, call OnFailed() instead of OnDocDone().
  void set_fail(bool fail) { fail_ = fail; }

  // PrintJobController:
  void CreatePrintJob(std::unique_ptr<printing::MetafileSkia> pdf,
                      std::unique_ptr<printing::PrintSettings> settings,
                      uint32_t page_count,
                      crosapi::mojom::PrintJob::Source source,
                      const std::string& source_id,
                      PrintJobCreatedCallback callback) override;

 private:
  void CreatePrintJobImpl(scoped_refptr<printing::PrintJob> job,
                          std::unique_ptr<printing::PrintSettings> settings);

  bool fail_ = false;
  // Current job id.
  int job_id_ = 0;

  base::WeakPtrFactory<FakePrintJobController> weak_ptr_factory_{this};
};

}  // namespace extensions

#endif  // CHROME_BROWSER_EXTENSIONS_API_PRINTING_FAKE_PRINT_JOB_CONTROLLER_H_