Simplified integration tests

This commit is contained in:
Michal Horejsek
2020-09-03 08:39:47 +02:00
parent c7578cf53c
commit 6671b78799
5 changed files with 17 additions and 29 deletions

View File

@ -130,7 +130,7 @@ func getLabelExclusive(name string) int {
return 0
}
func (ctl *Controller) AddUserMessage(username string, message *pmapi.Message) error {
func (ctl *Controller) AddUserMessage(username string, message *pmapi.Message) (string, error) {
if _, ok := ctl.messagesByUsername[username]; !ok {
ctl.messagesByUsername[username] = []*pmapi.Message{}
}
@ -138,7 +138,7 @@ func (ctl *Controller) AddUserMessage(username string, message *pmapi.Message) e
message.LabelIDs = append(message.LabelIDs, pmapi.AllMailLabel)
ctl.messagesByUsername[username] = append(ctl.messagesByUsername[username], message)
ctl.resetUsers()
return nil
return message.ID, nil
}
func (ctl *Controller) getFakeAPIForUser(userID string) *FakePMAPI {
@ -172,8 +172,4 @@ func (ctl *Controller) GetMessages(username, labelID string) ([]*pmapi.Message,
}
}
return messages, nil
func (ctl *Controller) GetLastMessageID(username string) string {
msgs := ctl.messagesByUsername[username]
return msgs[len(msgs)-1].ID
}