mirror of https://github.com/gophish/gophish
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 testspull/961/head
Shuhei Kitagawa
Jordan Wright
parent
55b0da4aa9
commit
44afb2b965
|
|
@ -0,0 +1,156 @@
|
||||||
|
package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/jinzhu/gorm"
|
||||||
|
"gopkg.in/check.v1"
|
||||||
|
)
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|
||||||
|
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")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ModelsSuite) TestGetGroupsNoGroups(c *check.C) {
|
||||||
|
groups, err := GetGroups(1)
|
||||||
|
c.Assert(err, check.Equals, nil)
|
||||||
|
c.Assert(len(groups), check.Equals, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ModelsSuite) TestGetGroup(c *check.C) {
|
||||||
|
// Add group.
|
||||||
|
originalGroup := &Group{
|
||||||
|
Name: "Test Group",
|
||||||
|
Targets: []Target{Target{Email: "test@example.com"}},
|
||||||
|
UserId: 1,
|
||||||
|
}
|
||||||
|
c.Assert(PostGroup(originalGroup), check.Equals, nil)
|
||||||
|
|
||||||
|
// Get group and test result.
|
||||||
|
group, err := GetGroup(originalGroup.Id, 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")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ModelsSuite) TestGetGroupNoGroups(c *check.C) {
|
||||||
|
_, err := GetGroup(1, 1)
|
||||||
|
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
|
||||||
|
}
|
||||||
|
|
||||||
|
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")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ModelsSuite) TestGetGroupByNameNoGroups(c *check.C) {
|
||||||
|
_, err := GetGroupByName("Test Group", 1)
|
||||||
|
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ModelsSuite) TestPutGroup(c *check.C) {
|
||||||
|
// Add test group.
|
||||||
|
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
|
||||||
|
PostGroup(&group)
|
||||||
|
|
||||||
|
// Update one of group's targets.
|
||||||
|
group.Targets[0].FirstName = "Updated"
|
||||||
|
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")
|
||||||
|
c.Assert(targets[0].FirstName, check.Equals, "Updated")
|
||||||
|
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")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ModelsSuite) TestPutGroupEmptyAttribute(c *check.C) {
|
||||||
|
// Add test group.
|
||||||
|
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
|
||||||
|
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)
|
||||||
|
c.Assert(targets[0].Email, check.Equals, "test1@example.com")
|
||||||
|
c.Assert(targets[0].FirstName, check.Equals, "")
|
||||||
|
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")
|
||||||
|
}
|
||||||
|
|
@ -1,16 +1,9 @@
|
||||||
package models
|
package models
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"net/mail"
|
|
||||||
"regexp"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/PuerkitoBio/goquery"
|
|
||||||
"github.com/gophish/gophish/config"
|
"github.com/gophish/gophish/config"
|
||||||
"github.com/jinzhu/gorm"
|
|
||||||
"gopkg.in/check.v1"
|
"gopkg.in/check.v1"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -93,374 +86,3 @@ func (s *ModelsSuite) createCampaign(ch *check.C) Campaign {
|
||||||
ch.Assert(PostCampaign(&c, c.UserId), check.Equals, nil)
|
ch.Assert(PostCampaign(&c, c.UserId), check.Equals, nil)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
|
|
||||||
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")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ModelsSuite) TestGetGroupsNoGroups(c *check.C) {
|
|
||||||
groups, err := GetGroups(1)
|
|
||||||
c.Assert(err, check.Equals, nil)
|
|
||||||
c.Assert(len(groups), check.Equals, 0)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ModelsSuite) TestGetGroup(c *check.C) {
|
|
||||||
// Add group.
|
|
||||||
originalGroup := &Group{
|
|
||||||
Name: "Test Group",
|
|
||||||
Targets: []Target{Target{Email: "test@example.com"}},
|
|
||||||
UserId: 1,
|
|
||||||
}
|
|
||||||
c.Assert(PostGroup(originalGroup), check.Equals, nil)
|
|
||||||
|
|
||||||
// Get group and test result.
|
|
||||||
group, err := GetGroup(originalGroup.Id, 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")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ModelsSuite) TestGetGroupNoGroups(c *check.C) {
|
|
||||||
_, err := GetGroup(1, 1)
|
|
||||||
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
|
|
||||||
}
|
|
||||||
|
|
||||||
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")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ModelsSuite) TestGetGroupByNameNoGroups(c *check.C) {
|
|
||||||
_, err := GetGroupByName("Test Group", 1)
|
|
||||||
c.Assert(err, check.Equals, gorm.ErrRecordNotFound)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ModelsSuite) TestPutGroup(c *check.C) {
|
|
||||||
// Add test group.
|
|
||||||
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
|
|
||||||
PostGroup(&group)
|
|
||||||
|
|
||||||
// Update one of group's targets.
|
|
||||||
group.Targets[0].FirstName = "Updated"
|
|
||||||
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")
|
|
||||||
c.Assert(targets[0].FirstName, check.Equals, "Updated")
|
|
||||||
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")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ModelsSuite) TestPutGroupEmptyAttribute(c *check.C) {
|
|
||||||
// Add test group.
|
|
||||||
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
|
|
||||||
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)
|
|
||||||
c.Assert(targets[0].Email, check.Equals, "test1@example.com")
|
|
||||||
c.Assert(targets[0].FirstName, check.Equals, "")
|
|
||||||
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")
|
|
||||||
}
|
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
|
|
||||||
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{
|
|
||||||
Name: "Test Page",
|
|
||||||
HTML: html,
|
|
||||||
RedirectURL: "http://example.com",
|
|
||||||
}
|
|
||||||
// Check the capturing credentials and passwords
|
|
||||||
p.CaptureCredentials = true
|
|
||||||
p.CapturePasswords = true
|
|
||||||
err := PostPage(&p)
|
|
||||||
c.Assert(err, check.Equals, nil)
|
|
||||||
c.Assert(p.RedirectURL, check.Equals, "http://example.com")
|
|
||||||
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
|
|
||||||
p.RedirectURL = ""
|
|
||||||
err = PutPage(&p)
|
|
||||||
c.Assert(err, check.Equals, nil)
|
|
||||||
c.Assert(p.RedirectURL, check.Equals, "")
|
|
||||||
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)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
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{
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,85 @@
|
||||||
|
package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/PuerkitoBio/goquery"
|
||||||
|
"gopkg.in/check.v1"
|
||||||
|
)
|
||||||
|
|
||||||
|
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{
|
||||||
|
Name: "Test Page",
|
||||||
|
HTML: html,
|
||||||
|
RedirectURL: "http://example.com",
|
||||||
|
}
|
||||||
|
// Check the capturing credentials and passwords
|
||||||
|
p.CaptureCredentials = true
|
||||||
|
p.CapturePasswords = true
|
||||||
|
err := PostPage(&p)
|
||||||
|
c.Assert(err, check.Equals, nil)
|
||||||
|
c.Assert(p.RedirectURL, check.Equals, "http://example.com")
|
||||||
|
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
|
||||||
|
p.RedirectURL = ""
|
||||||
|
err = PutPage(&p)
|
||||||
|
c.Assert(err, check.Equals, nil)
|
||||||
|
c.Assert(p.RedirectURL, check.Equals, "")
|
||||||
|
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)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,102 @@
|
||||||
|
package models
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"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{
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|
@ -6,6 +6,58 @@ import (
|
||||||
check "gopkg.in/check.v1"
|
check "gopkg.in/check.v1"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|
||||||
func (s *ModelsSuite) TestSMTPGetDialer(ch *check.C) {
|
func (s *ModelsSuite) TestSMTPGetDialer(ch *check.C) {
|
||||||
host := "localhost"
|
host := "localhost"
|
||||||
port := 25
|
port := 25
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue