chromeos/services/secure_channel/fake_active_connection_manager.h

// Copyright 2018 The Chromium 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 CHROMEOS_SERVICES_SECURE_CHANNEL_FAKE_ACTIVE_CONNECTION_MANAGER_H_
#define CHROMEOS_SERVICES_SECURE_CHANNEL_FAKE_ACTIVE_CONNECTION_MANAGER_H_

#include <memory>
#include <string>
#include <tuple>
#include <vector>

#include "base/containers/flat_map.h"
#include "base/macros.h"
#include "chromeos/services/secure_channel/active_connection_manager.h"
#include "chromeos/services/secure_channel/client_connection_parameters.h"
#include "chromeos/services/secure_channel/connection_details.h"
#include "chromeos/services/secure_channel/public/mojom/secure_channel.mojom.h"

namespace chromeos {

namespace secure_channel {

class AuthenticatedChannel;

// Test ActiveConnectionManager implementation.
class FakeActiveConnectionManager : public ActiveConnectionManager {
 public:
  FakeActiveConnectionManager(ActiveConnectionManager::Delegate* delegate);
  ~FakeActiveConnectionManager() override;

  using DetailsToMetadataMap = base::flat_map<
      ConnectionDetails,
      std::tuple<ConnectionState,
                 std::unique_ptr<AuthenticatedChannel>,
                 std::vector<std::unique_ptr<ClientConnectionParameters>>>>;

  DetailsToMetadataMap& connection_details_to_active_metadata_map() {
    return connection_details_to_active_metadata_map_;
  }

  void SetDisconnecting(const ConnectionDetails& connection_details);
  void SetDisconnected(const ConnectionDetails& connection_details);

 private:
  // ActiveConnectionManager:
  ConnectionState GetConnectionState(
      const ConnectionDetails& connection_details) const override;
  void PerformAddActiveConnection(
      std::unique_ptr<AuthenticatedChannel> authenticated_channel,
      std::vector<std::unique_ptr<ClientConnectionParameters>> initial_clients,
      const ConnectionDetails& connection_details) override;
  void PerformAddClientToChannel(
      std::unique_ptr<ClientConnectionParameters> client_connection_parameters,
      const ConnectionDetails& connection_details) override;

  DetailsToMetadataMap connection_details_to_active_metadata_map_;

  DISALLOW_COPY_AND_ASSIGN(FakeActiveConnectionManager);
};

// Test ActiveConnectionManager::Delegate implementation.
class FakeActiveConnectionManagerDelegate
    : public ActiveConnectionManager::Delegate {
 public:
  FakeActiveConnectionManagerDelegate();
  ~FakeActiveConnectionManagerDelegate() override;

  const base::flat_map<ConnectionDetails, size_t>&
  connection_details_to_num_disconnections_map() const {
    return connection_details_to_num_disconnections_map_;
  }

 private:
  void OnDisconnected(const ConnectionDetails& connection_details) override;

  base::flat_map<ConnectionDetails, size_t>
      connection_details_to_num_disconnections_map_;

  DISALLOW_COPY_AND_ASSIGN(FakeActiveConnectionManagerDelegate);
};

}  // namespace secure_channel

}  // namespace chromeos

#endif  // CHROMEOS_SERVICES_SECURE_CHANNEL_FAKE_ACTIVE_CONNECTION_MANAGER_H_