Add retries to TestEmailMention test (#6893)

1 files changed, 9 insertions, 4 deletions

diff --git a/api/post_test.go b/api/post_test.go
index c4e016b0b..c7bd7a04c 100644
--- a/api/post_test.go
+++ b/api/post_test.go
@@ -1073,10 +1073,15 @@ func TestEmailMention(t *testing.T) {
 		if !strings.ContainsAny(resultsMailbox[len(resultsMailbox)-1].To[0], th.BasicUser2.Email) {
 			t.Fatal("Wrong To recipient")
 		} else {
-			if resultsEmail, err := utils.GetMessageFromMailbox(th.BasicUser2.Email, resultsMailbox[len(resultsMailbox)-1].ID); err == nil {
-				if !strings.Contains(resultsEmail.Body.Text, post1.Message) {
-					t.Log(resultsEmail.Body.Text)
-					t.Fatal("Received wrong Message")
+			for i := 0; i < 5; i++ {
+				if resultsEmail, err := utils.GetMessageFromMailbox(th.BasicUser2.Email, resultsMailbox[len(resultsMailbox)-1].ID); err == nil {
+					if strings.Contains(resultsEmail.Body.Text, post1.Message) {
+						break
+					} else if i == 4 {
+						t.Log(resultsEmail.Body.Text)
+						t.Fatal("Received wrong Message")
+					}
+					time.Sleep(100 * time.Millisecond)
 				}
 			}
 		}