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

Add more tests for getting group(s)

pull/324/head
Rob Cutmore 2016-07-16 08:38:27 -04:00
parent 3a9bc91415
commit f9a15cfe3d
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
models/models_test.go
View File

@ -111,6 +111,12 @@ func (s *ModelsSuite) TestGetGroups(c *check.C) {
c.Assert(groups[1].Targets[0].Email, check.Equals, "test2@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) { func (s *ModelsSuite) TestGetGroup(c *check.C) {
// Add group. // Add group.
PostGroup(&Group{ PostGroup(&Group{
@ -127,6 +133,12 @@ func (s *ModelsSuite) TestGetGroup(c *check.C) {
c.Assert(group.Targets[0].Email, check.Equals, "test@example.com") c.Assert(group.Targets[0].Email, check.Equals, "test@example.com")
} }
func (s *ModelsSuite) TestGetGroupNoGroups(c *check.C) {
group, err := GetGroup(1, 1)
c.Assert(err, check.Not(check.Equals), nil)
c.Assert(group.Name, check.Not(check.Equals), "Test Group")
}
func (s *ModelsSuite) TestGetGroupByName(c *check.C) { func (s *ModelsSuite) TestGetGroupByName(c *check.C) {
// Add group. // Add group.
PostGroup(&Group{ PostGroup(&Group{
@ -143,6 +155,12 @@ func (s *ModelsSuite) TestGetGroupByName(c *check.C) {
c.Assert(group.Targets[0].Email, check.Equals, "test@example.com") c.Assert(group.Targets[0].Email, check.Equals, "test@example.com")
} }
func (s *ModelsSuite) TestGetGroupByNameNoGroups(c *check.C) {
group, err := GetGroupByName("Test Group", 1)
c.Assert(err, check.Not(check.Equals), nil)
c.Assert(group.Name, check.Not(check.Equals), "Test Group")
}
func (s *ModelsSuite) TestPutGroup(c *check.C) { func (s *ModelsSuite) TestPutGroup(c *check.C) {
// Add test group. // Add test group.
group := Group{Name: "Test Group"} group := Group{Name: "Test Group"}