// Copyright 2019 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.

#include <base/run_loop.h>
#include <brillo/message_loops/base_message_loop.h>
#include <gtest/gtest.h>
#include <mojo/core/core.h>
#include <mojo/core/embedder/embedder.h>
#include <mojo/public/cpp/bindings/binding.h>
#include <mojo/public/cpp/bindings/interface_request.h>

#include "midis/client.h"
#include "midis/device_tracker.h"
#include "mojo/midis.mojom.h"

// Local implementation of the mojo MidisClient interface.
class ClientImpl : public arc::mojom::MidisClient {
 public:
  ~ClientImpl() override{};
  void OnDeviceAdded(arc::mojom::MidisDeviceInfoPtr device) override {}

  void OnDeviceRemoved(arc::mojom::MidisDeviceInfoPtr device) override {}

  void BindClientPtr(arc::mojom::MidisClientPtr* ptr) {
    mojo::InterfaceRequest<arc::mojom::MidisClient> request =
        mojo::MakeRequest<arc::mojom::MidisClient>(ptr);
    binding_ = std::make_unique<mojo::Binding<arc::mojom::MidisClient>>(
        this, std::move(request));
  }

 private:
  std::unique_ptr<mojo::Binding<arc::mojom::MidisClient>> binding_;
};

namespace midis {

class ClientTest : public ::testing::Test {
 public:
  ClientTest() = default;
  ClientTest(const ClientTest&) = delete;
  ClientTest& operator=(const ClientTest&) = delete;

  ~ClientTest() override = default;

 protected:
  void SetUp() override {
    message_loop_.SetAsCurrent();
    mojo::core::Init();
  }

  void TearDown() override {
    auto core = mojo::core::Core::Get();
    std::vector<MojoHandle> leaks;
    core->GetActiveHandlesForTest(&leaks);
    EXPECT_TRUE(leaks.empty());
  }

 private:
  brillo::BaseMessageLoop message_loop_;
};

// Check that the MidisServer implementation sends back the correct
// number of devices.
TEST_F(ClientTest, ListDevices) {
  DeviceTracker tracker;
  arc::mojom::MidisServerPtr server;

  ClientImpl client;
  arc::mojom::MidisClientPtr ptr;
  client.BindClientPtr(&ptr);

  Client client_class(&tracker, 0, base::Bind([](uint32_t client_id) {}),
                      mojo::MakeRequest(&server), std::move(ptr));

  // Check that initially there are no devices listed.
  int64_t num_devices = -1;
  server->ListDevices(base::Bind(
      [](int64_t* num_devices,
         std::vector<arc::mojom::MidisDeviceInfoPtr> devices) {
        *num_devices = devices.size();
      },
      &num_devices));
  base::RunLoop().RunUntilIdle();
  EXPECT_EQ(num_devices, 0);

  // TODO(b/122623049): Add a device, then check that ListDevices works as
  // expected.
}

}  // namespace midis