| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807 |
- package api_test
- import (
- "encoding/json"
- "net/http"
- "net/http/httptest"
- "reflect"
- "strings"
- "testing"
- "time"
- "github.com/go-chi/chi"
- "github.com/porter-dev/porter/internal/config"
- "github.com/porter-dev/porter/internal/models"
- "github.com/porter-dev/porter/internal/repository"
- "github.com/porter-dev/porter/internal/repository/test"
- "github.com/porter-dev/porter/server/api"
- "github.com/porter-dev/porter/server/router"
- sessionstore "github.com/porter-dev/porter/internal/auth"
- lr "github.com/porter-dev/porter/internal/logger"
- vr "github.com/porter-dev/porter/internal/validator"
- )
- type tester struct {
- app *api.App
- repo *repository.Repository
- store *sessionstore.PGStore
- router *chi.Mux
- req *http.Request
- rr *httptest.ResponseRecorder
- cookie *http.Cookie
- }
- type userTest struct {
- initializers []func(t *tester)
- msg string
- method string
- endpoint string
- body string
- expStatus int
- expBody string
- useCookie bool
- validators []func(c *userTest, tester *tester, t *testing.T)
- }
- func (t *tester) execute() {
- t.router.ServeHTTP(t.rr, t.req)
- }
- func (t *tester) reset() {
- t.rr = httptest.NewRecorder()
- t.req = nil
- }
- func (t *tester) createUserSession(email string, pw string) {
- req, _ := http.NewRequest(
- "POST",
- "/api/users",
- strings.NewReader(`{"email":"`+email+`","password":"`+pw+`"}`),
- )
- t.req = req
- t.execute()
- if cookies := t.rr.Result().Cookies(); len(cookies) > 0 {
- t.cookie = cookies[0]
- }
- t.reset()
- }
- func initUserDefault(tester *tester) {
- tester.createUserSession("belanger@getporter.dev", "hello")
- }
- func initUserWithClusters(tester *tester) {
- initUserDefault(tester)
- user, _ := tester.repo.User.ReadUserByEmail("belanger@getporter.dev")
- user.Clusters = []models.ClusterConfig{
- models.ClusterConfig{
- Name: "cluster-test",
- Server: "https://localhost",
- Context: "context-test",
- User: "test-admin",
- },
- }
- user.RawKubeConfig = []byte("apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin")
- tester.repo.User.UpdateUser(user)
- }
- func newTester(canQuery bool) *tester {
- appConf := config.Conf{
- Debug: true,
- Server: config.ServerConf{
- Port: 8080,
- CookieName: "porter",
- CookieSecret: []byte("secret"),
- TimeoutRead: time.Second * 5,
- TimeoutWrite: time.Second * 10,
- TimeoutIdle: time.Second * 15,
- },
- // unimportant here
- Db: config.DBConf{},
- }
- logger := lr.NewConsole(appConf.Debug)
- validator := vr.New()
- repo := test.NewRepository(canQuery)
- store, _ := sessionstore.NewStore(repo, appConf.Server)
- app := api.New(logger, repo, validator, store, appConf.Server.CookieName)
- r := router.New(app, store, appConf.Server.CookieName)
- return &tester{
- app: app,
- repo: repo,
- store: store,
- router: r,
- req: nil,
- rr: httptest.NewRecorder(),
- cookie: nil,
- }
- }
- func testUserRequests(t *testing.T, tests []*userTest, canQuery bool) {
- for _, c := range tests {
- // create a new tester
- tester := newTester(canQuery)
- // if there's an initializer, call it
- for _, init := range c.initializers {
- init(tester)
- }
- req, err := http.NewRequest(
- c.method,
- c.endpoint,
- strings.NewReader(c.body),
- )
- tester.req = req
- if c.useCookie {
- req.AddCookie(tester.cookie)
- }
- if err != nil {
- t.Fatal(err)
- }
- tester.execute()
- rr := tester.rr
- // first, check that the status matches
- if status := rr.Code; status != c.expStatus {
- t.Errorf("%s, handler returned wrong status code: got %v want %v",
- c.msg, status, c.expStatus)
- }
- // if there's a validator, call it
- for _, validate := range c.validators {
- validate(c, tester, t)
- }
- }
- }
- var createUserTests = []*userTest{
- &userTest{
- msg: "Create user",
- method: "POST",
- endpoint: "/api/users",
- body: `{
- "email": "belanger@getporter.dev",
- "password": "hello"
- }`,
- expStatus: http.StatusCreated,
- expBody: "",
- },
- &userTest{
- msg: "Create user invalid email",
- method: "POST",
- endpoint: "/api/users",
- body: `{
- "email": "notanemail",
- "password": "hello"
- }`,
- expStatus: http.StatusUnprocessableEntity,
- expBody: `{"code":601,"errors":["email validation failed"]}`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- &userTest{
- msg: "Create user missing field",
- method: "POST",
- endpoint: "/api/users",
- body: `{
- "password": "hello"
- }`,
- expStatus: http.StatusUnprocessableEntity,
- expBody: `{"code":601,"errors":["required validation failed"]}`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Create user same email",
- method: "POST",
- endpoint: "/api/users",
- body: `{
- "email": "belanger@getporter.dev",
- "password": "hello"
- }`,
- expStatus: http.StatusUnprocessableEntity,
- expBody: `{"code":601,"errors":["email already taken"]}`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- &userTest{
- msg: "Create user invalid field type",
- method: "POST",
- endpoint: "/api/users",
- body: `{
- "email": "belanger@getporter.dev",
- "password": 0
- }`,
- expStatus: http.StatusBadRequest,
- expBody: `{"code":600,"errors":["could not process request"]}`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- }
- func TestHandleCreateUser(t *testing.T) {
- testUserRequests(t, createUserTests, true)
- }
- var createUserTestsWriteFail = []*userTest{
- &userTest{
- msg: "Create user db connection down",
- method: "POST",
- endpoint: "/api/users",
- body: `{
- "email": "belanger@getporter.dev",
- "password": "hello"
- }`,
- expStatus: http.StatusInternalServerError,
- expBody: `{"code":500,"errors":["could not read from database"]}`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- }
- func TestHandleCreateUserWriteFail(t *testing.T) {
- testUserRequests(t, createUserTestsWriteFail, false)
- }
- var loginUserTests = []*userTest{
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Login user successful",
- method: "POST",
- endpoint: "/api/login",
- body: `{
- "email": "belanger@getporter.dev",
- "password": "hello"
- }`,
- expStatus: http.StatusOK,
- expBody: ``,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Login user already logged in",
- method: "POST",
- endpoint: "/api/login",
- body: `{
- "email": "belanger@getporter.dev",
- "password": "hello"
- }`,
- expStatus: http.StatusOK,
- expBody: ``,
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- &userTest{
- msg: "Login user unregistered email",
- method: "POST",
- endpoint: "/api/login",
- body: `{
- "email": "belanger@getporter.dev",
- "password": "hello"
- }`,
- expStatus: http.StatusUnauthorized,
- expBody: `{"code":401,"errors":["email not registered"]}`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Login user incorrect password",
- method: "POST",
- endpoint: "/api/login",
- body: `{
- "email": "belanger@getporter.dev",
- "password": "notthepassword"
- }`,
- expStatus: http.StatusUnauthorized,
- expBody: `{"code":401,"errors":["incorrect password"]}`,
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- }
- func TestHandleLoginUser(t *testing.T) {
- testUserRequests(t, loginUserTests, true)
- }
- var logoutUserTests = []*userTest{
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Logout user successful",
- method: "POST",
- endpoint: "/api/logout",
- body: `{
- "email": "belanger@getporter.dev",
- "password": "hello"
- }`,
- expStatus: http.StatusOK,
- expBody: ``,
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- func(c *userTest, tester *tester, t *testing.T) {
- req, err := http.NewRequest(
- "GET",
- "/api/users/1",
- strings.NewReader(""),
- )
- req.AddCookie(tester.cookie)
- if err != nil {
- t.Fatal(err)
- }
- rr2 := httptest.NewRecorder()
- tester.router.ServeHTTP(rr2, req)
- if status := rr2.Code; status != http.StatusForbidden {
- t.Errorf("%s, handler returned wrong status: got %v want %v",
- "validator failed", status, http.StatusForbidden)
- }
- },
- },
- },
- }
- func TestHandleLogoutUser(t *testing.T) {
- testUserRequests(t, logoutUserTests, true)
- }
- var readUserTests = []*userTest{
- &userTest{
- initializers: []func(tester *tester){
- initUserWithClusters,
- },
- msg: "Read user successful",
- method: "GET",
- endpoint: "/api/users/1",
- body: "",
- expStatus: http.StatusOK,
- expBody: `{"id":1,"email":"belanger@getporter.dev","clusters":[{"name":"cluster-test","server":"https://localhost","context":"context-test","user":"test-admin"}],"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin"}`,
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- UserModelBodyValidator,
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Read user unauthorized",
- method: "GET",
- endpoint: "/api/users/2",
- body: "",
- expStatus: http.StatusForbidden,
- expBody: http.StatusText(http.StatusForbidden) + "\n",
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- }
- func TestHandleReadUser(t *testing.T) {
- testUserRequests(t, readUserTests, true)
- }
- var readUserClustersTests = []*userTest{
- &userTest{
- initializers: []func(tester *tester){
- initUserWithClusters,
- },
- msg: "Read user successful",
- method: "GET",
- endpoint: "/api/users/1/clusters",
- body: "",
- expStatus: http.StatusOK,
- useCookie: true,
- expBody: `[{"name":"cluster-test","server":"https://localhost","context":"context-test","user":"test-admin"}]`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- ClusterBodyValidator,
- },
- },
- }
- func TestHandleReadUserClusters(t *testing.T) {
- testUserRequests(t, readUserClustersTests, true)
- }
- var readUserClustersAllTests = []*userTest{
- &userTest{
- initializers: []func(tester *tester){
- initUserWithClusters,
- },
- msg: "Read user successful",
- method: "GET",
- endpoint: "/api/users/1/clusters/all",
- body: "",
- expStatus: http.StatusOK,
- useCookie: true,
- expBody: `[{"name":"cluster-test","server":"https://localhost","context":"context-test","user":"test-admin"}]`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- ClusterBodyValidator,
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserWithClusters,
- func(tester *tester) {
- initUserDefault(tester)
- user, _ := tester.repo.User.ReadUserByEmail("belanger@getporter.dev")
- user.Clusters = []models.ClusterConfig{}
- user.RawKubeConfig = []byte("apiVersion: \xc5\n")
- tester.repo.User.UpdateUser(user)
- },
- },
- msg: "Read user with invalid utf-8 \xc5 in kubeconfig",
- method: "GET",
- endpoint: "/api/users/1/clusters/all",
- body: "",
- expStatus: http.StatusBadRequest,
- useCookie: true,
- expBody: `{"code":600,"errors":["could not process request"]}`,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- ClusterBodyValidator,
- },
- },
- }
- func TestHandleReadUserClustersAll(t *testing.T) {
- testUserRequests(t, readUserClustersAllTests, true)
- }
- var updateUserTests = []*userTest{
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Update user successful",
- method: "PUT",
- endpoint: "/api/users/1",
- body: `{"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin", "allowedClusters":[]}`,
- expStatus: http.StatusNoContent,
- expBody: "",
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- func(c *userTest, tester *tester, t *testing.T) {
- req, err := http.NewRequest(
- "GET",
- "/api/users/1",
- strings.NewReader(""),
- )
- req.AddCookie(tester.cookie)
- if err != nil {
- t.Fatal(err)
- }
- rr2 := httptest.NewRecorder()
- tester.router.ServeHTTP(rr2, req)
- gotBody := &models.UserExternal{}
- expBody := &models.UserExternal{}
- json.Unmarshal(rr2.Body.Bytes(), gotBody)
- json.Unmarshal([]byte(`{"id":1,"email":"belanger@getporter.dev","clusters":[],"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin"}`), expBody)
- if !reflect.DeepEqual(gotBody, expBody) {
- t.Errorf("%s, handler returned wrong body: got %v want %v",
- "validator failed", gotBody, expBody)
- }
- },
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Update user successful without allowedClusters parameter",
- method: "PUT",
- endpoint: "/api/users/1",
- body: `{"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin"}`,
- expStatus: http.StatusNoContent,
- expBody: "",
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- func(c *userTest, tester *tester, t *testing.T) {
- req, err := http.NewRequest(
- "GET",
- "/api/users/1",
- strings.NewReader(""),
- )
- req.AddCookie(tester.cookie)
- if err != nil {
- t.Fatal(err)
- }
- rr2 := httptest.NewRecorder()
- tester.router.ServeHTTP(rr2, req)
- gotBody := &models.UserExternal{}
- expBody := &models.UserExternal{}
- json.Unmarshal(rr2.Body.Bytes(), gotBody)
- json.Unmarshal([]byte(`{"id":1,"email":"belanger@getporter.dev","clusters":[],"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin"}`), expBody)
- if !reflect.DeepEqual(gotBody, expBody) {
- t.Errorf("%s, handler returned wrong body: got %v want %v",
- "validator failed", gotBody, expBody)
- }
- },
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Update user successful with allowedClusters",
- method: "PUT",
- endpoint: "/api/users/1",
- body: `{"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin", "allowedClusters":["cluster-test"]}`,
- expStatus: http.StatusNoContent,
- expBody: "",
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- func(c *userTest, tester *tester, t *testing.T) {
- req, err := http.NewRequest(
- "GET",
- "/api/users/1",
- strings.NewReader(""),
- )
- req.AddCookie(tester.cookie)
- if err != nil {
- t.Fatal(err)
- }
- rr2 := httptest.NewRecorder()
- tester.router.ServeHTTP(rr2, req)
- gotBody := &models.UserExternal{}
- expBody := &models.UserExternal{}
- json.Unmarshal(rr2.Body.Bytes(), gotBody)
- json.Unmarshal([]byte(`{"id":1,"email":"belanger@getporter.dev","clusters":[{"name":"cluster-test","server":"https://localhost","context":"context-test","user":"test-admin"}],"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin"}`), expBody)
- if !reflect.DeepEqual(gotBody, expBody) {
- t.Errorf("%s, handler returned wrong body: got %v want %v",
- "validator failed", gotBody, expBody)
- }
- },
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserWithClusters,
- },
- msg: "Update user successful without rawKubeConfig",
- method: "PUT",
- endpoint: "/api/users/1",
- body: `{"allowedClusters":[]}`,
- expStatus: http.StatusNoContent,
- expBody: "",
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- func(c *userTest, tester *tester, t *testing.T) {
- req, err := http.NewRequest(
- "GET",
- "/api/users/1",
- strings.NewReader(""),
- )
- req.AddCookie(tester.cookie)
- if err != nil {
- t.Fatal(err)
- }
- rr2 := httptest.NewRecorder()
- tester.router.ServeHTTP(rr2, req)
- gotBody := &models.UserExternal{}
- expBody := &models.UserExternal{}
- json.Unmarshal(rr2.Body.Bytes(), gotBody)
- json.Unmarshal([]byte(`{"id":1,"email":"belanger@getporter.dev","clusters":[],"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin"}`), expBody)
- if !reflect.DeepEqual(gotBody, expBody) {
- t.Errorf("%s, handler returned wrong body: got %v want %v",
- "validator failed", gotBody, expBody)
- }
- },
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Update user invalid id",
- method: "PUT",
- endpoint: "/api/users/alsdfjk",
- body: `{"rawKubeConfig":"apiVersion: v1\nkind: Config\npreferences: {}\ncurrent-context: context-test\nclusters:\n- cluster:\n server: https://localhost\n name: cluster-test\ncontexts:\n- context:\n cluster: cluster-test\n user: test-admin\n name: context-test\nusers:\n- name: test-admin", "allowedClusters":[]}`,
- expStatus: http.StatusForbidden,
- expBody: http.StatusText(http.StatusForbidden) + "\n",
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Update user bad kubeconfig",
- method: "PUT",
- endpoint: "/api/users/1",
- body: `{"rawKubeConfig":"notvalidyaml", "allowedClusters":[]}`,
- expStatus: http.StatusBadRequest,
- expBody: `{"code":600,"errors":["could not process request"]}`,
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- }
- func TestHandleUpdateUser(t *testing.T) {
- testUserRequests(t, updateUserTests, true)
- }
- var deleteUserTests = []*userTest{
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Delete user successful",
- method: "DELETE",
- endpoint: "/api/users/1",
- body: `{"password":"hello"}`,
- expStatus: http.StatusNoContent,
- expBody: "",
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- func(c *userTest, tester *tester, t *testing.T) {
- req, err := http.NewRequest(
- "GET",
- "/api/users/1",
- strings.NewReader(""),
- )
- req.AddCookie(tester.cookie)
- if err != nil {
- t.Fatal(err)
- }
- rr2 := httptest.NewRecorder()
- tester.router.ServeHTTP(rr2, req)
- gotBody := &models.UserExternal{}
- expBody := &models.UserExternal{}
- if status := rr2.Code; status != 404 {
- t.Errorf("DELETE user validation, handler returned wrong status code: got %v want %v",
- status, 404)
- }
- json.Unmarshal(rr2.Body.Bytes(), gotBody)
- json.Unmarshal([]byte(`{"code":602,"errors":["could not find requested object"]}`), expBody)
- if !reflect.DeepEqual(gotBody, expBody) {
- t.Errorf("%s, handler returned wrong body: got %v want %v",
- "validator failed", gotBody, expBody)
- }
- },
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Delete user invalid id",
- method: "DELETE",
- endpoint: "/api/users/aldkjf",
- body: `{"password":"hello"}`,
- expStatus: http.StatusForbidden,
- expBody: http.StatusText(http.StatusForbidden) + "\n",
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- &userTest{
- initializers: []func(tester *tester){
- initUserDefault,
- },
- msg: "Delete user missing password",
- method: "DELETE",
- endpoint: "/api/users/1",
- body: `{}`,
- expStatus: http.StatusUnprocessableEntity,
- expBody: `{"code":601,"errors":["required validation failed"]}`,
- useCookie: true,
- validators: []func(c *userTest, tester *tester, t *testing.T){
- BasicBodyValidator,
- },
- },
- }
- func TestHandleDeleteUser(t *testing.T) {
- testUserRequests(t, deleteUserTests, true)
- }
- func BasicBodyValidator(c *userTest, tester *tester, t *testing.T) {
- if body := tester.rr.Body.String(); body != c.expBody {
- t.Errorf("%s, handler returned wrong body: got %v want %v",
- c.msg, body, c.expBody)
- }
- }
- func UserModelBodyValidator(c *userTest, tester *tester, t *testing.T) {
- gotBody := &models.UserExternal{}
- expBody := &models.UserExternal{}
- json.Unmarshal(tester.rr.Body.Bytes(), gotBody)
- json.Unmarshal([]byte(c.expBody), expBody)
- if !reflect.DeepEqual(gotBody, expBody) {
- t.Errorf("%s, handler returned wrong body: got %v want %v",
- c.msg, gotBody, expBody)
- }
- }
- func ClusterBodyValidator(c *userTest, tester *tester, t *testing.T) {
- // if status is expected to be 200, parse the body for UserExternal
- gotBody := &[]models.ClusterConfigExternal{}
- expBody := &[]models.ClusterConfigExternal{}
- json.Unmarshal(tester.rr.Body.Bytes(), gotBody)
- json.Unmarshal([]byte(c.expBody), expBody)
- if !reflect.DeepEqual(gotBody, expBody) {
- t.Errorf("%s, handler returned wrong body: got %v want %v",
- c.msg, gotBody, expBody)
- }
- }
|