chromeos/components/drivefs/pending_connection_manager_unittest.cc

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

#include "chromeos/components/drivefs/pending_connection_manager.h"

#include "base/test/bind_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace drivefs {

class PendingConnectionManagerTest : public testing::Test {
 protected:
  PendingConnectionManager connection_manager_;
};

namespace {

TEST_F(PendingConnectionManagerTest, Basic) {
  auto token = base::UnguessableToken::Create();
  int callback_calls = 0;
  connection_manager_.ExpectOpenIpcChannel(
      token,
      base::BindLambdaForTesting([&](base::ScopedFD fd) { callback_calls++; }));
  EXPECT_TRUE(connection_manager_.OpenIpcChannel(token.ToString(), {}));
  EXPECT_FALSE(connection_manager_.OpenIpcChannel(token.ToString(), {}));
  EXPECT_EQ(1, callback_calls);
}

TEST_F(PendingConnectionManagerTest, Cancel) {
  auto token = base::UnguessableToken::Create();
  connection_manager_.ExpectOpenIpcChannel(
      token, base::BindLambdaForTesting(
                 [&](base::ScopedFD fd) { FAIL() << "Unexpected call"; }));
  connection_manager_.CancelExpectedOpenIpcChannel(token);
  EXPECT_FALSE(connection_manager_.OpenIpcChannel(token.ToString(), {}));
}

TEST_F(PendingConnectionManagerTest, UnexpectedConnection) {
  EXPECT_FALSE(connection_manager_.OpenIpcChannel("invalid", {}));
}

}  // namespace
}  // namespace drivefs