Add tests for FsSizeGB in MachineRootfs

Author: dangra

machine_types_test.go

@@ -320,6 +320,16 @@ func TestMachineRootfsJSON(t *testing.T) {
 				input:  MachineConfig{Rootfs: &MachineRootfs{Persist: MachinePersistRootfsRestart}},
 				output: `{"init":{},"rootfs":{"persist":"restart"}}`,
 			},
+			{
+				name:   "rootfs with fs_size_gb",
+				input:  MachineConfig{Rootfs: &MachineRootfs{Persist: MachinePersistRootfsAlways, SizeGB: 10, FsSizeGB: 20}},
+				output: `{"init":{},"rootfs":{"persist":"always","size_gb":10,"fs_size_gb":20}}`,
+			},
+			{
+				name:   "rootfs with fs_size_gb only",
+				input:  MachineConfig{Rootfs: &MachineRootfs{FsSizeGB: 15}},
+				output: `{"init":{},"rootfs":{"fs_size_gb":15}}`,
+			},
 			{
 				name:   "nil rootfs omitted",
 				input:  MachineConfig{},
@@ -338,23 +348,26 @@ func TestMachineRootfsJSON(t *testing.T) {
 
 	t.Run("unmarshal", func(t *testing.T) {
 		cases := []struct {
-			name    string
-			input   string
-			persist MachinePersistRootfs
-			sizeGB  uint64
+			name     string
+			input    string
+			persist  MachinePersistRootfs
+			sizeGB   uint64
+			fsSizeGB uint64
 		}{
-			{"persist and size", `{"rootfs":{"persist":"always","size_gb":10}}`, MachinePersistRootfsAlways, 10},
-			{"persist only", `{"rootfs":{"persist":"restart"}}`, MachinePersistRootfsRestart, 0},
-			{"size only", `{"rootfs":{"size_gb":5}}`, MachinePersistRootfsNone, 5},
-			{"no rootfs", `{}`, MachinePersistRootfsNone, 0},
+			{"persist and size", `{"rootfs":{"persist":"always","size_gb":10}}`, MachinePersistRootfsAlways, 10, 0},
+			{"persist only", `{"rootfs":{"persist":"restart"}}`, MachinePersistRootfsRestart, 0, 0},
+			{"size only", `{"rootfs":{"size_gb":5}}`, MachinePersistRootfsNone, 5, 0},
+			{"fs_size_gb only", `{"rootfs":{"fs_size_gb":8}}`, MachinePersistRootfsNone, 0, 8},
+			{"all fields", `{"rootfs":{"persist":"always","size_gb":10,"fs_size_gb":20}}`, MachinePersistRootfsAlways, 10, 20},
+			{"no rootfs", `{}`, MachinePersistRootfsNone, 0, 0},
 		}
 		for _, tc := range cases {
 			var mc MachineConfig
 			if err := json.Unmarshal([]byte(tc.input), &mc); err != nil {
 				t.Errorf("%s: unexpected error: %v", tc.name, err)
 				continue
 			}
-			if tc.persist == MachinePersistRootfsNone && tc.sizeGB == 0 {
+			if tc.persist == MachinePersistRootfsNone && tc.sizeGB == 0 && tc.fsSizeGB == 0 {
 				if mc.Rootfs != nil {
 					t.Errorf("%s: expected nil rootfs, got %+v", tc.name, mc.Rootfs)
 				}
@@ -370,6 +383,9 @@ func TestMachineRootfsJSON(t *testing.T) {
 			if mc.Rootfs.SizeGB != tc.sizeGB {
 				t.Errorf("%s: size_gb got %d, want %d", tc.name, mc.Rootfs.SizeGB, tc.sizeGB)
 			}
+			if mc.Rootfs.FsSizeGB != tc.fsSizeGB {
+				t.Errorf("%s: fs_size_gb got %d, want %d", tc.name, mc.Rootfs.FsSizeGB, tc.fsSizeGB)
+			}
 		}
 	})
 }