system/xhost: wrap revert error correctly

This otherwise creates a confusing error message on a revert failure.

Signed-off-by: Ophestra <cat@gensokyo.uk>

1 files changed, 9 insertions, 0 deletions

diff --git a/system/dispatcher_test.go b/system/dispatcher_test.go
index f798047c..7bd4e6db 100644
--- a/system/dispatcher_test.go
+++ b/system/dispatcher_test.go
@@ -13,6 +13,7 @@ import (
 	"hakurei.app/container/stub"
 	"hakurei.app/system/acl"
 	"hakurei.app/system/dbus"
+	"hakurei.app/system/internal/xcb"
 )
 
 // call initialises a [stub.Call].
@@ -280,6 +281,14 @@ func (k *kstub) aclUpdate(name string, uid int, perms ...acl.Perm) error {
 		stub.CheckArgReflect(k.Stub, "perms", perms, 2))
 }
 
+func (k *kstub) xcbChangeHosts(mode xcb.HostMode, family xcb.Family, address string) error {
+	k.Helper()
+	return k.Expects("xcbChangeHosts").Error(
+		stub.CheckArg(k.Stub, "mode", mode, 0),
+		stub.CheckArg(k.Stub, "family", family, 1),
+		stub.CheckArg(k.Stub, "address", address, 2))
+}
+
 func (k *kstub) dbusAddress() (session, system string) {
 	k.Helper()
 	ret := k.Expects("dbusAddress").Ret.([2]string)