Properly returning 404 error if the requested sending profile isn't found.

This commit is contained in:
Jordan Wright 2018-10-07 12:37:15 -05:00
parent 10aa98b760
commit 81da804761
2 changed files with 8 additions and 0 deletions

View file

@ -146,6 +146,7 @@ func GetSMTP(id int64, uid int64) (SMTP, error) {
err := db.Where("user_id=? and id=?", uid, id).Find(&s).Error
if err != nil {
log.Error(err)
return s, err
}
err = db.Where("smtp_id=?", s.Id).Find(&s.Headers).Error
if err != nil && err != gorm.ErrRecordNotFound {

View file

@ -3,6 +3,8 @@ package models
import (
"fmt"
"github.com/jinzhu/gorm"
check "gopkg.in/check.v1"
)
@ -74,3 +76,8 @@ func (s *ModelsSuite) TestSMTPGetDialer(ch *check.C) {
ch.Assert(dialer.TLSConfig.ServerName, check.Equals, smtp.Host)
ch.Assert(dialer.TLSConfig.InsecureSkipVerify, check.Equals, smtp.IgnoreCertErrors)
}
func (s *ModelsSuite) TestGetInvalidSMTP(ch *check.C) {
_, err := GetSMTP(-1, 1)
ch.Assert(err, check.Equals, gorm.ErrRecordNotFound)
}