89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
package server
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.javil.eu/jacob1123/budgeteer/bcrypt"
|
|
"git.javil.eu/jacob1123/budgeteer/jwt"
|
|
"git.javil.eu/jacob1123/budgeteer/postgres"
|
|
txdb "github.com/DATA-DOG/go-txdb"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func init() { //nolint:gochecknoinits
|
|
txdb.Register("pgtx", "pgx", "postgres://budgeteer_test:budgeteer_test@localhost:5432/budgeteer_test")
|
|
}
|
|
|
|
func TestRegisterUser(t *testing.T) { //nolint:funlen
|
|
t.Parallel()
|
|
database, err := postgres.Connect("pgtx", "example")
|
|
if err != nil {
|
|
fmt.Printf("could not connect to db: %s\n", err)
|
|
t.Skip()
|
|
return
|
|
}
|
|
|
|
h := Handler{
|
|
Service: database,
|
|
TokenVerifier: &jwt.TokenVerifier{
|
|
Secret: "this_is_my_demo_secret_for_unit_tests",
|
|
},
|
|
CredentialsVerifier: &bcrypt.Verifier{},
|
|
}
|
|
|
|
recorder := httptest.NewRecorder()
|
|
context, engine := gin.CreateTestContext(recorder)
|
|
h.LoadRoutes(engine)
|
|
|
|
t.Run("RegisterUser", func(t *testing.T) {
|
|
t.Parallel()
|
|
context.Request, err = http.NewRequest(
|
|
http.MethodPost,
|
|
"/api/v1/user/register",
|
|
strings.NewReader(`{"password":"pass","email":"info@example.com","name":"Test"}`))
|
|
if err != nil {
|
|
t.Errorf("error creating request: %s", err)
|
|
return
|
|
}
|
|
|
|
h.registerPost(context)
|
|
|
|
if recorder.Code != http.StatusOK {
|
|
t.Errorf("handler returned wrong status code: got %v want %v", recorder.Code, http.StatusOK)
|
|
}
|
|
|
|
var response LoginResponse
|
|
err = json.NewDecoder(recorder.Body).Decode(&response)
|
|
if err != nil {
|
|
t.Error(err.Error())
|
|
t.Error("Error registering")
|
|
}
|
|
if len(response.Token) == 0 {
|
|
t.Error("Did not get a token")
|
|
}
|
|
})
|
|
|
|
t.Run("GetTransactions", func(t *testing.T) {
|
|
t.Parallel()
|
|
context.Request, err = http.NewRequest(http.MethodGet, "/account/accountid/transactions", nil)
|
|
if recorder.Code != http.StatusOK {
|
|
t.Errorf("handler returned wrong status code: got %v want %v", recorder.Code, http.StatusOK)
|
|
}
|
|
|
|
var response TransactionsResponse
|
|
err = json.NewDecoder(recorder.Body).Decode(&response)
|
|
if err != nil {
|
|
t.Error(err.Error())
|
|
t.Error("Error retreiving list of transactions.")
|
|
}
|
|
if len(response.Transactions) == 0 {
|
|
t.Error("Did not get any transactions.")
|
|
}
|
|
})
|
|
}
|