user_handler_test.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742
  1. package api_test
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "net/http"
  6. "net/http/httptest"
  7. "reflect"
  8. "strings"
  9. "testing"
  10. "github.com/porter-dev/porter/internal/models"
  11. )
  12. // ------------------------- TEST TYPES AND MAIN LOOP ------------------------- //
  13. type userTest struct {
  14. initializers []func(t *tester)
  15. msg string
  16. method string
  17. endpoint string
  18. body string
  19. expStatus int
  20. expBody string
  21. useCookie bool
  22. validators []func(c *userTest, tester *tester, t *testing.T)
  23. }
  24. func testUserRequests(t *testing.T, tests []*userTest, canQuery bool) {
  25. for _, c := range tests {
  26. // create a new tester
  27. tester := newTester(canQuery)
  28. // if there's an initializer, call it
  29. for _, init := range c.initializers {
  30. init(tester)
  31. }
  32. req, err := http.NewRequest(
  33. c.method,
  34. c.endpoint,
  35. strings.NewReader(c.body),
  36. )
  37. tester.req = req
  38. if c.useCookie {
  39. req.AddCookie(tester.cookie)
  40. }
  41. if err != nil {
  42. t.Fatal(err)
  43. }
  44. tester.execute()
  45. rr := tester.rr
  46. // first, check that the status matches
  47. if status := rr.Code; status != c.expStatus {
  48. t.Errorf("%s, handler returned wrong status code: got %v want %v",
  49. c.msg, status, c.expStatus)
  50. }
  51. // if there's a validator, call it
  52. for _, validate := range c.validators {
  53. validate(c, tester, t)
  54. }
  55. }
  56. }
  57. // ------------------------- TEST FIXTURES AND FUNCTIONS ------------------------- //
  58. var authCheckTests = []*userTest{
  59. &userTest{
  60. initializers: []func(tester *tester){
  61. initUserDefault,
  62. },
  63. msg: "Auth check successful. User is logged in.",
  64. method: "GET",
  65. endpoint: "/api/auth/check",
  66. expStatus: http.StatusOK,
  67. body: "",
  68. expBody: `{"id":1,"email":"belanger@getporter.dev","contexts":null,"rawKubeConfig":""}`,
  69. useCookie: true,
  70. validators: []func(c *userTest, tester *tester, t *testing.T){
  71. userBasicBodyValidator,
  72. },
  73. },
  74. &userTest{
  75. initializers: []func(tester *tester){
  76. initUserDefault,
  77. },
  78. msg: "Auth check failure. User is not logged in.",
  79. method: "GET",
  80. endpoint: "/api/auth/check",
  81. body: "",
  82. expStatus: http.StatusForbidden,
  83. expBody: http.StatusText(http.StatusForbidden) + "\n",
  84. validators: []func(c *userTest, tester *tester, t *testing.T){
  85. userBasicBodyValidator,
  86. },
  87. },
  88. }
  89. func TestHandleAuthCheck(t *testing.T) {
  90. testUserRequests(t, authCheckTests, true)
  91. }
  92. var createUserTests = []*userTest{
  93. &userTest{
  94. msg: "Create user",
  95. method: "POST",
  96. endpoint: "/api/users",
  97. body: `{
  98. "email": "belanger@getporter.dev",
  99. "password": "hello"
  100. }`,
  101. expStatus: http.StatusCreated,
  102. expBody: `{"id":1,"email":"belanger@getporter.dev","contexts":null,"rawKubeConfig":""}`,
  103. validators: []func(c *userTest, tester *tester, t *testing.T){
  104. userModelBodyValidator,
  105. },
  106. },
  107. &userTest{
  108. msg: "Create user invalid email",
  109. method: "POST",
  110. endpoint: "/api/users",
  111. body: `{
  112. "email": "notanemail",
  113. "password": "hello"
  114. }`,
  115. expStatus: http.StatusUnprocessableEntity,
  116. expBody: `{"code":601,"errors":["email validation failed"]}`,
  117. validators: []func(c *userTest, tester *tester, t *testing.T){
  118. userBasicBodyValidator,
  119. },
  120. },
  121. &userTest{
  122. msg: "Create user missing field",
  123. method: "POST",
  124. endpoint: "/api/users",
  125. body: `{
  126. "password": "hello"
  127. }`,
  128. expStatus: http.StatusUnprocessableEntity,
  129. expBody: `{"code":601,"errors":["required validation failed"]}`,
  130. validators: []func(c *userTest, tester *tester, t *testing.T){
  131. userBasicBodyValidator,
  132. },
  133. },
  134. &userTest{
  135. initializers: []func(tester *tester){
  136. initUserDefault,
  137. },
  138. msg: "Create user same email",
  139. method: "POST",
  140. endpoint: "/api/users",
  141. body: `{
  142. "email": "belanger@getporter.dev",
  143. "password": "hello"
  144. }`,
  145. expStatus: http.StatusUnprocessableEntity,
  146. expBody: `{"code":601,"errors":["email already taken"]}`,
  147. validators: []func(c *userTest, tester *tester, t *testing.T){
  148. userBasicBodyValidator,
  149. },
  150. },
  151. &userTest{
  152. msg: "Create user invalid field type",
  153. method: "POST",
  154. endpoint: "/api/users",
  155. body: `{
  156. "email": "belanger@getporter.dev",
  157. "password": 0
  158. }`,
  159. expStatus: http.StatusBadRequest,
  160. expBody: `{"code":600,"errors":["could not process request"]}`,
  161. validators: []func(c *userTest, tester *tester, t *testing.T){
  162. userBasicBodyValidator,
  163. },
  164. },
  165. }
  166. func TestHandleCreateUser(t *testing.T) {
  167. testUserRequests(t, createUserTests, true)
  168. }
  169. var createUserTestsWriteFail = []*userTest{
  170. &userTest{
  171. msg: "Create user db connection down",
  172. method: "POST",
  173. endpoint: "/api/users",
  174. body: `{
  175. "email": "belanger@getporter.dev",
  176. "password": "hello"
  177. }`,
  178. expStatus: http.StatusInternalServerError,
  179. expBody: `{"code":500,"errors":["could not read from database"]}`,
  180. validators: []func(c *userTest, tester *tester, t *testing.T){
  181. userBasicBodyValidator,
  182. },
  183. },
  184. }
  185. func TestHandleCreateUserWriteFail(t *testing.T) {
  186. testUserRequests(t, createUserTestsWriteFail, false)
  187. }
  188. var loginUserTests = []*userTest{
  189. &userTest{
  190. initializers: []func(tester *tester){
  191. initUserDefault,
  192. },
  193. msg: "Login user successful",
  194. method: "POST",
  195. endpoint: "/api/login",
  196. body: `{
  197. "email": "belanger@getporter.dev",
  198. "password": "hello"
  199. }`,
  200. expStatus: http.StatusOK,
  201. expBody: `{"id":1,"email":"belanger@getporter.dev","contexts":null,"rawKubeConfig":""}`,
  202. validators: []func(c *userTest, tester *tester, t *testing.T){
  203. userBasicBodyValidator,
  204. },
  205. },
  206. &userTest{
  207. initializers: []func(tester *tester){
  208. initUserDefault,
  209. },
  210. msg: "Login user already logged in",
  211. method: "POST",
  212. endpoint: "/api/login",
  213. body: `{
  214. "email": "belanger@getporter.dev",
  215. "password": "hello"
  216. }`,
  217. expStatus: http.StatusOK,
  218. expBody: `{"id":1,"email":"belanger@getporter.dev","contexts":null,"rawKubeConfig":""}`,
  219. useCookie: true,
  220. validators: []func(c *userTest, tester *tester, t *testing.T){
  221. userBasicBodyValidator,
  222. },
  223. },
  224. &userTest{
  225. msg: "Login user unregistered email",
  226. method: "POST",
  227. endpoint: "/api/login",
  228. body: `{
  229. "email": "belanger@getporter.dev",
  230. "password": "hello"
  231. }`,
  232. expStatus: http.StatusUnauthorized,
  233. expBody: `{"code":401,"errors":["email not registered"]}`,
  234. validators: []func(c *userTest, tester *tester, t *testing.T){
  235. userBasicBodyValidator,
  236. },
  237. },
  238. &userTest{
  239. initializers: []func(tester *tester){
  240. initUserDefault,
  241. },
  242. msg: "Login user incorrect password",
  243. method: "POST",
  244. endpoint: "/api/login",
  245. body: `{
  246. "email": "belanger@getporter.dev",
  247. "password": "notthepassword"
  248. }`,
  249. expStatus: http.StatusUnauthorized,
  250. expBody: `{"code":401,"errors":["incorrect password"]}`,
  251. useCookie: true,
  252. validators: []func(c *userTest, tester *tester, t *testing.T){
  253. userBasicBodyValidator,
  254. },
  255. },
  256. }
  257. func TestHandleLoginUser(t *testing.T) {
  258. testUserRequests(t, loginUserTests, true)
  259. }
  260. var logoutUserTests = []*userTest{
  261. &userTest{
  262. initializers: []func(tester *tester){
  263. initUserDefault,
  264. },
  265. msg: "Logout user successful",
  266. method: "POST",
  267. endpoint: "/api/logout",
  268. body: `{
  269. "email": "belanger@getporter.dev",
  270. "password": "hello"
  271. }`,
  272. expStatus: http.StatusOK,
  273. expBody: ``,
  274. useCookie: true,
  275. validators: []func(c *userTest, tester *tester, t *testing.T){
  276. func(c *userTest, tester *tester, t *testing.T) {
  277. req, err := http.NewRequest(
  278. "GET",
  279. "/api/users/1",
  280. strings.NewReader(""),
  281. )
  282. req.AddCookie(tester.cookie)
  283. if err != nil {
  284. t.Fatal(err)
  285. }
  286. rr2 := httptest.NewRecorder()
  287. tester.router.ServeHTTP(rr2, req)
  288. if status := rr2.Code; status != http.StatusForbidden {
  289. t.Errorf("%s, handler returned wrong status: got %v want %v",
  290. "validator failed", status, http.StatusForbidden)
  291. }
  292. },
  293. },
  294. },
  295. }
  296. func TestHandleLogoutUser(t *testing.T) {
  297. testUserRequests(t, logoutUserTests, true)
  298. }
  299. var readUserTests = []*userTest{
  300. &userTest{
  301. initializers: []func(tester *tester){
  302. initUserWithContexts,
  303. },
  304. msg: "Read user successful",
  305. method: "GET",
  306. endpoint: "/api/users/1",
  307. body: "",
  308. expStatus: http.StatusOK,
  309. expBody: `{"id":1,"email":"belanger@getporter.dev","contexts":["context-test"],"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"}`,
  310. useCookie: true,
  311. validators: []func(c *userTest, tester *tester, t *testing.T){
  312. userModelBodyValidator,
  313. },
  314. },
  315. &userTest{
  316. initializers: []func(tester *tester){
  317. initUserDefault,
  318. },
  319. msg: "Read user unauthorized",
  320. method: "GET",
  321. endpoint: "/api/users/2",
  322. body: "",
  323. expStatus: http.StatusForbidden,
  324. expBody: http.StatusText(http.StatusForbidden) + "\n",
  325. validators: []func(c *userTest, tester *tester, t *testing.T){
  326. userBasicBodyValidator,
  327. },
  328. },
  329. }
  330. func TestHandleReadUser(t *testing.T) {
  331. testUserRequests(t, readUserTests, true)
  332. }
  333. var readUserContextsTests = []*userTest{
  334. &userTest{
  335. initializers: []func(tester *tester){
  336. initUserWithContexts,
  337. },
  338. msg: "Read user context selected successful",
  339. method: "GET",
  340. endpoint: "/api/users/1/contexts",
  341. body: "",
  342. expStatus: http.StatusOK,
  343. useCookie: true,
  344. expBody: `[{"name":"context-test","server":"https://localhost","cluster":"cluster-test","user":"test-admin","selected":true}]`,
  345. validators: []func(c *userTest, tester *tester, t *testing.T){
  346. userContextBodyValidator,
  347. },
  348. },
  349. &userTest{
  350. initializers: []func(tester *tester){
  351. func(tester *tester) {
  352. initUserDefault(tester)
  353. user, _ := tester.repo.User.ReadUserByEmail("belanger@getporter.dev")
  354. user.Contexts = ""
  355. 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")
  356. tester.repo.User.UpdateUser(user)
  357. },
  358. },
  359. msg: "Read user context not selected successful",
  360. method: "GET",
  361. endpoint: "/api/users/1/contexts",
  362. body: "",
  363. expStatus: http.StatusOK,
  364. useCookie: true,
  365. expBody: `[{"name":"context-test","server":"https://localhost","cluster":"cluster-test","user":"test-admin","selected":false}]`,
  366. validators: []func(c *userTest, tester *tester, t *testing.T){
  367. userContextBodyValidator,
  368. },
  369. },
  370. }
  371. func TestHandleReadUserContexts(t *testing.T) {
  372. testUserRequests(t, readUserContextsTests, true)
  373. }
  374. var updateUserTests = []*userTest{
  375. &userTest{
  376. initializers: []func(tester *tester){
  377. initUserDefault,
  378. },
  379. msg: "Update user successful",
  380. method: "PUT",
  381. endpoint: "/api/users/1",
  382. 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", "allowedContexts":[]}`,
  383. expStatus: http.StatusNoContent,
  384. expBody: "",
  385. useCookie: true,
  386. validators: []func(c *userTest, tester *tester, t *testing.T){
  387. func(c *userTest, tester *tester, t *testing.T) {
  388. req, err := http.NewRequest(
  389. "GET",
  390. "/api/users/1",
  391. strings.NewReader(""),
  392. )
  393. req.AddCookie(tester.cookie)
  394. if err != nil {
  395. t.Fatal(err)
  396. }
  397. rr2 := httptest.NewRecorder()
  398. tester.router.ServeHTTP(rr2, req)
  399. gotBody := &models.UserExternal{}
  400. expBody := &models.UserExternal{}
  401. json.Unmarshal(rr2.Body.Bytes(), gotBody)
  402. json.Unmarshal([]byte(`{"id":1,"email":"belanger@getporter.dev","contexts":[],"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)
  403. fmt.Println(rr2.Body.String())
  404. if !reflect.DeepEqual(gotBody, expBody) {
  405. t.Errorf("%s, handler returned wrong body: got %v want %v",
  406. "validator failed", gotBody, expBody)
  407. }
  408. },
  409. },
  410. },
  411. &userTest{
  412. initializers: []func(tester *tester){
  413. initUserDefault,
  414. },
  415. msg: "Update user successful without allowedContexts parameter",
  416. method: "PUT",
  417. endpoint: "/api/users/1",
  418. 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"}`,
  419. expStatus: http.StatusNoContent,
  420. expBody: "",
  421. useCookie: true,
  422. validators: []func(c *userTest, tester *tester, t *testing.T){
  423. func(c *userTest, tester *tester, t *testing.T) {
  424. req, err := http.NewRequest(
  425. "GET",
  426. "/api/users/1",
  427. strings.NewReader(""),
  428. )
  429. req.AddCookie(tester.cookie)
  430. if err != nil {
  431. t.Fatal(err)
  432. }
  433. rr2 := httptest.NewRecorder()
  434. tester.router.ServeHTTP(rr2, req)
  435. gotBody := &models.UserExternal{}
  436. expBody := &models.UserExternal{}
  437. json.Unmarshal(rr2.Body.Bytes(), gotBody)
  438. json.Unmarshal([]byte(`{"id":1,"email":"belanger@getporter.dev","contexts":[],"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)
  439. if !reflect.DeepEqual(gotBody, expBody) {
  440. t.Errorf("%s, handler returned wrong body: got %v want %v",
  441. "validator failed", gotBody, expBody)
  442. }
  443. },
  444. },
  445. },
  446. &userTest{
  447. initializers: []func(tester *tester){
  448. initUserDefault,
  449. },
  450. msg: "Update user successful with allowedContexts",
  451. method: "PUT",
  452. endpoint: "/api/users/1",
  453. 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", "allowedContexts":["context-test"]}`,
  454. expStatus: http.StatusNoContent,
  455. expBody: "",
  456. useCookie: true,
  457. validators: []func(c *userTest, tester *tester, t *testing.T){
  458. func(c *userTest, tester *tester, t *testing.T) {
  459. req, err := http.NewRequest(
  460. "GET",
  461. "/api/users/1",
  462. strings.NewReader(""),
  463. )
  464. req.AddCookie(tester.cookie)
  465. if err != nil {
  466. t.Fatal(err)
  467. }
  468. rr2 := httptest.NewRecorder()
  469. tester.router.ServeHTTP(rr2, req)
  470. gotBody := &models.UserExternal{}
  471. expBody := &models.UserExternal{}
  472. json.Unmarshal(rr2.Body.Bytes(), gotBody)
  473. json.Unmarshal([]byte(`{"id":1,"email":"belanger@getporter.dev","contexts":["context-test"],"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)
  474. if !reflect.DeepEqual(gotBody, expBody) {
  475. t.Errorf("%s, handler returned wrong body: got %v want %v",
  476. "validator failed", gotBody, expBody)
  477. }
  478. },
  479. },
  480. },
  481. &userTest{
  482. initializers: []func(tester *tester){
  483. initUserWithContexts,
  484. },
  485. msg: "Update user successful without rawKubeConfig",
  486. method: "PUT",
  487. endpoint: "/api/users/1",
  488. body: `{"allowedContexts":[]}`,
  489. expStatus: http.StatusNoContent,
  490. expBody: "",
  491. useCookie: true,
  492. validators: []func(c *userTest, tester *tester, t *testing.T){
  493. func(c *userTest, tester *tester, t *testing.T) {
  494. req, err := http.NewRequest(
  495. "GET",
  496. "/api/users/1",
  497. strings.NewReader(""),
  498. )
  499. req.AddCookie(tester.cookie)
  500. if err != nil {
  501. t.Fatal(err)
  502. }
  503. rr2 := httptest.NewRecorder()
  504. tester.router.ServeHTTP(rr2, req)
  505. gotBody := &models.UserExternal{}
  506. expBody := &models.UserExternal{}
  507. json.Unmarshal(rr2.Body.Bytes(), gotBody)
  508. json.Unmarshal([]byte(`{"id":1,"email":"belanger@getporter.dev","contexts":[],"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)
  509. if !reflect.DeepEqual(gotBody, expBody) {
  510. t.Errorf("%s, handler returned wrong body: got %v want %v",
  511. "validator failed", gotBody, expBody)
  512. }
  513. },
  514. },
  515. },
  516. &userTest{
  517. initializers: []func(tester *tester){
  518. initUserDefault,
  519. },
  520. msg: "Update user invalid id",
  521. method: "PUT",
  522. endpoint: "/api/users/alsdfjk",
  523. 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", "allowedContexts":[]}`,
  524. expStatus: http.StatusForbidden,
  525. expBody: http.StatusText(http.StatusForbidden) + "\n",
  526. validators: []func(c *userTest, tester *tester, t *testing.T){
  527. userBasicBodyValidator,
  528. },
  529. },
  530. &userTest{
  531. initializers: []func(tester *tester){
  532. initUserDefault,
  533. },
  534. msg: "Update user bad kubeconfig",
  535. method: "PUT",
  536. endpoint: "/api/users/1",
  537. body: `{"rawKubeConfig":"notvalidyaml", "allowedContexts":[]}`,
  538. expStatus: http.StatusBadRequest,
  539. expBody: `{"code":600,"errors":["could not process request"]}`,
  540. useCookie: true,
  541. validators: []func(c *userTest, tester *tester, t *testing.T){
  542. userBasicBodyValidator,
  543. },
  544. },
  545. }
  546. func TestHandleUpdateUser(t *testing.T) {
  547. testUserRequests(t, updateUserTests, true)
  548. }
  549. var deleteUserTests = []*userTest{
  550. &userTest{
  551. initializers: []func(tester *tester){
  552. initUserDefault,
  553. },
  554. msg: "Delete user successful",
  555. method: "DELETE",
  556. endpoint: "/api/users/1",
  557. body: `{"password":"hello"}`,
  558. expStatus: http.StatusNoContent,
  559. expBody: "",
  560. useCookie: true,
  561. validators: []func(c *userTest, tester *tester, t *testing.T){
  562. func(c *userTest, tester *tester, t *testing.T) {
  563. req, err := http.NewRequest(
  564. "GET",
  565. "/api/users/1",
  566. strings.NewReader(""),
  567. )
  568. req.AddCookie(tester.cookie)
  569. if err != nil {
  570. t.Fatal(err)
  571. }
  572. rr2 := httptest.NewRecorder()
  573. tester.router.ServeHTTP(rr2, req)
  574. gotBody := &models.UserExternal{}
  575. expBody := &models.UserExternal{}
  576. if status := rr2.Code; status != 404 {
  577. t.Errorf("DELETE user validation, handler returned wrong status code: got %v want %v",
  578. status, 404)
  579. }
  580. json.Unmarshal(rr2.Body.Bytes(), gotBody)
  581. json.Unmarshal([]byte(`{"code":602,"errors":["could not find requested object"]}`), expBody)
  582. if !reflect.DeepEqual(gotBody, expBody) {
  583. t.Errorf("%s, handler returned wrong body: got %v want %v",
  584. "validator failed", gotBody, expBody)
  585. }
  586. },
  587. },
  588. },
  589. &userTest{
  590. initializers: []func(tester *tester){
  591. initUserDefault,
  592. },
  593. msg: "Delete user invalid id",
  594. method: "DELETE",
  595. endpoint: "/api/users/aldkjf",
  596. body: `{"password":"hello"}`,
  597. expStatus: http.StatusForbidden,
  598. expBody: http.StatusText(http.StatusForbidden) + "\n",
  599. validators: []func(c *userTest, tester *tester, t *testing.T){
  600. userBasicBodyValidator,
  601. },
  602. },
  603. &userTest{
  604. initializers: []func(tester *tester){
  605. initUserDefault,
  606. },
  607. msg: "Delete user missing password",
  608. method: "DELETE",
  609. endpoint: "/api/users/1",
  610. body: `{}`,
  611. expStatus: http.StatusUnprocessableEntity,
  612. expBody: `{"code":601,"errors":["required validation failed"]}`,
  613. useCookie: true,
  614. validators: []func(c *userTest, tester *tester, t *testing.T){
  615. userBasicBodyValidator,
  616. },
  617. },
  618. }
  619. func TestHandleDeleteUser(t *testing.T) {
  620. testUserRequests(t, deleteUserTests, true)
  621. }
  622. // ------------------------- INITIALIZERS AND VALIDATORS ------------------------- //
  623. func initUserDefault(tester *tester) {
  624. tester.createUserSession("belanger@getporter.dev", "hello")
  625. }
  626. func initUserWithContexts(tester *tester) {
  627. initUserDefault(tester)
  628. user, _ := tester.repo.User.ReadUserByEmail("belanger@getporter.dev")
  629. user.Contexts = "context-test"
  630. 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")
  631. tester.repo.User.UpdateUser(user)
  632. }
  633. func userBasicBodyValidator(c *userTest, tester *tester, t *testing.T) {
  634. if body := tester.rr.Body.String(); strings.TrimSpace(body) != strings.TrimSpace(c.expBody) {
  635. t.Errorf("%s, handler returned wrong body: got %v want %v",
  636. c.msg, body, c.expBody)
  637. }
  638. }
  639. func userModelBodyValidator(c *userTest, tester *tester, t *testing.T) {
  640. gotBody := &models.UserExternal{}
  641. expBody := &models.UserExternal{}
  642. json.Unmarshal(tester.rr.Body.Bytes(), gotBody)
  643. json.Unmarshal([]byte(c.expBody), expBody)
  644. if !reflect.DeepEqual(gotBody, expBody) {
  645. t.Errorf("%s, handler returned wrong body: got %v want %v",
  646. c.msg, gotBody, expBody)
  647. }
  648. }
  649. func userContextBodyValidator(c *userTest, tester *tester, t *testing.T) {
  650. gotBody := &[]models.Context{}
  651. expBody := &[]models.Context{}
  652. json.Unmarshal(tester.rr.Body.Bytes(), gotBody)
  653. json.Unmarshal([]byte(c.expBody), expBody)
  654. if !reflect.DeepEqual(gotBody, expBody) {
  655. t.Errorf("%s, handler returned wrong body: got %v want %v",
  656. c.msg, gotBody, expBody)
  657. }
  658. }