internal/efi: add some extra tests

canonical · Jul 23, 2024 · 046e611 · 046e611
1 parent d0de70e
commit 046e611
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/internal/efi/default_env_test.go b/internal/efi/default_env_test.go
@@ -279,3 +279,29 @@ func (s *defaultEnvSuite) TestDevicesForClassIOMMU(c *C) {
 	c.Check(devices[0].Path(), Equals, filepath.Join(path, "devices/virtual/iommu/dmar0"))
 	c.Check(devices[0].Subsystem(), Equals, "iommu")
 }
+
+func (s *defaultEnvSuite) TestDevicesForClassNotExist(c *C) {
+	_, restore := s.mockSysfsPath(c, "testdata/sys.tar")
+	defer restore()
+
+	devices, err := DefaultEnv.DevicesForClass("foo")
+	c.Check(err, IsNil)
+	c.Assert(devices, HasLen, 0)
+}
+
+func (s *defaultEnvSuite) TestSysfsDeviceAttributeReaderNoAttr(c *C) {
+	_, restore := s.mockSysfsPath(c, "testdata/sys.tar")
+	defer restore()
+
+	devices, err := DefaultEnv.DevicesForClass("mei")
+	c.Check(err, IsNil)
+	c.Assert(devices, HasLen, 1)
+
+	_, err = devices[0].AttributeReader("uevent")
+	c.Check(err, Equals, ErrNoDeviceAttribute)
+	_, err = devices[0].AttributeReader("foo")
+	c.Check(err, Equals, ErrNoDeviceAttribute)
+	_, err = devices[0].AttributeReader("subsystem")
+	c.Check(err, Equals, ErrNoDeviceAttribute)
+
+}