test: use _cleanup_ attribute

Signed-off-by: Anthony G. Basile <blueness@gentoo.org>

1 files changed, 4 insertions, 4 deletions

diff --git a/test/test-libudev.c b/test/test-libudev.c
index 0e1202b449..912b61396a 100644
--- a/test/test-libudev.c
+++ b/test/test-libudev.c
@@ -30,6 +30,7 @@
 #include <sys/epoll.h>
 
 #include "libudev.h"
+#include "udev-util.h"
 #include "util.h"
 
 #define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
@@ -116,7 +117,7 @@ static void print_device(struct udev_device *device) {
 }
 
 static int test_device(struct udev *udev, const char *syspath) {
-        struct udev_device *device;
+        _cleanup_udev_device_unref_ struct udev_device *device;
 
         printf("looking at device: %s\n", syspath);
         device = udev_device_new_from_syspath(udev, syspath);
@@ -125,12 +126,12 @@ static int test_device(struct udev *udev, const char *syspath) {
                 return -1;
         }
         print_device(device);
-        udev_device_unref(device);
+
         return 0;
 }
 
 static int test_device_parents(struct udev *udev, const char *syspath) {
-        struct udev_device *device;
+        _cleanup_udev_device_unref_ struct udev_device *device;
         struct udev_device *device_parent;
 
         printf("looking at device: %s\n", syspath);
@@ -151,7 +152,6 @@ static int test_device_parents(struct udev *udev, const char *syspath) {
                 print_device(device_parent);
                 device_parent = udev_device_get_parent(device_parent);
         } while (device_parent != NULL);
-        udev_device_unref(device);
 
         return 0;
 }