mirror of
https://github.com/gophish/gophish
synced 2024-11-14 16:27:23 +00:00
Properly returning 404 error if the requested sending profile isn't found.
This commit is contained in:
parent
10aa98b760
commit
81da804761
2 changed files with 8 additions and 0 deletions
|
@ -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 {
|
||||
|
|
|
@ -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)
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue