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

522 lines
15 KiB
Go
Raw Normal View History

Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
package models
import (
Email refactoring (#878) The initial pass at refactoring the way we send emails.
2017-12-09 21:42:07 +00:00
"fmt"
Phishing emails are now sent in "First Last <email@domain.com>" format.
2017-06-17 03:21:08 +00:00
"net/mail"
Changed result ID's to be 7 random alphanumeric characters. Fixes #518
2017-01-28 03:16:47 +00:00
"regexp"
Adding tests
2016-02-26 01:58:02 +00:00
"strings"
Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
"testing"
Email refactoring (#878) The initial pass at refactoring the way we send emails.
2017-12-09 21:42:07 +00:00
"time"
Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
Adding tests
2016-02-26 01:58:02 +00:00
"github.com/PuerkitoBio/goquery"
Update bcrypt dependency and code moved to gophish group
2016-01-10 17:03:17 +00:00
"github.com/gophish/gophish/config"
Check for specific error in group tests
2016-07-18 22:47:28 +00:00
"github.com/jinzhu/gorm"
Added ability to delete campaign from campaign results - needs some cleanup, but it works
2015-02-21 07:27:32 +00:00
"gopkg.in/check.v1"
Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
)
// Hook up gocheck into the "go test" runner.
Added ability to delete campaign from campaign results - needs some cleanup, but it works
2015-02-21 07:27:32 +00:00
func Test(t *testing.T) { check.TestingT(t) }
Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
type ModelsSuite struct{}
Added ability to delete campaign from campaign results - needs some cleanup, but it works
2015-02-21 07:27:32 +00:00
var _ = check.Suite(&ModelsSuite{})
Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
Added ability to delete campaign from campaign results - needs some cleanup, but it works
2015-02-21 07:27:32 +00:00
func (s *ModelsSuite) SetUpSuite(c *check.C) {
Adding support for Mysql (#442) Thanks, @svigne1! Fixes #53
2016-11-19 16:37:22 +00:00
config.Conf.DBName = "sqlite3"
Let's try the :memory: db for testing
2015-08-15 09:34:41 +00:00
config.Conf.DBPath = ":memory:"
Adding support for Mysql (#442) Thanks, @svigne1! Fixes #53
2016-11-19 16:37:22 +00:00
config.Conf.MigrationsPath = "../db/db_sqlite3/migrations/"
Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
err := Setup()
if err != nil {
c.Fatalf("Failed creating database: %v", err)
}
}
Add cleanup between tests for models
2016-07-15 11:28:11 +00:00
func (s *ModelsSuite) TearDownTest(c *check.C) {
// Clear database tables between each test. If new tables are
// used in this test suite they will need to be cleaned up here.
db.Delete(Group{})
db.Delete(Target{})
db.Delete(GroupTarget{})
db.Delete(SMTP{})
db.Delete(Page{})
Email refactoring (#878) The initial pass at refactoring the way we send emails.
2017-12-09 21:42:07 +00:00
db.Delete(Result{})
db.Delete(MailLog{})
db.Delete(Campaign{})
Add cleanup between tests for models
2016-07-15 11:28:11 +00:00
// Reset users table to default state.
db.Not("id", 1).Delete(User{})
db.Model(User{}).Update("username", "admin")
}
Email refactoring (#878) The initial pass at refactoring the way we send emails.
2017-12-09 21:42:07 +00:00
func (s *ModelsSuite) createCampaignDependencies(ch *check.C) Campaign {
group := Group{Name: "Test Group"}
group.Targets = []Target{
Target{Email: "test1@example.com", FirstName: "First", LastName: "Example"},
Target{Email: "test2@example.com", FirstName: "Second", LastName: "Example"},
}
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}
return c
}
func (s *ModelsSuite) createCampaign(ch *check.C) Campaign {
c := s.createCampaignDependencies(ch)
// Setup and "launch" our campaign
ch.Assert(PostCampaign(&c, c.UserId), check.Equals, nil)
return c
}
Remove redundant return nil statement in user.go (#921) Removing some redundant statements in user.go. Adding test cases for user models.
2018-01-12 00:11:57 +00:00
func (s *ModelsSuite) TestGetUserExists(c *check.C) {
Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
u, err := GetUser(1)
Added ability to delete campaign from campaign results - needs some cleanup, but it works
2015-02-21 07:27:32 +00:00
c.Assert(err, check.Equals, nil)
c.Assert(u.Username, check.Equals, "admin")
Added better testing with gocheck Fixed some typos leading to syntax errors TODO: Finish up gorm integration into templates
2014-03-27 02:42:07 +00:00
}
Refactor GetUserByUsername method not to suppress an error (#920) Also adding some other tests for the User models.
2018-01-12 00:37:38 +00:00
func (s *ModelsSuite) TestGetUserByUsernameWithExistingUser(c *check.C) {
u, err := GetUserByUsername("admin")
c.Assert(err, check.Equals, nil)
c.Assert(u.Username, check.Equals, "admin")
}
Remove redundant return nil statement in user.go (#921) Removing some redundant statements in user.go. Adding test cases for user models.
2018-01-12 00:11:57 +00:00
func (s *ModelsSuite) TestGetUserDoesNotExist(c *check.C) {
u, err := GetUser(100)
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
c.Assert(u.Username, check.Equals, "")
}
func (s *ModelsSuite) TestGetUserByAPIKeyWithExistingAPIKey(c *check.C) {
u, err := GetUser(1)
c.Assert(err, check.Equals, nil)
u, err = GetUserByAPIKey(u.ApiKey)
}
func (s *ModelsSuite) TestGetUserByAPIKeyWithNotExistingAPIKey(c *check.C) {
u, err := GetUser(1)
c.Assert(err, check.Equals, nil)
u, err = GetUserByAPIKey(u.ApiKey + "test")
Refactor GetUserByUsername method not to suppress an error (#920) Also adding some other tests for the User models.
2018-01-12 00:37:38 +00:00
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
c.Assert(u.Username, check.Equals, "")
}
func (s *ModelsSuite) TestGetUserByUsernameWithNotExistingUser(c *check.C) {
u, err := GetUserByUsername("test user does not exist")
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
Remove redundant return nil statement in user.go (#921) Removing some redundant statements in user.go. Adding test cases for user models.
2018-01-12 00:11:57 +00:00
c.Assert(u.Username, check.Equals, "")
}
Put User tests together
2016-07-15 11:31:44 +00:00
func (s *ModelsSuite) TestPutUser(c *check.C) {
u, err := GetUser(1)
u.Username = "admin_changed"
err = PutUser(&u)
c.Assert(err, check.Equals, nil)
u, err = GetUser(1)
c.Assert(u.Username, check.Equals, "admin_changed")
}
Added test for #60
2016-01-13 02:52:13 +00:00
func (s *ModelsSuite) TestGeneratedAPIKey(c *check.C) {
u, err := GetUser(1)
c.Assert(err, check.Equals, nil)
c.Assert(u.ApiKey, check.Not(check.Equals), "12345678901234567890123456789012")
}
Added ability to delete campaign from campaign results - needs some cleanup, but it works
2015-02-21 07:27:32 +00:00
func (s *ModelsSuite) TestPostGroup(c *check.C) {
g := Group{Name: "Test Group"}
g.Targets = []Target{Target{Email: "test@example.com"}}
g.UserId = 1
err := PostGroup(&g)
c.Assert(err, check.Equals, nil)
c.Assert(g.Name, check.Equals, "Test Group")
c.Assert(g.Targets[0].Email, check.Equals, "test@example.com")
}
func (s *ModelsSuite) TestPostGroupNoName(c *check.C) {
g := Group{Name: ""}
g.Targets = []Target{Target{Email: "test@example.com"}}
g.UserId = 1
err := PostGroup(&g)
c.Assert(err, check.Equals, ErrGroupNameNotSpecified)
}
func (s *ModelsSuite) TestPostGroupNoTargets(c *check.C) {
g := Group{Name: "No Target Group"}
g.Targets = []Target{}
g.UserId = 1
err := PostGroup(&g)
c.Assert(err, check.Equals, ErrNoTargetsSpecified)
}
Add tests for getting group(s)
2016-07-16 11:53:56 +00:00
func (s *ModelsSuite) TestGetGroups(c *check.C) {
// Add groups.
PostGroup(&Group{
Name: "Test Group 1",
Targets: []Target{Target{Email: "test1@example.com"}},
UserId: 1,
})
PostGroup(&Group{
Name: "Test Group 2",
Targets: []Target{Target{Email: "test2@example.com"}},
UserId: 1,
})
// Get groups and test result.
groups, err := GetGroups(1)
c.Assert(err, check.Equals, nil)
c.Assert(len(groups), check.Equals, 2)
c.Assert(len(groups[0].Targets), check.Equals, 1)
c.Assert(len(groups[1].Targets), check.Equals, 1)
c.Assert(groups[0].Name, check.Equals, "Test Group 1")
c.Assert(groups[1].Name, check.Equals, "Test Group 2")
c.Assert(groups[0].Targets[0].Email, check.Equals, "test1@example.com")
c.Assert(groups[1].Targets[0].Email, check.Equals, "test2@example.com")
}
Add more tests for getting group(s)
2016-07-16 12:38:27 +00:00
func (s *ModelsSuite) TestGetGroupsNoGroups(c *check.C) {
groups, err := GetGroups(1)
c.Assert(err, check.Equals, nil)
c.Assert(len(groups), check.Equals, 0)
}
Add tests for getting group(s)
2016-07-16 11:53:56 +00:00
func (s *ModelsSuite) TestGetGroup(c *check.C) {
// Add group.
Email refactoring (#878) The initial pass at refactoring the way we send emails.
2017-12-09 21:42:07 +00:00
originalGroup := &Group{
Add tests for getting group(s)
2016-07-16 11:53:56 +00:00
Name: "Test Group",
Targets: []Target{Target{Email: "test@example.com"}},
UserId: 1,
Email refactoring (#878) The initial pass at refactoring the way we send emails.
2017-12-09 21:42:07 +00:00
}
c.Assert(PostGroup(originalGroup), check.Equals, nil)
Add tests for getting group(s)
2016-07-16 11:53:56 +00:00
// Get group and test result.
Email refactoring (#878) The initial pass at refactoring the way we send emails.
2017-12-09 21:42:07 +00:00
group, err := GetGroup(originalGroup.Id, 1)
Add tests for getting group(s)
2016-07-16 11:53:56 +00:00
c.Assert(err, check.Equals, nil)
c.Assert(len(group.Targets), check.Equals, 1)
c.Assert(group.Name, check.Equals, "Test Group")
c.Assert(group.Targets[0].Email, check.Equals, "test@example.com")
}
Add more tests for getting group(s)
2016-07-16 12:38:27 +00:00
func (s *ModelsSuite) TestGetGroupNoGroups(c *check.C) {
Check for specific error in group tests
2016-07-18 22:47:28 +00:00
_, err := GetGroup(1, 1)
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
Add more tests for getting group(s)
2016-07-16 12:38:27 +00:00
}
Add tests for getting group(s)
2016-07-16 11:53:56 +00:00
func (s *ModelsSuite) TestGetGroupByName(c *check.C) {
// Add group.
PostGroup(&Group{
Name: "Test Group",
Targets: []Target{Target{Email: "test@example.com"}},
UserId: 1,
})
// Get group and test result.
group, err := GetGroupByName("Test Group", 1)
c.Assert(err, check.Equals, nil)
c.Assert(len(group.Targets), check.Equals, 1)
c.Assert(group.Name, check.Equals, "Test Group")
c.Assert(group.Targets[0].Email, check.Equals, "test@example.com")
}
Add more tests for getting group(s)
2016-07-16 12:38:27 +00:00
func (s *ModelsSuite) TestGetGroupByNameNoGroups(c *check.C) {
Check for specific error in group tests
2016-07-18 22:47:28 +00:00
_, err := GetGroupByName("Test Group", 1)
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
Add more tests for getting group(s)
2016-07-16 12:38:27 +00:00
}
Add test for PutGroup function
2016-07-02 22:22:44 +00:00
func (s *ModelsSuite) TestPutGroup(c *check.C) {
// Add test group.
Use generic names for Group tests
2016-07-15 11:37:16 +00:00
group := Group{Name: "Test Group"}
Add test for PutGroup function
2016-07-02 22:22:44 +00:00
group.Targets = []Target{
Target{Email: "test1@example.com", FirstName: "First", LastName: "Example"},
Target{Email: "test2@example.com", FirstName: "Second", LastName: "Example"},
}
group.UserId = 1
PostGroup(&group)
// Update one of group's targets.
Use generic names for Group tests
2016-07-15 11:37:16 +00:00
group.Targets[0].FirstName = "Updated"
Add test for PutGroup function
2016-07-02 22:22:44 +00:00
err := PutGroup(&group)
c.Assert(err, check.Equals, nil)
// Verify updated target information.
targets, _ := GetTargets(group.Id)
c.Assert(targets[0].Email, check.Equals, "test1@example.com")
Use generic names for Group tests
2016-07-15 11:37:16 +00:00
c.Assert(targets[0].FirstName, check.Equals, "Updated")
Add test for PutGroup function
2016-07-02 22:22:44 +00:00
c.Assert(targets[0].LastName, check.Equals, "Example")
c.Assert(targets[1].Email, check.Equals, "test2@example.com")
c.Assert(targets[1].FirstName, check.Equals, "Second")
c.Assert(targets[1].LastName, check.Equals, "Example")
}
Add empty attribute test for PutGroup If passing a Target instance to the database Updates method then empty attributes will be ignored. Using a map[string]interface ensures all changed attributes, even empty ones, are persisted to database.
2016-07-04 10:39:38 +00:00
func (s *ModelsSuite) TestPutGroupEmptyAttribute(c *check.C) {
// Add test group.
Use generic names for Group tests
2016-07-15 11:37:16 +00:00
group := Group{Name: "Test Group"}
Add empty attribute test for PutGroup If passing a Target instance to the database Updates method then empty attributes will be ignored. Using a map[string]interface ensures all changed attributes, even empty ones, are persisted to database.
2016-07-04 10:39:38 +00:00
group.Targets = []Target{
Use generic names for Group tests
2016-07-15 11:37:16 +00:00
Target{Email: "test1@example.com", FirstName: "First", LastName: "Example"},
Target{Email: "test2@example.com", FirstName: "Second", LastName: "Example"},
Add empty attribute test for PutGroup If passing a Target instance to the database Updates method then empty attributes will be ignored. Using a map[string]interface ensures all changed attributes, even empty ones, are persisted to database.
2016-07-04 10:39:38 +00:00
}
group.UserId = 1
PostGroup(&group)
// Update one of group's targets.
group.Targets[0].FirstName = ""
err := PutGroup(&group)
c.Assert(err, check.Equals, nil)
// Verify updated empty attribute was saved.
targets, _ := GetTargets(group.Id)
Use generic names for Group tests
2016-07-15 11:37:16 +00:00
c.Assert(targets[0].Email, check.Equals, "test1@example.com")
Add empty attribute test for PutGroup If passing a Target instance to the database Updates method then empty attributes will be ignored. Using a map[string]interface ensures all changed attributes, even empty ones, are persisted to database.
2016-07-04 10:39:38 +00:00
c.Assert(targets[0].FirstName, check.Equals, "")
c.Assert(targets[0].LastName, check.Equals, "Example")
Use generic names for Group tests
2016-07-15 11:37:16 +00:00
c.Assert(targets[1].Email, check.Equals, "test2@example.com")
c.Assert(targets[1].FirstName, check.Equals, "Second")
Add empty attribute test for PutGroup If passing a Target instance to the database Updates method then empty attributes will be ignored. Using a map[string]interface ensures all changed attributes, even empty ones, are persisted to database.
2016-07-04 10:39:38 +00:00
c.Assert(targets[1].LastName, check.Equals, "Example")
}
Added tests for SMTP
2016-02-22 04:13:06 +00:00
func (s *ModelsSuite) TestPostSMTP(c *check.C) {
smtp := SMTP{
Name: "Test SMTP",
Host: "1.1.1.1:25",
FromAddress: "Foo Bar <foo@example.com>",
UserId: 1,
}
err = PostSMTP(&smtp)
c.Assert(err, check.Equals, nil)
ss, err := GetSMTPs(1)
c.Assert(err, check.Equals, nil)
c.Assert(len(ss), check.Equals, 1)
}
func (s *ModelsSuite) TestPostSMTPNoHost(c *check.C) {
smtp := SMTP{
Name: "Test SMTP",
FromAddress: "Foo Bar <foo@example.com>",
UserId: 1,
}
err = PostSMTP(&smtp)
c.Assert(err, check.Equals, ErrHostNotSpecified)
}
func (s *ModelsSuite) TestPostSMTPNoFrom(c *check.C) {
smtp := SMTP{
Name: "Test SMTP",
UserId: 1,
Host: "1.1.1.1:25",
}
err = PostSMTP(&smtp)
c.Assert(err, check.Equals, ErrFromAddressNotSpecified)
}
Adding support for custom headers in sending profiles (#544) Closes #215 Closes #128
2017-02-20 00:43:08 +00:00
func (s *ModelsSuite) TestPostSMTPValidHeader(c *check.C) {
smtp := SMTP{
Name: "Test SMTP",
Host: "1.1.1.1:25",
FromAddress: "Foo Bar <foo@example.com>",
UserId: 1,
Headers: []Header{
Header{Key: "Reply-To", Value: "test@example.com"},
Header{Key: "X-Mailer", Value: "gophish"},
},
}
err = PostSMTP(&smtp)
c.Assert(err, check.Equals, nil)
ss, err := GetSMTPs(1)
c.Assert(err, check.Equals, nil)
c.Assert(len(ss), check.Equals, 1)
}
Adding tests
2016-02-26 01:58:02 +00:00
func (s *ModelsSuite) TestPostPage(c *check.C) {
html := `<html>
<head></head>
<body><form action="example.com">
<input name="username"/>
<input name="password" type="password"/>
</form></body>
</html>`
p := Page{
Added support for redirect URL's after creds are submitted. Fixes #210
2016-03-19 01:19:13 +00:00
Name: "Test Page",
HTML: html,
RedirectURL: "http://example.com",
Adding tests
2016-02-26 01:58:02 +00:00
}
// Check the capturing credentials and passwords
p.CaptureCredentials = true
p.CapturePasswords = true
err := PostPage(&p)
c.Assert(err, check.Equals, nil)
Added support for redirect URL's after creds are submitted. Fixes #210
2016-03-19 01:19:13 +00:00
c.Assert(p.RedirectURL, check.Equals, "http://example.com")
Adding tests
2016-02-26 01:58:02 +00:00
d, err := goquery.NewDocumentFromReader(strings.NewReader(p.HTML))
c.Assert(err, check.Equals, nil)
forms := d.Find("form")
forms.Each(func(i int, f *goquery.Selection) {
// Check the action has been set
a, _ := f.Attr("action")
c.Assert(a, check.Equals, "")
// Check the password still has a name
_, ok := f.Find("input[type=\"password\"]").Attr("name")
c.Assert(ok, check.Equals, true)
// Check the username is still correct
u, ok := f.Find("input").Attr("name")
c.Assert(ok, check.Equals, true)
c.Assert(u, check.Equals, "username")
})
// Check what happens when we don't capture passwords
p.CapturePasswords = false
p.HTML = html
Added support for redirect URL's after creds are submitted. Fixes #210
2016-03-19 01:19:13 +00:00
p.RedirectURL = ""
Adding tests
2016-02-26 01:58:02 +00:00
err = PutPage(&p)
c.Assert(err, check.Equals, nil)
Added support for redirect URL's after creds are submitted. Fixes #210
2016-03-19 01:19:13 +00:00
c.Assert(p.RedirectURL, check.Equals, "")
Adding tests
2016-02-26 01:58:02 +00:00
d, err = goquery.NewDocumentFromReader(strings.NewReader(p.HTML))
c.Assert(err, check.Equals, nil)
forms = d.Find("form")
forms.Each(func(i int, f *goquery.Selection) {
// Check the action has been set
a, _ := f.Attr("action")
c.Assert(a, check.Equals, "")
// Check the password still has a name
_, ok := f.Find("input[type=\"password\"]").Attr("name")
c.Assert(ok, check.Equals, false)
// Check the username is still correct
u, ok := f.Find("input").Attr("name")
c.Assert(ok, check.Equals, true)
c.Assert(u, check.Equals, "username")
})
// Finally, check when we don't capture credentials
p.CaptureCredentials = false
p.HTML = html
err = PutPage(&p)
c.Assert(err, check.Equals, nil)
d, err = goquery.NewDocumentFromReader(strings.NewReader(p.HTML))
c.Assert(err, check.Equals, nil)
forms = d.Find("form")
forms.Each(func(i int, f *goquery.Selection) {
// Check the action has been set
a, _ := f.Attr("action")
c.Assert(a, check.Equals, "")
// Check the password still has a name
_, ok := f.Find("input[type=\"password\"]").Attr("name")
c.Assert(ok, check.Equals, false)
// Check the username is still correct
_, ok = f.Find("input").Attr("name")
c.Assert(ok, check.Equals, false)
})
}
Changed result ID's to be 7 random alphanumeric characters. Fixes #518
2017-01-28 03:16:47 +00:00
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)
}
Phishing emails are now sent in "First Last <email@domain.com>" format.
2017-06-17 03:21:08 +00:00
func (s *ModelsSuite) TestFormatAddress(c *check.C) {
r := Result{
FirstName: "John",
LastName: "Doe",
Email: "johndoe@example.com",
}
expected := &mail.Address{
Name: "John Doe",
Address: "johndoe@example.com",
}
c.Assert(r.FormatAddress(), check.Equals, expected.String())
r = Result{
Email: "johndoe@example.com",
}
c.Assert(r.FormatAddress(), check.Equals, r.Email)
}
Email refactoring (#878) The initial pass at refactoring the way we send emails.
2017-12-09 21:42:07 +00:00
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
fmt.Println("Campaign STATUS")
fmt.Println(c.Status)
for _, r := range c.Results {
ch.Assert(r.Status, check.Equals, STATUS_SENDING)
}
}
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)
}
}
Removing duplicates in campaigns. Fixes #855
2017-12-12 04:12:17 +00:00
func (s *ModelsSuite) TestDuplicateResults(ch *check.C) {
group := Group{Name: "Test Group"}
group.Targets = []Target{
Target{Email: "test1@example.com", FirstName: "First", LastName: "Example"},
Target{Email: "test1@example.com", FirstName: "Duplicate", LastName: "Duplicate"},
Target{Email: "test2@example.com", FirstName: "Second", LastName: "Example"},
}
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)
}