67 lines
2.0 KiB
Go
67 lines
2.0 KiB
Go
|
package tripsgo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"strconv"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/gorilla/mux"
|
||
|
|
"github.com/joho/godotenv"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
// APITestCase needs to be exported to be accessed for test dir
|
||
|
|
type APITestCase struct {
|
||
|
|
Tag string
|
||
|
|
Method string
|
||
|
|
URL string
|
||
|
|
Body string
|
||
|
|
Status int
|
||
|
|
ExpectedResponse string
|
||
|
|
ActualResponse string
|
||
|
|
}
|
||
|
|
|
||
|
|
func testAPI(router *mux.Router, method, URL, body string) *httptest.ResponseRecorder {
|
||
|
|
req, _ := http.NewRequest(method, URL, bytes.NewBufferString(body))
|
||
|
|
req.Header.Set("Content-Type", "application/json")
|
||
|
|
res := httptest.NewRecorder()
|
||
|
|
router.ServeHTTP(res, req)
|
||
|
|
return res
|
||
|
|
}
|
||
|
|
|
||
|
|
// RunAPITests needs to be exported to be accessed for test dir
|
||
|
|
func RunAPITests(t *testing.T, router *mux.Router, tests []APITestCase) {
|
||
|
|
for i := 0; i < len(tests); i++ {
|
||
|
|
res := testAPI(router, tests[i].Method, tests[i].URL, tests[i].Body)
|
||
|
|
tests[i].ActualResponse = res.Body.String()
|
||
|
|
Debug.Println(tests[i].Tag + " - " + tests[i].ActualResponse)
|
||
|
|
assert.Equal(t, tests[i].Status, res.Code, tests[i].Tag)
|
||
|
|
Info.Println(tests[i].Tag + "- Response Code:" + strconv.Itoa(res.Code))
|
||
|
|
if tests[i].ExpectedResponse != "" {
|
||
|
|
assert.JSONEq(t, tests[i].ExpectedResponse, res.Body.String(), tests[i].Tag)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func RunAPITestsPlainText(t *testing.T, router *mux.Router, tests []APITestCase) {
|
||
|
|
for i := 0; i < len(tests); i++ {
|
||
|
|
res := testAPI(router, tests[i].Method, tests[i].URL, tests[i].Body)
|
||
|
|
tests[i].ActualResponse = res.Body.String()
|
||
|
|
Debug.Println(tests[i].Tag + " - " + tests[i].ActualResponse)
|
||
|
|
assert.Equal(t, tests[i].Status, res.Code, tests[i].Tag)
|
||
|
|
Info.Println(tests[i].Tag + "- Response Code:" + strconv.Itoa(res.Code))
|
||
|
|
if tests[i].ExpectedResponse != "" {
|
||
|
|
assert.Equal(t, tests[i].ExpectedResponse, res.Body.String(), tests[i].Tag)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func resetDataAccessEnvVars() {
|
||
|
|
var fls bool = false
|
||
|
|
debug = &fls
|
||
|
|
godotenv.Overload()
|
||
|
|
RebindDataAccessEnvironmentVariables()
|
||
|
|
}
|