| // Copyright 2019 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/disks/mount_point.h" |
| |
| #include "base/run_loop.h" |
| #include "base/test/bind_test_util.h" |
| #include "base/test/gmock_callback_support.h" |
| #include "base/test/task_environment.h" |
| #include "chromeos/disks/mock_disk_mount_manager.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using ::testing::_; |
| using ::testing::WithoutArgs; |
| |
| namespace chromeos { |
| namespace disks { |
| namespace { |
| |
| constexpr char kSourcePath[] = "/source/path"; |
| constexpr char kMountPath[] = "/mount/path"; |
| constexpr char kOtherPath[] = "/other/path"; |
| |
| class MountPointTest : public testing::Test { |
| public: |
| MountPointTest() = default; |
| |
| protected: |
| base::test::TaskEnvironment task_environment_; |
| MockDiskMountManager disk_mount_manager_; |
| }; |
| |
| TEST_F(MountPointTest, Mount) { |
| EXPECT_CALL(disk_mount_manager_, |
| MountPath(kSourcePath, "", "", _, MOUNT_TYPE_DEVICE, |
| MOUNT_ACCESS_MODE_READ_WRITE)) |
| .WillOnce(WithoutArgs([this]() { |
| // Ignore other mount events. |
| disk_mount_manager_.NotifyMountEvent( |
| DiskMountManager::MOUNTING, MOUNT_ERROR_NONE, |
| DiskMountManager::MountPointInfo(kOtherPath, kOtherPath, |
| MOUNT_TYPE_DEVICE, |
| MOUNT_CONDITION_NONE)); |
| disk_mount_manager_.NotifyMountEvent( |
| DiskMountManager::UNMOUNTING, MOUNT_ERROR_NONE, |
| DiskMountManager::MountPointInfo(kSourcePath, kOtherPath, |
| MOUNT_TYPE_DEVICE, |
| MOUNT_CONDITION_NONE)); |
| disk_mount_manager_.NotifyMountEvent( |
| DiskMountManager::MOUNTING, MOUNT_ERROR_NONE, |
| DiskMountManager::MountPointInfo(kSourcePath, kOtherPath, |
| MOUNT_TYPE_ARCHIVE, |
| MOUNT_CONDITION_NONE)); |
| |
| // This is the real mount event. |
| disk_mount_manager_.NotifyMountEvent( |
| DiskMountManager::MOUNTING, MOUNT_ERROR_NONE, |
| DiskMountManager::MountPointInfo(kSourcePath, kMountPath, |
| MOUNT_TYPE_DEVICE, |
| MOUNT_CONDITION_NONE)); |
| })); |
| EXPECT_CALL(disk_mount_manager_, UnmountPath(kMountPath, _)).Times(1); |
| |
| base::RunLoop run_loop; |
| MountPoint::Mount(&disk_mount_manager_, kSourcePath, "", "", {}, |
| MOUNT_TYPE_DEVICE, MOUNT_ACCESS_MODE_READ_WRITE, |
| base::BindLambdaForTesting( |
| [&run_loop](MountError mount_error, |
| std::unique_ptr<MountPoint> mount) { |
| EXPECT_EQ(MOUNT_ERROR_NONE, mount_error); |
| EXPECT_EQ(kMountPath, mount->mount_path().value()); |
| run_loop.Quit(); |
| })); |
| run_loop.Run(); |
| } |
| |
| TEST_F(MountPointTest, MountFailure) { |
| EXPECT_CALL(disk_mount_manager_, |
| MountPath(kSourcePath, "", "", _, MOUNT_TYPE_DEVICE, |
| MOUNT_ACCESS_MODE_READ_WRITE)) |
| .WillOnce(WithoutArgs([this]() { |
| disk_mount_manager_.NotifyMountEvent( |
| DiskMountManager::MOUNTING, MOUNT_ERROR_UNKNOWN, |
| DiskMountManager::MountPointInfo( |
| kSourcePath, kMountPath, MOUNT_TYPE_DEVICE, |
| MOUNT_CONDITION_UNSUPPORTED_FILESYSTEM)); |
| })); |
| EXPECT_CALL(disk_mount_manager_, UnmountPath(_, _)).Times(0); |
| |
| base::RunLoop run_loop; |
| MountPoint::Mount(&disk_mount_manager_, kSourcePath, "", "", {}, |
| MOUNT_TYPE_DEVICE, MOUNT_ACCESS_MODE_READ_WRITE, |
| base::BindLambdaForTesting( |
| [&run_loop](MountError mount_error, |
| std::unique_ptr<MountPoint> mount) { |
| EXPECT_EQ(MOUNT_ERROR_UNKNOWN, mount_error); |
| EXPECT_FALSE(mount); |
| run_loop.Quit(); |
| })); |
| run_loop.Run(); |
| } |
| |
| TEST_F(MountPointTest, Unmount) { |
| EXPECT_CALL(disk_mount_manager_, UnmountPath(kMountPath, _)) |
| .WillOnce(base::test::RunOnceCallback<1>(MOUNT_ERROR_INTERNAL)); |
| |
| base::RunLoop run_loop; |
| MountPoint mount_point(base::FilePath(kMountPath), &disk_mount_manager_); |
| mount_point.Unmount(base::BindLambdaForTesting([&run_loop](MountError error) { |
| EXPECT_EQ(MOUNT_ERROR_INTERNAL, error); |
| run_loop.Quit(); |
| })); |
| run_loop.Run(); |
| } |
| |
| TEST_F(MountPointTest, UnmountOnDestruction) { |
| EXPECT_CALL(disk_mount_manager_, UnmountPath(kMountPath, _)).Times(1); |
| |
| MountPoint mount_point(base::FilePath(kMountPath), &disk_mount_manager_); |
| } |
| |
| } // namespace |
| } // namespace disks |
| } // namespace chromeos |