test: Ensure initial sync has finished before running tests

ProtonMail · Sep 2, 2022 · 6073aff · 6073aff
1 parent 0410288
commit 6073aff
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/connector/dummy.go b/connector/dummy.go
@@ -245,10 +245,14 @@ func (conn *Dummy) MarkMessagesFlagged(ctx context.Context, messageIDs []imap.Me
 
 func (conn *Dummy) Sync(ctx context.Context) error {
 	for _, mailbox := range conn.state.getLabels() {
-		conn.updateCh <- imap.NewMailboxCreated(mailbox)
+		update := imap.NewMailboxCreated(mailbox)
+		defer update.Wait()
+
+		conn.updateCh <- update
 	}
 
 	update := imap.NewMessagesCreated()
+	defer update.Wait()
 
 	for _, message := range conn.state.getMessages() {
 		if err := update.Add(message, conn.state.getLiteral(message.ID), conn.state.getLabelIDs(message.ID)...); err != nil {