cristiancmoises
/
gophish
mirror of https://github.com/gophish/gophish
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
gophish/models/result_test.go

104 lines
3.0 KiB
Go
Raw Normal View History

Isolate model tests (#933) * isolates Group model tests * move SMTP related tests from models_test.go to smtp_test.go * isolate Page model tests * isolate Result model tests
2018-01-20 00:11:00 +00:00
package models
import (
"net/mail"
"regexp"
"time"
"gopkg.in/check.v1"
)
func (s *ModelsSuite) TestGenerateResultId(c *check.C) {
r := Result{}
r.GenerateId()
match, err := regexp.Match("[a-zA-Z0-9]{7}", []byte(r.RId))
c.Assert(err, check.Equals, nil)
c.Assert(match, check.Equals, true)
}
func (s *ModelsSuite) TestFormatAddress(c *check.C) {
r := Result{
Implemented the ability to preview landing pages when sending a test email.
2018-06-09 02:20:52 +00:00
BaseRecipient: BaseRecipient{
FirstName: "John",
LastName: "Doe",
Email: "johndoe@example.com",
},
Isolate model tests (#933) * isolates Group model tests * move SMTP related tests from models_test.go to smtp_test.go * isolate Page model tests * isolate Result model tests
2018-01-20 00:11:00 +00:00
}
expected := &mail.Address{
Name: "John Doe",
Address: "johndoe@example.com",
}
c.Assert(r.FormatAddress(), check.Equals, expected.String())
r = Result{
Implemented the ability to preview landing pages when sending a test email.
2018-06-09 02:20:52 +00:00
BaseRecipient: BaseRecipient{Email: "johndoe@example.com"},
Isolate model tests (#933) * isolates Group model tests * move SMTP related tests from models_test.go to smtp_test.go * isolate Page model tests * isolate Result model tests
2018-01-20 00:11:00 +00:00
}
c.Assert(r.FormatAddress(), check.Equals, r.Email)
}
func (s *ModelsSuite) TestResultSendingStatus(ch *check.C) {
c := s.createCampaignDependencies(ch)
ch.Assert(PostCampaign(&c, c.UserId), check.Equals, nil)
// This campaign wasn't scheduled, so we expect the status to
// be sending
for _, r := range c.Results {
ch.Assert(r.Status, check.Equals, STATUS_SENDING)
Refactored result updating to be in result.go. Added the modified_date field to results so it's easy to keep track of the last results that were modified without having to parse every event. Updated the tests to reflect the changes.
2018-05-27 02:26:34 +00:00
ch.Assert(r.ModifiedDate, check.Equals, c.CreatedDate)
Isolate model tests (#933) * isolates Group model tests * move SMTP related tests from models_test.go to smtp_test.go * isolate Page model tests * isolate Result model tests
2018-01-20 00:11:00 +00:00
}
}
func (s *ModelsSuite) TestResultScheduledStatus(ch *check.C) {
c := s.createCampaignDependencies(ch)
c.LaunchDate = time.Now().UTC().Add(time.Hour * time.Duration(1))
ch.Assert(PostCampaign(&c, c.UserId), check.Equals, nil)
// This campaign wasn't scheduled, so we expect the status to
// be sending
for _, r := range c.Results {
ch.Assert(r.Status, check.Equals, STATUS_SCHEDULED)
Refactored result updating to be in result.go. Added the modified_date field to results so it's easy to keep track of the last results that were modified without having to parse every event. Updated the tests to reflect the changes.
2018-05-27 02:26:34 +00:00
ch.Assert(r.ModifiedDate, check.Equals, c.CreatedDate)
Isolate model tests (#933) * isolates Group model tests * move SMTP related tests from models_test.go to smtp_test.go * isolate Page model tests * isolate Result model tests
2018-01-20 00:11:00 +00:00
}
}
func (s *ModelsSuite) TestDuplicateResults(ch *check.C) {
group := Group{Name: "Test Group"}
group.Targets = []Target{
Implemented the ability to preview landing pages when sending a test email.
2018-06-09 02:20:52 +00:00
Target{BaseRecipient: BaseRecipient{Email: "test1@example.com", FirstName: "First", LastName: "Example"}},
Target{BaseRecipient: BaseRecipient{Email: "test1@example.com", FirstName: "Duplicate", LastName: "Duplicate"}},
Target{BaseRecipient: BaseRecipient{Email: "test2@example.com", FirstName: "Second", LastName: "Example"}},
Isolate model tests (#933) * isolates Group model tests * move SMTP related tests from models_test.go to smtp_test.go * isolate Page model tests * isolate Result model tests
2018-01-20 00:11:00 +00:00
}
group.UserId = 1
ch.Assert(PostGroup(&group), check.Equals, nil)
// Add a template
t := Template{Name: "Test Template"}
t.Subject = "{{.RId}} - Subject"
t.Text = "{{.RId}} - Text"
t.HTML = "{{.RId}} - HTML"
t.UserId = 1
ch.Assert(PostTemplate(&t), check.Equals, nil)
// Add a landing page
p := Page{Name: "Test Page"}
p.HTML = "<html>Test</html>"
p.UserId = 1
ch.Assert(PostPage(&p), check.Equals, nil)
// Add a sending profile
smtp := SMTP{Name: "Test Page"}
smtp.UserId = 1
smtp.Host = "example.com"
smtp.FromAddress = "test@test.com"
ch.Assert(PostSMTP(&smtp), check.Equals, nil)
c := Campaign{Name: "Test campaign"}
c.UserId = 1
c.Template = t
c.Page = p
c.SMTP = smtp
c.Groups = []Group{group}
ch.Assert(PostCampaign(&c, c.UserId), check.Equals, nil)
ch.Assert(len(c.Results), check.Equals, 2)
ch.Assert(c.Results[0].Email, check.Equals, group.Targets[0].Email)
ch.Assert(c.Results[1].Email, check.Equals, group.Targets[2].Email)
}