src/developer/forensics/testing/stubs/channel_provider.cc

// Copyright 2019 The Fuchsia 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 "src/developer/forensics/testing/stubs/channel_provider.h"

#include <zircon/errors.h>

namespace forensics {
namespace stubs {

void ChannelProvider::GetCurrent(GetCurrentCallback callback) { callback(channel_); }

void ChannelProviderReturnsEmptyChannel::GetCurrent(GetCurrentCallback callback) { callback(""); }

void ChannelProviderClosesFirstConnection::GetCurrent(GetCurrentCallback callback) {
  if (first_call_) {
    first_call_ = false;
    CloseAllConnections();
    return;
  }

  callback(channel_);
}

ChannelProviderExpectsOneCall::~ChannelProviderExpectsOneCall() {
  FX_CHECK(!first_call_) << "No call was made";
}

void ChannelProviderExpectsOneCall::GetCurrent(GetCurrentCallback callback) {
  FX_CHECK(first_call_) << "Only one call to GetCurrent should be made";
  first_call_ = false;

  callback(channel_);
}

}  // namespace stubs
}  // namespace forensics