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

185 lines
5.1 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 (
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"
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"
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) {
Let's try the :memory: db for testing
2015-08-15 09:34:41 +00:00
config.Conf.DBPath = ":memory:"
Adding first round of database migrations using goose
2016-01-19 03:13:32 +00:00
config.Conf.MigrationsPath = "../db/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)
}
}
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) TestGetUser(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
}
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)
}
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 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)
})
}
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) TestPutUser(c *check.C) {
Integrated gorm with templates - Gorm integration complete Updated groups to belong to one user. I may make a Team and TeamGroups, and TeamUsers m2m relationships later. Added another test - more on the way soon.
2014-03-27 18:19:57 +00:00
u, err := GetUser(1)
u.Username = "admin_changed"
err = PutUser(&u)
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)
Integrated gorm with templates - Gorm integration complete Updated groups to belong to one user. I may make a Team and TeamGroups, and TeamUsers m2m relationships later. Added another test - more on the way soon.
2014-03-27 18:19:57 +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(u.Username, check.Equals, "admin_changed")
Integrated gorm with templates - Gorm integration complete Updated groups to belong to one user. I may make a Team and TeamGroups, and TeamUsers m2m relationships later. Added another test - more on the way soon.
2014-03-27 18:19:57 +00:00
}