140 lines
4.4 KiB
Go
140 lines
4.4 KiB
Go
package persontest
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
person "example.com/database/person"
|
|
testrequest "example.com/test/testrequest"
|
|
testsetup "example.com/test/testsetup"
|
|
grouptest "example.com/unit/grouptest"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type TestPerson struct {
|
|
Result []person.Person `json:"result"`
|
|
}
|
|
|
|
type PersonTestSuite struct {
|
|
suite.Suite
|
|
globalAuthHeader string
|
|
router *gin.Engine
|
|
person person.Person
|
|
personJSON []byte
|
|
personUpdate person.Person
|
|
personUpdateJSON []byte
|
|
}
|
|
|
|
func (s *PersonTestSuite) SetupTestSuite() {
|
|
s.globalAuthHeader, s.router = testsetup.SetupTestSuite()
|
|
// Create Test Data
|
|
fixtures, _ := os.ReadFile("fixtures/persons.json")
|
|
var personFixtures struct {
|
|
Persons []person.Person `json:"persons"`
|
|
PersonUpdate person.Person `json:"person_update"`
|
|
}
|
|
json.Unmarshal(fixtures, &personFixtures)
|
|
s.person = personFixtures.Persons[0]
|
|
var personGroupsIDs []uint
|
|
for _, group := range s.person.Groups {
|
|
personGroupsIDs = append(personGroupsIDs, group.ID)
|
|
}
|
|
s.personJSON, _ = json.Marshal(person.PersonParams{
|
|
Name: s.person.Name,
|
|
Groups: personGroupsIDs,
|
|
})
|
|
s.personUpdate = personFixtures.PersonUpdate
|
|
var personUpdateGroupsIDs []uint
|
|
for _, group := range s.personUpdate.Groups {
|
|
personUpdateGroupsIDs = append(personUpdateGroupsIDs, group.ID)
|
|
}
|
|
s.personUpdateJSON, _ = json.Marshal(person.PersonParams{
|
|
Name: s.personUpdate.Name,
|
|
Groups: personUpdateGroupsIDs,
|
|
})
|
|
}
|
|
|
|
func (s *PersonTestSuite) TearDownTestSuite() {
|
|
os.Remove("db/main.db")
|
|
}
|
|
|
|
func (s *PersonTestSuite) Test01CreatePerson() {
|
|
// Setup variables
|
|
var output TestPerson
|
|
target := fmt.Sprintf("/person?id=%d", s.person.ID)
|
|
|
|
// Attempt to create person
|
|
status := testrequest.MakePostRequest(&s.personJSON, "/person", s.globalAuthHeader, s.router)
|
|
// Check that the request was successful
|
|
s.Equal(status, "200 OK", "person created successfully")
|
|
|
|
// Attempt to get person
|
|
body := testrequest.MakeGetRequest(target, s.router)
|
|
// Unmarshal the result
|
|
json.Unmarshal(body, &output)
|
|
fmt.Println(string(body))
|
|
// Check that the person listed matches the created person
|
|
s.Equal(s.person.Name, output.Result[0].Name, "input name matches output name")
|
|
}
|
|
|
|
func (s *PersonTestSuite) Test02GetPerson() {
|
|
// Setup variables
|
|
var output TestPerson
|
|
|
|
// Attempt to get persons
|
|
body := testrequest.MakeGetRequest("/person", s.router)
|
|
// Unmarshal the result
|
|
json.Unmarshal(body, &output)
|
|
// Check that the persons listed match the created persons
|
|
s.Equal(s.person.Name, output.Result[1].Name, "input name matches output name")
|
|
}
|
|
|
|
func (s *PersonTestSuite) Test03UpdatePerson() {
|
|
// Setup variables
|
|
var output TestPerson
|
|
target := fmt.Sprintf("/person?id=%d", s.person.ID)
|
|
|
|
// Create needed field objects
|
|
var groupOutput grouptest.TestGroup
|
|
for _, group := range s.personUpdate.Groups {
|
|
groupBody, _ := json.Marshal(group)
|
|
groupStatus := testrequest.MakePostRequest(&groupBody, "/group", s.globalAuthHeader, s.router)
|
|
s.Equal(groupStatus, "200 OK", "group created successfully")
|
|
groupTarget := fmt.Sprintf("/group?id=%d", group.ID)
|
|
groupBody = testrequest.MakeGetRequest(groupTarget, s.router)
|
|
json.Unmarshal(groupBody, &groupOutput)
|
|
s.Equal(group.Name, groupOutput.Result[0].Name, "input group name matches output group name")
|
|
}
|
|
|
|
// Attempt to update person
|
|
status := testrequest.MakePutRequest(&s.personUpdateJSON, target, s.globalAuthHeader, s.router)
|
|
// Check that the request was successful
|
|
s.Equal(status, "200 OK", "person created successfully")
|
|
|
|
// Attempt to get person
|
|
body := testrequest.MakeGetRequest(target, s.router)
|
|
// Unmarshal the result
|
|
json.Unmarshal(body, &output)
|
|
// Check that the person listed matches the updated person
|
|
s.Equal(s.personUpdate.Name, output.Result[0].Name, "input name matches output name")
|
|
}
|
|
|
|
func (s *PersonTestSuite) Test04DeletePerson() {
|
|
// Setup variables
|
|
var output TestPerson
|
|
target := fmt.Sprintf("/person?id=%d", s.person.ID)
|
|
|
|
// Attempt to delete person
|
|
status := testrequest.MakeDeleteRequest(target, s.globalAuthHeader, s.router)
|
|
// Check that the request was successful
|
|
s.Equal(status, "200 OK", "person deleted successfully")
|
|
|
|
// Attempt to get person
|
|
body := testrequest.MakeGetRequest(target, s.router)
|
|
// Unmarshal the result
|
|
json.Unmarshal(body, &output)
|
|
// Check that the person list is empty
|
|
s.Equal(1, len(output.Result), "person list is empty")
|
|
}
|