src/storage/blobfs/test/unit/node-reserver-test.cc

// Copyright 2018 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/storage/blobfs/allocator/node-reserver.h"

#include <bitmap/rle-bitmap.h>
#include <gtest/gtest.h>

#include "src/storage/blobfs/allocator/extent-reserver.h"

namespace blobfs {
namespace {

// Test that reserving a node actually changes the node count, and that RAII releases the node.
TEST(NodeReserver, Reserve) {
  NodeReserver reserver;
  {
    const uint32_t ino = 3;
    ReservedNode reserved_node(&reserver, ino);
    EXPECT_EQ(1u, reserver.ReservedNodeCount());
  }
  EXPECT_EQ(0u, reserver.ReservedNodeCount());
}

TEST(NodeReserver, ReserveReset) {
  NodeReserver reserver;
  {
    const uint32_t ino = 3;
    ReservedNode reserved_node(&reserver, ino);
    EXPECT_EQ(1u, reserver.ReservedNodeCount());
    reserved_node.Reset();
    EXPECT_EQ(0u, reserver.ReservedNodeCount());
  }
  EXPECT_EQ(0u, reserver.ReservedNodeCount());
}

// Test the constructors of the reserved node.
TEST(NodeReserver, Constructor) {
  NodeReserver reserver;
  // Test the constructor.
  {
    ReservedNode reserved_node(&reserver, 3);
    EXPECT_EQ(3u, reserved_node.index());
    EXPECT_EQ(1u, reserver.ReservedNodeCount());
  }
  EXPECT_EQ(0u, reserver.ReservedNodeCount());
}

TEST(NodeReserver, MoveConstructor) {
  NodeReserver reserver;
  // Test the move constructor.
  {
    ReservedNode reserved_node(&reserver, 3);
    EXPECT_EQ(3u, reserved_node.index());
    EXPECT_EQ(1u, reserver.ReservedNodeCount());

    ReservedNode dest_node(std::move(reserved_node));
    EXPECT_EQ(3u, dest_node.index());
    EXPECT_EQ(1u, reserver.ReservedNodeCount());
  }
  EXPECT_EQ(0u, reserver.ReservedNodeCount());
}

TEST(NodeReserver, MoveAssignment) {
  NodeReserver reserver;
  // Test the move assignment operator.
  {
    ReservedNode reserved_node(&reserver, 3);
    EXPECT_EQ(3u, reserved_node.index());
    EXPECT_EQ(1u, reserver.ReservedNodeCount());

    ReservedNode dest_node = std::move(reserved_node);
    EXPECT_EQ(3u, dest_node.index());
    EXPECT_EQ(1u, reserver.ReservedNodeCount());
  }
  EXPECT_EQ(0u, reserver.ReservedNodeCount());
}

}  // namespace
}  // namespace blobfs