gadget: fix unit test after rebasing

canonical · Dec 2, 2024 · ef807c5 · ef807c5
1 parent 382f1dc
commit ef807c5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/gadget/gadget_emmc_test.go b/gadget/gadget_emmc_test.go
@@ -345,7 +345,7 @@ func (s *gadgetYamlEMMCSuite) TestUpdateApplyHappy(c *C) {
 
 	rollbackDir := c.MkDir()
 
-	restore := gadget.MockVolumeStructureToLocationMap(func(gd gadget.GadgetData, gm gadget.Model, gv map[string]*gadget.Volume) (map[string]map[int]gadget.StructureLocation, map[string]map[int]*gadget.OnDiskStructure, error) {
+	restore := gadget.MockVolumeStructureToLocationMap(func(_ gadget.Model, _, newVolumes map[string]*gadget.Volume) (map[string]map[int]gadget.StructureLocation, map[string]map[int]*gadget.OnDiskStructure, error) {
 		return map[string]map[int]gadget.StructureLocation{
 				"volumename": {
 					0: {
@@ -363,8 +363,8 @@ func (s *gadgetYamlEMMCSuite) TestUpdateApplyHappy(c *C) {
 					},
 				},
 			}, map[string]map[int]*gadget.OnDiskStructure{
-				"volumename": gadget.OnDiskStructsFromGadget(gd.Info.Volumes["volumename"]),
-				"my-emmc":    gadget.OnDiskStructsFromGadget(gd.Info.Volumes["my-emmc"]),
+				"volumename": gadget.OnDiskStructsFromGadget(newVolumes["volumename"]),
+				"my-emmc":    gadget.OnDiskStructsFromGadget(newVolumes["my-emmc"]),
 			}, nil
 	})
 	defer restore()

diff --git a/gadget/install/install_test.go b/gadget/install/install_test.go
@@ -102,11 +102,6 @@ func (s *installSuite) TestVolumeAssignmentDeviceNameMissing(c *C) {
 			DevNode: "/dev/fakedevice0",
 			DevPath: "/sys/block/fakedevice0",
 		},
-		filepath.Join(s.dir, "/dev/fakedevice1p1"): {
-			DevNum:  "43:0",
-			DevNode: "/dev/fakedevice1",
-			DevPath: "/sys/block/fakedevice1",
-		},
 	}
 
 	restore := disks.MockPartitionDeviceNodeToDiskMapping(m)