Fix integration tests - compiting message flags

pkg/message/flags.go

@@ -60,6 +60,7 @@ func GetFlags(m *pmapi.Message) (flags []string) {
 }
 
 func ParseFlags(m *pmapi.Message, flags []string) {
+	// Consider to use ComputeMessageFlagsByLabels to keep logic in one place.
 	if (m.Flags & pmapi.FlagSent) == 0 {
 		m.Flags |= pmapi.FlagReceived
 	}

pkg/pmapi/messages.go

@@ -839,3 +839,22 @@ func (c *client) EmptyFolder(labelID, addressID string) (err error) {
 	err = res.Err()
 	return
 }
+
+// ComputeMessageFlagsByLabels returns flags based on labels.
+func ComputeMessageFlagsByLabels(labels []string) (flag int64) {
+	for _, labelID := range labels {
+		switch labelID {
+		case SentLabel:
+			flag = (flag | FlagSent)
+		case ArchiveLabel, InboxLabel:
+			flag = (flag | FlagReceived)
+		}
+	}
+
+	// NOTE: if the labels are custom only
+	if flag == 0 {
+		flag = FlagReceived
+	}
+
+	return flag
+}

test/liveapi/messages.go

@@ -36,6 +36,10 @@ func (ctl *Controller) AddUserMessage(username string, message *pmapi.Message) e
 		return fmt.Errorf("user %s does not exist", username)
 	}
 
+	if message.Flags == 0 {
+		message.Flags = pmapi.ComputeMessageFlagsByLabels(message.LabelIDs)
+	}
+
 	body, err := buildMessage(client, message)
 	if err != nil {
 		return errors.Wrap(err, "failed to build message")