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

#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/functional/callback_helpers.h"
#include "base/run_loop.h"
#include "base/task/sequenced_task_runner.h"
#include "base/test/task_environment.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "mojo/public/cpp/bindings/self_owned_receiver.h"
#include "services/service_manager/public/cpp/binder_registry.h"
#include "services/service_manager/public/cpp/connector.h"
#include "services/service_manager/public/cpp/service.h"
#include "services/service_manager/public/cpp/service_receiver.h"
#include "services/service_manager/public/cpp/test/test_connector_factory.h"
#include "services/service_manager/tests/test_support.test-mojom.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace service_manager {

namespace {

// TestBImpl and TestCImpl are simple test interfaces whose methods invokes
// their callback when called without doing anything.
class TestBImpl : public mojom::TestB {
 public:
  TestBImpl() = default;

  TestBImpl(const TestBImpl&) = delete;
  TestBImpl& operator=(const TestBImpl&) = delete;

  ~TestBImpl() override = default;

 private:
  // TestB:
  void B(BCallback callback) override { std::move(callback).Run(); }
  void CallC(CallCCallback callback) override { std::move(callback).Run(); }
};

class TestCImpl : public mojom::TestC {
 public:
  TestCImpl() = default;

  TestCImpl(const TestCImpl&) = delete;
  TestCImpl& operator=(const TestCImpl&) = delete;

  ~TestCImpl() override = default;

 private:
  // TestC:
  void C(CCallback callback) override { std::move(callback).Run(); }
};

void OnTestBReceiver(mojo::PendingReceiver<mojom::TestB> receiver) {
  mojo::MakeSelfOwnedReceiver(std::make_unique<TestBImpl>(),
                              std::move(receiver));
}

void OnTestCReceiver(mojo::PendingReceiver<mojom::TestC> receiver) {
  mojo::MakeSelfOwnedReceiver(std::make_unique<TestCImpl>(),
                              std::move(receiver));
}

class TestBServiceImpl : public Service {
 public:
  explicit TestBServiceImpl(mojo::PendingReceiver<mojom::Service> receiver)
      : service_receiver_(this, std::move(receiver)) {
    registry_.AddInterface(base::BindRepeating(&OnTestBReceiver));
  }

  TestBServiceImpl(const TestBServiceImpl&) = delete;
  TestBServiceImpl& operator=(const TestBServiceImpl&) = delete;

  ~TestBServiceImpl() override = default;

 private:
  // Service:
  void OnBindInterface(const BindSourceInfo& source_info,
                       const std::string& interface_name,
                       mojo::ScopedMessagePipeHandle interface_pipe) override {
    registry_.BindInterface(interface_name, std::move(interface_pipe));
  }

  service_manager::ServiceReceiver service_receiver_;
  service_manager::BinderRegistry registry_;
};

class TestCServiceImpl : public Service {
 public:
  explicit TestCServiceImpl(mojo::PendingReceiver<mojom::Service> receiver)
      : service_receiver_(this, std::move(receiver)) {
    registry_.AddInterface(base::BindRepeating(&OnTestCReceiver));
  }

  TestCServiceImpl(const TestCServiceImpl&) = delete;
  TestCServiceImpl& operator=(const TestCServiceImpl&) = delete;

  ~TestCServiceImpl() override = default;

 private:
  // Service:
  void OnBindInterface(const BindSourceInfo& source_info,
                       const std::string& interface_name,
                       mojo::ScopedMessagePipeHandle interface_pipe) override {
    registry_.BindInterface(interface_name, std::move(interface_pipe));
  }

  service_manager::ServiceReceiver service_receiver_;
  service_manager::BinderRegistry registry_;
};

constexpr char kServiceBName[] = "ServiceB";
constexpr char kServiceCName[] = "ServiceC";

}  // namespace

TEST(ServiceManagerTestSupport, TestConnectorFactoryUniqueService) {
  base::test::TaskEnvironment task_environment;

  TestConnectorFactory factory;
  TestCServiceImpl c_service(factory.RegisterInstance(kServiceCName));
  auto* connector = factory.GetDefaultConnector();

  mojo::Remote<mojom::TestC> c;
  connector->Connect(kServiceCName, c.BindNewPipeAndPassReceiver());
  base::RunLoop loop;
  c->C(loop.QuitClosure());
  loop.Run();

  // Give the service a chance to process disconnection and clean up.
  c.reset();
  base::RunLoop cleanup_loop;
  base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
      FROM_HERE, cleanup_loop.QuitClosure());
  cleanup_loop.Run();
}

TEST(ServiceManagerTestSupport, TestConnectorFactoryMultipleServices) {
  base::test::TaskEnvironment task_environment;

  TestConnectorFactory factory;
  TestBServiceImpl b_service(factory.RegisterInstance(kServiceBName));
  TestCServiceImpl c_service(factory.RegisterInstance(kServiceCName));
  auto* connector = factory.GetDefaultConnector();

  {
    mojo::Remote<mojom::TestB> b;
    connector->Connect(kServiceBName, b.BindNewPipeAndPassReceiver());
    base::RunLoop loop;
    b->B(loop.QuitClosure());
    loop.Run();
  }

  {
    mojo::Remote<mojom::TestC> c;
    connector->Connect(kServiceCName, c.BindNewPipeAndPassReceiver());
    base::RunLoop loop;
    c->C(loop.QuitClosure());
    loop.Run();
  }

  // Give the services a chance to process disconnection and clean up.
  base::RunLoop cleanup_loop;
  base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
      FROM_HERE, cleanup_loop.QuitClosure());
  cleanup_loop.Run();
}

}  // namespace service_manager