sandbox: wrap fmsg interface

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

1 files changed, 3 insertions, 1 deletions

diff --git a/dbus/dbus_test.go b/dbus/dbus_test.go
index b8391c3b..bba06f6c 100644
--- a/dbus/dbus_test.go
+++ b/dbus/dbus_test.go
@@ -14,6 +14,7 @@ import (
 	"git.gensokyo.uk/security/fortify/dbus"
 	"git.gensokyo.uk/security/fortify/helper"
 	"git.gensokyo.uk/security/fortify/internal"
+	"git.gensokyo.uk/security/fortify/internal/fmsg"
 	"git.gensokyo.uk/security/fortify/internal/sandbox"
 )
 
@@ -244,5 +245,6 @@ func TestHelperInit(t *testing.T) {
 	if len(os.Args) != 5 || os.Args[4] != "init" {
 		return
 	}
-	sandbox.Init(internal.Exit)
+	sandbox.SetOutput(fmsg.Output{})
+	sandbox.Init(fmsg.Prepare, internal.InstallFmsg)
 }