Merge pull request #614 from SchSeba/fix_device_spec_rdma_2

Fix unit tests
k8snetworkplumbingwg · Dec 11, 2024 · 12510dd · 12510dd
2 parents 2b3ddae + 8f09dbe
commit 12510dd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/netdevice/netResourcePool_test.go b/pkg/netdevice/netResourcePool_test.go
@@ -149,6 +149,8 @@ var _ = Describe("NetResourcePool", func() {
 						}
 						return nil
 					})
+				nadutils.On("CleanDeviceInfoFile", "fakeOrg.io/fakeResource", "fake1").Return(nil)
+				nadutils.On("CleanDeviceInfoFile", "fakeOrg.io/fakeResource", "fake2").Return(nil)
 				rp := netdevice.NewNetResourcePool(nadutils, rc, pcis)
 				err := rp.StoreDeviceInfoFile("fakeOrg.io", []string{"fake1", "fake2"})
 				nadutils.AssertExpectations(t)
@@ -216,6 +218,9 @@ var _ = Describe("NetResourcePool", func() {
 						}
 						return nil
 					})
+				nadutils.On("CleanDeviceInfoFile", "fakeOrg.io/fakeResource", "fake1").Return(nil)
+				nadutils.On("CleanDeviceInfoFile", "fakeOrg.io/fakeResource", "fake2").Return(nil)
+
 				rp := netdevice.NewNetResourcePool(nadutils, rc, pcis)
 				err := rp.StoreDeviceInfoFile("fakeOrg.io", []string{"fake1", "fake2"})
 				Expect(err).ToNot(HaveOccurred())