src/connectivity/bluetooth/core/bt-host/l2cap/fake_channel_test.cc

// Copyright 2017 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/connectivity/bluetooth/core/bt-host/public/pw_bluetooth_sapphire/internal/host/l2cap/fake_channel_test.h"

#include "src/connectivity/bluetooth/core/bt-host/public/pw_bluetooth_sapphire/internal/host/common/log.h"
#include "src/connectivity/bluetooth/core/bt-host/public/pw_bluetooth_sapphire/internal/host/testing/test_helpers.h"

namespace bt::l2cap::testing {

void FakeChannelTest::SetUp() {}

std::unique_ptr<FakeChannel> FakeChannelTest::CreateFakeChannel(
    const ChannelOptions& options) {
  auto fake_chan = std::make_unique<FakeChannel>(
      options.id,
      options.remote_id,
      options.conn_handle,
      options.link_type,
      ChannelInfo::MakeBasicMode(options.mtu, options.mtu));
  fake_chan_ = fake_chan->AsWeakPtr();
  return fake_chan;
}

bool FakeChannelTest::Expect(const ByteBuffer& expected) {
  return ExpectAfterMaybeReceiving(std::nullopt, expected);
}

bool FakeChannelTest::ReceiveAndExpect(const ByteBuffer& packet,
                                       const ByteBuffer& expected_response) {
  return ExpectAfterMaybeReceiving(packet.view(), expected_response);
}

bool FakeChannelTest::ExpectAfterMaybeReceiving(
    std::optional<BufferView> packet, const ByteBuffer& expected) {
  if (!fake_chan().is_alive()) {
    bt_log(ERROR, "testing", "no channel, failing!");
    return false;
  }

  bool success = false;
  auto cb = [&expected, &success](auto cb_packet) {
    success = ContainersEqual(expected, *cb_packet);
  };

  fake_chan()->SetSendCallback(cb, dispatcher());
  if (packet.has_value()) {
    fake_chan()->Receive(packet.value());
  }
  RunUntilIdle();

  return success;
}

}  // namespace bt::l2cap::testing