1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74


content / browser / smart_card / mock_smart_card_context_factory.cc [blame]

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

#include "content/browser/smart_card/mock_smart_card_context_factory.h"
#include "base/test/gmock_callback_support.h"

using base::test::RunOnceCallback;
using device::mojom::SmartCardCreateContextResult;
using device::mojom::SmartCardProtocol;
using device::mojom::SmartCardShareMode;
using testing::_;

namespace content {

MockSmartCardContextFactory::MockSmartCardContextFactory() {
  context_receivers_.set_disconnect_handler(
      base::BindRepeating(&MockSmartCardContextFactory::ContextDisconnected,
                          base::Unretained(this)));
}

MockSmartCardContextFactory::~MockSmartCardContextFactory() = default;

mojo::PendingRemote<device::mojom::SmartCardContextFactory>
MockSmartCardContextFactory::GetRemote() {
  mojo::PendingRemote<device::mojom::SmartCardContextFactory> pending_remote;
  receivers_.Add(this, pending_remote.InitWithNewPipeAndPassReceiver());
  return pending_remote;
}

void MockSmartCardContextFactory::CreateContext(
    CreateContextCallback callback) {
  mojo::PendingRemote<device::mojom::SmartCardContext> context_remote;
  context_receivers_.Add(this, context_remote.InitWithNewPipeAndPassReceiver());

  std::move(callback).Run(
      SmartCardCreateContextResult::NewContext(std::move(context_remote)));
}

void MockSmartCardContextFactory::ExpectConnectFakeReaderSharedT1(
    mojo::Receiver<device::mojom::SmartCardConnection>& connection_receiver) {
  EXPECT_CALL(*this, Connect("Fake reader", SmartCardShareMode::kShared, _, _))
      .WillOnce([&connection_receiver](
                    const std::string& reader,
                    device::mojom::SmartCardShareMode share_mode,
                    device::mojom::SmartCardProtocolsPtr preferred_protocols,
                    SmartCardContext::ConnectCallback callback) {
        EXPECT_FALSE(preferred_protocols->t0);
        EXPECT_TRUE(preferred_protocols->t1);
        EXPECT_FALSE(preferred_protocols->raw);

        auto success = device::mojom::SmartCardConnectSuccess::New(
            connection_receiver.BindNewPipeAndPassRemote(),
            SmartCardProtocol::kT1);

        std::move(callback).Run(
            device::mojom::SmartCardConnectResult::NewSuccess(
                std::move(success)));
      });
}

void MockSmartCardContextFactory::ExpectListReaders(
    std::vector<std::string> readers) {
  EXPECT_CALL(*this, ListReaders(_))
      .WillOnce(RunOnceCallback<0>(
          device::mojom::SmartCardListReadersResult::NewReaders(
              std::move(readers))));
}

void MockSmartCardContextFactory::ClearContextReceivers() {
  context_receivers_.Clear();
}

}  // namespace content