From 279a9c44ed9de31b3d488f39ad67fa5cc7ef5148 Mon Sep 17 00:00:00 2001
From: Mike Fleetwood <mike.fleetwood@googlemail.com>
Date: Sat, 27 Jul 2019 09:59:22 +0100
Subject: [PATCH] Add offline ext2 resizing tests (!49)

Closes !49 - Add file system interface tests
---
 tests/test_ext2.cc | 66 +++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 59 insertions(+), 7 deletions(-)

diff --git a/tests/test_ext2.cc b/tests/test_ext2.cc
index be7a5aea..34acda67 100644
--- a/tests/test_ext2.cc
+++ b/tests/test_ext2.cc
@@ -178,19 +178,25 @@ std::ostream& operator<<(std::ostream& out, const OperationDetail& od)
 	}
 
 
+const Byte_Value IMAGESIZE_Default = 256*MEBIBYTE;
+const Byte_Value IMAGESIZE_Larger  = 512*MEBIBYTE;
+
+
 class ext2Test : public ::testing::Test
 {
 protected:
 	// Initialise top-level operation detail object with description ...
 	ext2Test() : m_operation_detail("Operation details:", STATUS_NONE)  {};
 
-	virtual void extra_setup();
+	virtual void extra_setup(Byte_Value size = IMAGESIZE_Default);
 	virtual void TearDown();
 
 	static void SetUpTestCase();
 	static void TearDownTestCase();
 
 	virtual void reload_partition();
+	virtual void resize_image(Byte_Value new_size);
+	virtual void shrink_partition(Byte_Value size);
 
 	static FileSystem* s_ext2_obj;
 	static FS          s_ext2_support;
@@ -206,17 +212,15 @@ FS          ext2Test::s_ext2_support;
 const char* ext2Test::s_image_name   = "test_ext2.img";
 
 
-void ext2Test::extra_setup()
+void ext2Test::extra_setup(Byte_Value size)
 {
-	const Byte_Value ImageSize = 256*MEBIBYTE;
-
-	// Create new 256M image file to work with.
+	// Create new image file to work with.
 	unlink(s_image_name);
 	int fd = open(s_image_name, O_WRONLY|O_CREAT|O_NONBLOCK, 0666);
 	ASSERT_GE(fd, 0) << "Failed to create image file '" << s_image_name << "'.  errno="
 	                 << errno << "," << strerror(errno);
-	ASSERT_EQ(ftruncate(fd, (off_t)ImageSize), 0) << "Failed to set image file '" << s_image_name << "' to size "
-	                                              << ImageSize << ".  errno=" << errno << "," << strerror(errno);
+	ASSERT_EQ(ftruncate(fd, (off_t)size), 0) << "Failed to set image file '" << s_image_name << "' to size "
+	                                         << size << ".  errno=" << errno << "," << strerror(errno);
 	close(fd);
 
 	reload_partition();
@@ -269,6 +273,25 @@ void ext2Test::reload_partition()
 }
 
 
+void ext2Test::resize_image(Byte_Value new_size)
+{
+	int fd = open(s_image_name, O_WRONLY|O_NONBLOCK);
+	ASSERT_GE(fd, 0) << "Failed to open image file '" << s_image_name << "'.  errno="
+	                 << errno << "," << strerror(errno);
+	ASSERT_EQ(ftruncate(fd, (off_t)new_size), 0) << "Failed to resize image file '" << s_image_name << "' to size "
+	                                             << new_size << ".  errno=" << errno << "," << strerror(errno);
+	close(fd);
+}
+
+
+void ext2Test::shrink_partition(Byte_Value new_size)
+{
+	ASSERT_LE(new_size, m_partition.get_byte_length()) << __func__ << "(): TEST_BUG: Cannot grow Partition object size";
+	Sector new_sectors = (new_size + m_partition.sector_size - 1) / m_partition.sector_size;
+	m_partition.sector_end = new_sectors;
+}
+
+
 TEST_F(ext2Test, Create)
 {
 	SKIP_IF_FS_DOESNT_SUPPORT(create);
@@ -398,6 +421,35 @@ TEST_F(ext2Test, CreateAndRemove)
 }
 
 
+TEST_F(ext2Test, CreateAndGrow)
+{
+	SKIP_IF_FS_DOESNT_SUPPORT(create);
+	SKIP_IF_FS_DOESNT_SUPPORT(grow);
+
+	extra_setup(IMAGESIZE_Default);
+	ASSERT_TRUE(s_ext2_obj->create(m_partition, m_operation_detail)) << m_operation_detail;
+
+	// Test growing the file system is successful.
+	resize_image(IMAGESIZE_Larger);
+	reload_partition();
+	ASSERT_TRUE(s_ext2_obj->resize(m_partition, m_operation_detail, true)) << m_operation_detail;
+}
+
+
+TEST_F(ext2Test, CreateAndShrink)
+{
+	SKIP_IF_FS_DOESNT_SUPPORT(create);
+	SKIP_IF_FS_DOESNT_SUPPORT(shrink);
+
+	extra_setup(IMAGESIZE_Larger);
+	ASSERT_TRUE(s_ext2_obj->create(m_partition, m_operation_detail)) << m_operation_detail;
+
+	// Test shrinking the file system is successful.
+	shrink_partition(IMAGESIZE_Default);
+	ASSERT_TRUE(s_ext2_obj->resize(m_partition, m_operation_detail, false)) << m_operation_detail;
+}
+
+
 }  // namespace GParted