test: TestGetMountablePartitions with unpartitioned device

mondoohq · Nov 19, 2024 · 924a671 · 924a671
1 parent ab3545c
commit 924a671
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/providers/os/connection/snapshot/blockdevices_test.go b/providers/os/connection/snapshot/blockdevices_test.go
@@ -285,6 +285,21 @@ func TestGetMountablePartitions(t *testing.T) {
 		}
 		require.ElementsMatch(t, expected, parts)
 	})
+
+	t.Run("get all non-mounted partitions (unpartitioned)", func(t *testing.T) {
+		block := BlockDevice{
+			Name:   "sda",
+			FsType: "xfs",
+			Label:  "ROOT",
+			Uuid:   "1234",
+		}
+		parts, err := block.GetMountablePartitions(true)
+		require.NoError(t, err)
+		expected := []*PartitionInfo{
+			{Name: "/dev/sda", FsType: "xfs", Uuid: "1234", Label: "ROOT"},
+		}
+		require.ElementsMatch(t, expected, parts)
+	})
 }
 
 func TestLongestMatchingSuffix(t *testing.T) {