1 files changed, 7 insertions, 2 deletions
diff --git a/test/syscalls/linux/socket_netlink_util.cc b/test/syscalls/linux/socket_netlink_util.cc
index 728d25434..36b6560c2 100644
--- a/test/syscalls/linux/socket_netlink_util.cc
+++ b/test/syscalls/linux/socket_netlink_util.cc
@@ -54,7 +54,8 @@ PosixErrorOr<uint32_t> NetlinkPortID(int fd) {
 
 PosixError NetlinkRequestResponse(
     const FileDescriptor& fd, void* request, size_t len,
-    const std::function<void(const struct nlmsghdr* hdr)>& fn) {
+    const std::function<void(const struct nlmsghdr* hdr)>& fn,
+    bool expect_nlmsgerr) {
   struct iovec iov = {};
   iov.iov_base = request;
   iov.iov_len = len;
@@ -93,7 +94,11 @@ PosixError NetlinkRequestResponse(
     }
   } while (type != NLMSG_DONE && type != NLMSG_ERROR);
 
-  EXPECT_EQ(type, NLMSG_DONE);
+  if (expect_nlmsgerr) {
+    EXPECT_EQ(type, NLMSG_ERROR);
+  } else {
+    EXPECT_EQ(type, NLMSG_DONE);
+  }
   return NoError();
 }