fix: Enable tracing for demo and set server flags

ProtonMail · Jul 11, 2022 · 3134478 · 3134478
1 parent b8ae801
commit 3134478
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/demo/demo.go b/demo/demo.go
@@ -22,8 +22,14 @@ func main() {
 		logrus.SetLevel(level)
 	}
 
+	loggerIn := logrus.StandardLogger().WriterLevel(logrus.TraceLevel)
+	loggerOut := logrus.StandardLogger().WriterLevel(logrus.TraceLevel)
+
 	ctx := context.Background()
-	server, err := gluon.New(temp())
+	server, err := gluon.New(temp(), gluon.WithLogger(
+		loggerIn,
+		loggerOut,
+	))
 
 	defer server.Close(ctx)
 
@@ -56,8 +62,8 @@ func addUser(ctx context.Context, server *gluon.Server, addresses []string, pass
 		addresses,
 		password,
 		time.Second,
-		imap.NewFlagSet(),
-		imap.NewFlagSet(),
+		imap.NewFlagSet(`\Answered`, `\Seen`, `\Flagged`, `\Deleted`),
+		imap.NewFlagSet(`\Answered`, `\Seen`, `\Flagged`, `\Deleted`),
 		imap.NewFlagSet(),
 	)