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

#ifndef DIAGNOSTICS_WILCO_DTC_SUPPORTD_FAKE_BROWSER_H_
#define DIAGNOSTICS_WILCO_DTC_SUPPORTD_FAKE_BROWSER_H_

#include <string>

#include <base/callback.h>
#include <base/macros.h>
#include <dbus/mock_exported_object.h>
#include <mojo/public/cpp/bindings/binding.h>

#include "diagnostics/wilco_dtc_supportd/mock_mojom_wilco_dtc_supportd_client.h"
#include "diagnostics/wilco_dtc_supportd/mojo_test_utils.h"
#include "mojo/diagnosticsd.mojom.h"

namespace diagnostics {

// Helper class that allows to test communication between the browser and the
// tested code of the wilco_dtc_supportd daemon.
class FakeBrowser final {
 public:
  using DBusMethodCallCallback = dbus::ExportedObject::MethodCallCallback;

  using MojomDiagnosticsdClient =
      chromeos::diagnosticsd::mojom::DiagnosticsdClient;
  using MojomDiagnosticsdClientPtr =
      chromeos::diagnosticsd::mojom::DiagnosticsdClientPtr;
  using MojomDiagnosticsdServicePtr =
      chromeos::diagnosticsd::mojom::DiagnosticsdServicePtr;
  using MojomDiagnosticsdServiceFactoryPtr =
      chromeos::diagnosticsd::mojom::DiagnosticsdServiceFactoryPtr;

  // |wilco_dtc_supportd_service_factory_ptr| is a pointer to the tested
  // DiagnosticsdServiceFactory instance.
  // |bootstrap_mojo_connection_dbus_method| is a fake substitute for the
  // BootstrapMojoConnection() D-Bus method.
  FakeBrowser(MojomDiagnosticsdServiceFactoryPtr*
                  wilco_dtc_supportd_service_factory_ptr,
              DBusMethodCallCallback bootstrap_mojo_connection_dbus_method);
  ~FakeBrowser();

  // Returns a mock DiagnosticsdClient instance, whose methods are invoked when
  // FakeBrowser receives incoming Mojo calls from the tested code.
  MockMojomWilcoDtcSupportdClient* wilco_dtc_supportd_client() {
    return &wilco_dtc_supportd_client_;
  }

  // Call the BootstrapMojoConnection D-Bus method. Returns whether the D-Bus
  // call returned success.
  // |mojo_fd| is the fake file descriptor.
  // |bootstrap_mojo_connection_dbus_method| is the callback that the tested
  // code exposed as the BootstrapMojoConnection D-Bus method.
  //
  // It's not allowed to call this method again after a successful completion.
  bool BootstrapMojoConnection(FakeMojoFdGenerator* fake_mojo_fd_generator);

  // Call the |SendUiMessageToDiagnosticsProcessor| Mojo method
  // on wilco_dtc_supportd daemon, which will call the |HandleMessageFromUi|
  // gRPC method on wilco_dtc.
  //
  // It simulates message sent from diagnostics UI extension to wilco_dtc.
  //
  // Returns false when we were not able to copy |json_message| into shared
  // buffer.
  //
  // Must be called only after a successful invocation of
  // BootstrapMojoConnection().
  bool SendUiMessageToDiagnosticsProcessor(
      const std::string& json_message,
      const base::Callback<void(mojo::ScopedHandle)>& callback);

 private:
  // Calls |bootstrap_mojo_connection_dbus_method_| with a fake file descriptor.
  // Returns whether the method call succeeded (it's expected to happen
  // synchronously).
  bool CallBootstrapMojoConnectionDBusMethod(
      FakeMojoFdGenerator* fake_mojo_fd_generator);
  // Calls GetService() Mojo method on
  // |wilco_dtc_supportd_service_factory_ptr_|, initializes
  // |wilco_dtc_supportd_service_ptr_| so that it points to the tested service,
  // registers |wilco_dtc_supportd_client_| to handle incoming Mojo requests.
  void CallGetServiceMojoMethod();

  // Unowned. Points to the tested DiagnosticsdServiceFactory instance.
  MojomDiagnosticsdServiceFactoryPtr* const
      wilco_dtc_supportd_service_factory_ptr_;
  // Fake substitute for the BootstrapMojoConnection() D-Bus method.
  DBusMethodCallCallback bootstrap_mojo_connection_dbus_method_;

  // Mock DiagnosticsdClient instance. After an invocation of
  // CallGetServiceMojoMethod() it becomes registered to receive incoming Mojo
  // requests from the tested code.
  MockMojomWilcoDtcSupportdClient wilco_dtc_supportd_client_;
  // Mojo binding that is associated with |wilco_dtc_supportd_client_|.
  mojo::Binding<MojomDiagnosticsdClient> wilco_dtc_supportd_client_binding_;

  // Mojo interface pointer to the DiagnosticsdService service exposed by the
  // tested code. Gets initialized after a call to CallGetServiceMojoMethod().
  MojomDiagnosticsdServicePtr wilco_dtc_supportd_service_ptr_;

  DISALLOW_COPY_AND_ASSIGN(FakeBrowser);
};

}  // namespace diagnostics

#endif  // DIAGNOSTICS_WILCO_DTC_SUPPORTD_FAKE_BROWSER_H_