release_handler_test.go 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507
  1. package api_test
  2. // import (
  3. // "encoding/json"
  4. // "net/http"
  5. // "net/http/httptest"
  6. // "net/url"
  7. // "reflect"
  8. // "strings"
  9. // "testing"
  10. // "github.com/porter-dev/porter/internal/helm"
  11. // "helm.sh/helm/v3/pkg/chart"
  12. // "helm.sh/helm/v3/pkg/release"
  13. // "helm.sh/helm/v3/pkg/storage/driver"
  14. // )
  15. // type releaseStub struct {
  16. // name string
  17. // namespace string
  18. // version int
  19. // releaseVersion string
  20. // status release.Status
  21. // }
  22. // // ------------------------- TEST TYPES AND MAIN LOOP ------------------------- //
  23. // type releaseTest struct {
  24. // initializers []func(tester *tester)
  25. // namespace string
  26. // msg string
  27. // method string
  28. // endpoint string
  29. // body string
  30. // expStatus int
  31. // expBody string
  32. // useCookie bool
  33. // validators []func(c *releaseTest, tester *tester, t *testing.T)
  34. // }
  35. // func testReleaseRequests(t *testing.T, tests []*releaseTest, canQuery bool) {
  36. // for _, c := range tests {
  37. // // create a new tester
  38. // tester := newTester(canQuery)
  39. // // if there's an initializer, call it
  40. // for _, init := range c.initializers {
  41. // init(tester)
  42. // }
  43. // tester.app.TestAgents.HelmAgent.ActionConfig.Releases.Driver.(*driver.Memory).SetNamespace(c.namespace)
  44. // req, err := http.NewRequest(
  45. // c.method,
  46. // c.endpoint,
  47. // strings.NewReader(c.body),
  48. // )
  49. // tester.req = req
  50. // if c.useCookie {
  51. // req.AddCookie(tester.cookie)
  52. // }
  53. // if err != nil {
  54. // t.Fatal(err)
  55. // }
  56. // tester.execute()
  57. // rr := tester.rr
  58. // // first, check that the status matches
  59. // if status := rr.Code; status != c.expStatus {
  60. // t.Errorf("%s, handler returned wrong status code: got %v want %v",
  61. // c.msg, status, c.expStatus)
  62. // }
  63. // // if there's a validator, call it
  64. // for _, validate := range c.validators {
  65. // validate(c, tester, t)
  66. // }
  67. // }
  68. // }
  69. // // ------------------------- TEST FIXTURES AND FUNCTIONS ------------------------- //
  70. // var listReleasesTests = []*releaseTest{
  71. // {
  72. // initializers: []func(tester *tester){
  73. // initDefaultReleases,
  74. // },
  75. // msg: "List releases no namespace",
  76. // method: "GET",
  77. // endpoint: "/api/projects/1/releases?" + url.Values{
  78. // "namespace": []string{""},
  79. // "cluster_id": []string{"1"},
  80. // "storage": []string{"memory"},
  81. // "limit": []string{"20"},
  82. // "skip": []string{"0"},
  83. // "byDate": []string{"false"},
  84. // "statusFilter": []string{"deployed"},
  85. // }.Encode(),
  86. // body: "",
  87. // expStatus: http.StatusOK,
  88. // expBody: releaseStubsToReleaseJSON(sampleReleaseStubs),
  89. // useCookie: true,
  90. // validators: []func(c *releaseTest, tester *tester, t *testing.T){
  91. // releaseReleaseArrBodyValidator,
  92. // },
  93. // },
  94. // {
  95. // initializers: []func(tester *tester){
  96. // initDefaultReleases,
  97. // },
  98. // msg: "List releases with namespace",
  99. // method: "GET",
  100. // namespace: "default",
  101. // endpoint: "/api/projects/1/releases?" + url.Values{
  102. // "namespace": []string{"default"},
  103. // "cluster_id": []string{"1"},
  104. // "storage": []string{"memory"},
  105. // "limit": []string{"20"},
  106. // "skip": []string{"0"},
  107. // "byDate": []string{"false"},
  108. // "statusFilter": []string{"deployed"},
  109. // }.Encode(),
  110. // body: "",
  111. // expStatus: http.StatusOK,
  112. // expBody: releaseStubsToReleaseJSON([]releaseStub{
  113. // sampleReleaseStubs[0],
  114. // sampleReleaseStubs[2],
  115. // }),
  116. // useCookie: true,
  117. // validators: []func(c *releaseTest, tester *tester, t *testing.T){
  118. // releaseReleaseArrBodyValidator,
  119. // },
  120. // },
  121. // }
  122. // func TestHandleListReleases(t *testing.T) {
  123. // testReleaseRequests(t, listReleasesTests, true)
  124. // }
  125. // var getReleaseTests = []*releaseTest{
  126. // {
  127. // initializers: []func(tester *tester){
  128. // initDefaultReleases,
  129. // },
  130. // msg: "Get releases",
  131. // method: "GET",
  132. // namespace: "default",
  133. // endpoint: "/api/projects/1/releases/airwatch/1?" + url.Values{
  134. // "namespace": []string{"default"},
  135. // "cluster_id": []string{"1"},
  136. // "storage": []string{"memory"},
  137. // }.Encode(),
  138. // body: "",
  139. // expStatus: http.StatusOK,
  140. // expBody: releaseStubToReleaseJSON(sampleReleaseStubs[0]),
  141. // useCookie: true,
  142. // validators: []func(c *releaseTest, tester *tester, t *testing.T){
  143. // releaseReleaseBodyValidator,
  144. // },
  145. // },
  146. // {
  147. // initializers: []func(tester *tester){
  148. // initDefaultReleases,
  149. // },
  150. // msg: "Release not found",
  151. // method: "GET",
  152. // namespace: "default",
  153. // endpoint: "/api/projects/1/releases/airwatch/5?" + url.Values{
  154. // "namespace": []string{""},
  155. // "cluster_id": []string{"1"},
  156. // "storage": []string{"memory"},
  157. // }.Encode(),
  158. // body: "",
  159. // expStatus: http.StatusNotFound,
  160. // expBody: `{"code":602,"errors":["release not found"]}`,
  161. // useCookie: true,
  162. // validators: []func(c *releaseTest, tester *tester, t *testing.T){
  163. // releaseBasicBodyValidator,
  164. // },
  165. // },
  166. // }
  167. // func TestHandleGetRelease(t *testing.T) {
  168. // testReleaseRequests(t, getReleaseTests, true)
  169. // }
  170. // var listReleaseHistoryTests = []*releaseTest{
  171. // {
  172. // initializers: []func(tester *tester){
  173. // initHistoryReleases,
  174. // },
  175. // msg: "List release history",
  176. // method: "GET",
  177. // namespace: "default",
  178. // endpoint: "/api/projects/1/releases/wordpress/history?" + url.Values{
  179. // "namespace": []string{""},
  180. // "cluster_id": []string{"1"},
  181. // "storage": []string{"memory"},
  182. // }.Encode(),
  183. // body: "",
  184. // expStatus: http.StatusOK,
  185. // expBody: releaseStubsToReleaseJSON(historyReleaseStubs),
  186. // useCookie: true,
  187. // validators: []func(c *releaseTest, tester *tester, t *testing.T){
  188. // releaseReleaseArrBodyValidator,
  189. // },
  190. // },
  191. // {
  192. // initializers: []func(tester *tester){
  193. // initDefaultReleases,
  194. // },
  195. // msg: "Release not found",
  196. // method: "GET",
  197. // namespace: "default",
  198. // endpoint: "/api/projects/1/releases/asldfkja/history?" + url.Values{
  199. // "namespace": []string{""},
  200. // "cluster_id": []string{"1"},
  201. // "storage": []string{"memory"},
  202. // }.Encode(),
  203. // body: "",
  204. // expStatus: http.StatusNotFound,
  205. // expBody: `{"code":602,"errors":["release not found"]}`,
  206. // useCookie: true,
  207. // validators: []func(c *releaseTest, tester *tester, t *testing.T){
  208. // releaseBasicBodyValidator,
  209. // },
  210. // },
  211. // }
  212. // func TestHandleListReleaseHistory(t *testing.T) {
  213. // testReleaseRequests(t, listReleaseHistoryTests, true)
  214. // }
  215. // var upgradeReleaseTests = []*releaseTest{
  216. // {
  217. // initializers: []func(tester *tester){
  218. // initHistoryReleases,
  219. // },
  220. // msg: "Upgrade relase",
  221. // method: "POST",
  222. // namespace: "default",
  223. // endpoint: "/api/projects/1/releases/wordpress/upgrade?" + url.Values{
  224. // "cluster_id": []string{"1"},
  225. // }.Encode(),
  226. // body: `
  227. // {
  228. // "namespace": "default",
  229. // "storage": "memory",
  230. // "values": "\nfoo: bar\n"
  231. // }
  232. // `,
  233. // expStatus: http.StatusOK,
  234. // expBody: ``,
  235. // useCookie: true,
  236. // validators: []func(c *releaseTest, tester *tester, t *testing.T){
  237. // func(c *releaseTest, tester *tester, t *testing.T) {
  238. // req, err := http.NewRequest(
  239. // "GET",
  240. // "/api/projects/1/releases/wordpress/3?"+url.Values{
  241. // "namespace": []string{"default"},
  242. // "cluster_id": []string{"1"},
  243. // "storage": []string{"memory"},
  244. // }.Encode(),
  245. // strings.NewReader(""),
  246. // )
  247. // req.AddCookie(tester.cookie)
  248. // if err != nil {
  249. // t.Fatal(err)
  250. // }
  251. // rr2 := httptest.NewRecorder()
  252. // tester.router.ServeHTTP(rr2, req)
  253. // gotBody := &release.Release{}
  254. // expBody := &release.Release{}
  255. // expBodyJSON := releaseStubToReleaseJSON(releaseStub{"wordpress", "default", 3, "1.0.2", release.StatusDeployed})
  256. // json.Unmarshal(rr2.Body.Bytes(), gotBody)
  257. // json.Unmarshal([]byte(expBodyJSON), expBody)
  258. // // just check name and version match, other items will be different
  259. // if gotBody.Name != expBody.Name {
  260. // t.Errorf("%s, validation wrong body: got %v want %v",
  261. // c.msg, gotBody.Name, expBody.Name)
  262. // }
  263. // if gotBody.Version != expBody.Version {
  264. // t.Errorf("%s, validation wrong body: got %v want %v",
  265. // c.msg, gotBody.Version, expBody.Version)
  266. // }
  267. // expConfig := map[string]interface{}{
  268. // "foo": "bar",
  269. // }
  270. // if !reflect.DeepEqual(gotBody.Config, expConfig) {
  271. // t.Errorf("%s, validation wrong config: got %v want %v",
  272. // c.msg, gotBody.Config, expConfig)
  273. // }
  274. // },
  275. // },
  276. // },
  277. // }
  278. // func TestUpgradeRelease(t *testing.T) {
  279. // testReleaseRequests(t, upgradeReleaseTests, true)
  280. // }
  281. // var rollbackReleaseTests = []*releaseTest{
  282. // {
  283. // initializers: []func(tester *tester){
  284. // initHistoryReleases,
  285. // },
  286. // msg: "Rollback release",
  287. // method: "POST",
  288. // namespace: "default",
  289. // endpoint: "/api/projects/1/releases/wordpress/rollback?" + url.Values{
  290. // "cluster_id": []string{"1"},
  291. // }.Encode(),
  292. // body: `
  293. // {
  294. // "namespace": "default",
  295. // "storage": "memory",
  296. // "revision": 1
  297. // }
  298. // `,
  299. // expStatus: http.StatusOK,
  300. // expBody: ``,
  301. // useCookie: true,
  302. // validators: []func(c *releaseTest, tester *tester, t *testing.T){
  303. // func(c *releaseTest, tester *tester, t *testing.T) {
  304. // req, err := http.NewRequest(
  305. // "GET",
  306. // "/api/projects/1/releases/wordpress/3?"+url.Values{
  307. // "namespace": []string{"default"},
  308. // "cluster_id": []string{"1"},
  309. // "storage": []string{"memory"},
  310. // }.Encode(),
  311. // strings.NewReader(""),
  312. // )
  313. // req.AddCookie(tester.cookie)
  314. // if err != nil {
  315. // t.Fatal(err)
  316. // }
  317. // rr2 := httptest.NewRecorder()
  318. // tester.router.ServeHTTP(rr2, req)
  319. // gotBody := &release.Release{}
  320. // expBody := &release.Release{}
  321. // expBodyJSON := releaseStubToReleaseJSON(releaseStub{"wordpress", "default", 3, "1.0.1", release.StatusDeployed})
  322. // json.Unmarshal(rr2.Body.Bytes(), gotBody)
  323. // json.Unmarshal([]byte(expBodyJSON), expBody)
  324. // // just check name and version match, other items will be different
  325. // if gotBody.Name != expBody.Name {
  326. // t.Errorf("%s, validation wrong body: got %v want %v",
  327. // c.msg, gotBody.Name, expBody.Name)
  328. // }
  329. // if gotBody.Version != expBody.Version {
  330. // t.Errorf("%s, validation wrong body: got %v want %v",
  331. // c.msg, gotBody.Version, expBody.Version)
  332. // }
  333. // },
  334. // },
  335. // },
  336. // }
  337. // func TestRollbackRelease(t *testing.T) {
  338. // testReleaseRequests(t, rollbackReleaseTests, true)
  339. // }
  340. // // ------------------------- INITIALIZERS AND VALIDATORS ------------------------- //
  341. // func initDefaultReleases(tester *tester) {
  342. // initUserDefault(tester)
  343. // initProject(tester)
  344. // initProjectClusterDefault(tester)
  345. // agent := tester.app.TestAgents.HelmAgent
  346. // makeReleases(agent, sampleReleaseStubs)
  347. // // calling agent.ActionConfig.Releases.Create in makeReleases will automatically set the
  348. // // namespace, so we have to reset the namespace of the storage driver
  349. // agent.ActionConfig.Releases.Driver.(*driver.Memory).SetNamespace("")
  350. // }
  351. // func initHistoryReleases(tester *tester) {
  352. // initUserDefault(tester)
  353. // initProject(tester)
  354. // initProjectClusterDefault(tester)
  355. // agent := tester.app.TestAgents.HelmAgent
  356. // makeReleases(agent, historyReleaseStubs)
  357. // // calling agent.ActionConfig.Releases.Create in makeReleases will automatically set the
  358. // // namespace, so we have to reset the namespace of the storage driver
  359. // agent.ActionConfig.Releases.Driver.(*driver.Memory).SetNamespace("")
  360. // }
  361. // var sampleReleaseStubs = []releaseStub{
  362. // {"airwatch", "default", 1, "1.0.0", release.StatusDeployed},
  363. // {"not-in-default-namespace", "other", 1, "1.0.1", release.StatusDeployed},
  364. // {"wordpress", "default", 1, "1.0.2", release.StatusDeployed},
  365. // }
  366. // var historyReleaseStubs = []releaseStub{
  367. // {"wordpress", "default", 1, "1.0.1", release.StatusSuperseded},
  368. // {"wordpress", "default", 2, "1.0.2", release.StatusDeployed},
  369. // }
  370. // func releaseStubsToReleaseJSON(rels []releaseStub) string {
  371. // releases := make([]*release.Release, 0)
  372. // for _, r := range rels {
  373. // rel := releaseStubToRelease(r)
  374. // releases = append(releases, rel)
  375. // }
  376. // str, _ := json.Marshal(releases)
  377. // return string(str)
  378. // }
  379. // func releaseStubToReleaseJSON(r releaseStub) string {
  380. // rel := releaseStubToRelease(r)
  381. // str, _ := json.Marshal(rel)
  382. // return string(str)
  383. // }
  384. // func releaseStubToRelease(r releaseStub) *release.Release {
  385. // return &release.Release{
  386. // Name: r.name,
  387. // Namespace: r.namespace,
  388. // Version: r.version,
  389. // Info: &release.Info{
  390. // Status: r.status,
  391. // },
  392. // Chart: &chart.Chart{
  393. // Metadata: &chart.Metadata{
  394. // Version: r.releaseVersion,
  395. // Icon: "https://example.com/icon.png",
  396. // },
  397. // },
  398. // }
  399. // }
  400. // func makeReleases(agent *helm.Agent, rels []releaseStub) {
  401. // storage := agent.ActionConfig.Releases
  402. // for _, r := range rels {
  403. // rel := releaseStubToRelease(r)
  404. // storage.Create(rel)
  405. // }
  406. // }
  407. // func releaseBasicBodyValidator(c *releaseTest, tester *tester, t *testing.T) {
  408. // if body := tester.rr.Body.String(); strings.TrimSpace(body) != strings.TrimSpace(c.expBody) {
  409. // t.Errorf("%s, handler returned wrong body: got %v want %v",
  410. // c.msg, body, c.expBody)
  411. // }
  412. // }
  413. // func releaseReleaseBodyValidator(c *releaseTest, tester *tester, t *testing.T) {
  414. // gotBody := &release.Release{}
  415. // expBody := &release.Release{}
  416. // json.Unmarshal(tester.rr.Body.Bytes(), gotBody)
  417. // json.Unmarshal([]byte(c.expBody), expBody)
  418. // if !reflect.DeepEqual(gotBody, expBody) {
  419. // t.Errorf("%s, handler returned wrong body: got %v want %v",
  420. // c.msg, gotBody, expBody)
  421. // }
  422. // }
  423. // func releaseReleaseArrBodyValidator(c *releaseTest, tester *tester, t *testing.T) {
  424. // gotBody := &[]release.Release{}
  425. // expBody := &[]release.Release{}
  426. // json.Unmarshal(tester.rr.Body.Bytes(), gotBody)
  427. // json.Unmarshal([]byte(c.expBody), expBody)
  428. // if !reflect.DeepEqual(gotBody, expBody) {
  429. // t.Errorf("%s, handler returned wrong body: got %v want %v",
  430. // c.msg, gotBody, expBody)
  431. // }
  432. // }