gitea源码

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213
  1. // Copyright 2019 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package webhook
  4. import (
  5. "testing"
  6. webhook_model "code.gitea.io/gitea/models/webhook"
  7. "code.gitea.io/gitea/modules/json"
  8. api "code.gitea.io/gitea/modules/structs"
  9. webhook_module "code.gitea.io/gitea/modules/webhook"
  10. "github.com/stretchr/testify/assert"
  11. "github.com/stretchr/testify/require"
  12. )
  13. func TestSlackPayload(t *testing.T) {
  14. sc := slackConvertor{}
  15. t.Run("Create", func(t *testing.T) {
  16. p := createTestPayload()
  17. pl, err := sc.Create(p)
  18. require.NoError(t, err)
  19. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>:<http://localhost:3000/test/repo/src/branch/test|test>] branch created by user1", pl.Text)
  20. })
  21. t.Run("Delete", func(t *testing.T) {
  22. p := deleteTestPayload()
  23. pl, err := sc.Delete(p)
  24. require.NoError(t, err)
  25. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>:test] branch deleted by user1", pl.Text)
  26. })
  27. t.Run("Fork", func(t *testing.T) {
  28. p := forkTestPayload()
  29. pl, err := sc.Fork(p)
  30. require.NoError(t, err)
  31. assert.Equal(t, "<http://localhost:3000/test/repo2|test/repo2> is forked to <http://localhost:3000/test/repo|test/repo>", pl.Text)
  32. })
  33. t.Run("Push", func(t *testing.T) {
  34. p := pushTestPayload()
  35. pl, err := sc.Push(p)
  36. require.NoError(t, err)
  37. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>:<http://localhost:3000/test/repo/src/branch/test|test>] 2 new commits pushed by user1", pl.Text)
  38. })
  39. t.Run("Issue", func(t *testing.T) {
  40. p := issueTestPayload()
  41. p.Action = api.HookIssueOpened
  42. pl, err := sc.Issue(p)
  43. require.NoError(t, err)
  44. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] Issue opened: <http://localhost:3000/test/repo/issues/2|#2 crash> by <https://try.gitea.io/user1|user1>", pl.Text)
  45. p.Action = api.HookIssueClosed
  46. pl, err = sc.Issue(p)
  47. require.NoError(t, err)
  48. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] Issue closed: <http://localhost:3000/test/repo/issues/2|#2 crash> by <https://try.gitea.io/user1|user1>", pl.Text)
  49. })
  50. t.Run("IssueComment", func(t *testing.T) {
  51. p := issueCommentTestPayload()
  52. pl, err := sc.IssueComment(p)
  53. require.NoError(t, err)
  54. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] New comment on issue <http://localhost:3000/test/repo/issues/2|#2 crash> by <https://try.gitea.io/user1|user1>", pl.Text)
  55. })
  56. t.Run("PullRequest", func(t *testing.T) {
  57. p := pullRequestTestPayload()
  58. pl, err := sc.PullRequest(p)
  59. require.NoError(t, err)
  60. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] Pull request opened: <http://localhost:3000/test/repo/pulls/12|#12 Fix bug> by <https://try.gitea.io/user1|user1>", pl.Text)
  61. })
  62. t.Run("PullRequestComment", func(t *testing.T) {
  63. p := pullRequestCommentTestPayload()
  64. pl, err := sc.IssueComment(p)
  65. require.NoError(t, err)
  66. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] New comment on pull request <http://localhost:3000/test/repo/pulls/12|#12 Fix bug> by <https://try.gitea.io/user1|user1>", pl.Text)
  67. })
  68. t.Run("Review", func(t *testing.T) {
  69. p := pullRequestTestPayload()
  70. p.Action = api.HookIssueReviewed
  71. pl, err := sc.Review(p, webhook_module.HookEventPullRequestReviewApproved)
  72. require.NoError(t, err)
  73. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] Pull request review approved: [#12 Fix bug](http://localhost:3000/test/repo/pulls/12) by <https://try.gitea.io/user1|user1>", pl.Text)
  74. })
  75. t.Run("Repository", func(t *testing.T) {
  76. p := repositoryTestPayload()
  77. pl, err := sc.Repository(p)
  78. require.NoError(t, err)
  79. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] Repository created by <https://try.gitea.io/user1|user1>", pl.Text)
  80. })
  81. t.Run("Package", func(t *testing.T) {
  82. p := packageTestPayload()
  83. pl, err := sc.Package(p)
  84. require.NoError(t, err)
  85. assert.Equal(t, "Package created: <http://localhost:3000/user1/-/packages/container/GiteaContainer/latest|GiteaContainer:latest> by <https://try.gitea.io/user1|user1>", pl.Text)
  86. })
  87. t.Run("Wiki", func(t *testing.T) {
  88. p := wikiTestPayload()
  89. p.Action = api.HookWikiCreated
  90. pl, err := sc.Wiki(p)
  91. require.NoError(t, err)
  92. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] New wiki page '<http://localhost:3000/test/repo/wiki/index|index>' (Wiki change comment) by <https://try.gitea.io/user1|user1>", pl.Text)
  93. p.Action = api.HookWikiEdited
  94. pl, err = sc.Wiki(p)
  95. require.NoError(t, err)
  96. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] Wiki page '<http://localhost:3000/test/repo/wiki/index|index>' edited (Wiki change comment) by <https://try.gitea.io/user1|user1>", pl.Text)
  97. p.Action = api.HookWikiDeleted
  98. pl, err = sc.Wiki(p)
  99. require.NoError(t, err)
  100. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] Wiki page '<http://localhost:3000/test/repo/wiki/index|index>' deleted by <https://try.gitea.io/user1|user1>", pl.Text)
  101. })
  102. t.Run("Release", func(t *testing.T) {
  103. p := pullReleaseTestPayload()
  104. pl, err := sc.Release(p)
  105. require.NoError(t, err)
  106. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>] Release created: <http://localhost:3000/test/repo/releases/tag/v1.0|v1.0> by <https://try.gitea.io/user1|user1>", pl.Text)
  107. })
  108. }
  109. func TestSlackJSONPayload(t *testing.T) {
  110. p := pushTestPayload()
  111. data, err := p.JSONPayload()
  112. require.NoError(t, err)
  113. hook := &webhook_model.Webhook{
  114. RepoID: 3,
  115. IsActive: true,
  116. Type: webhook_module.SLACK,
  117. URL: "https://slack.example.com/",
  118. Meta: `{}`,
  119. HTTPMethod: "POST",
  120. }
  121. task := &webhook_model.HookTask{
  122. HookID: hook.ID,
  123. EventType: webhook_module.HookEventPush,
  124. PayloadContent: string(data),
  125. PayloadVersion: 2,
  126. }
  127. req, reqBody, err := newSlackRequest(t.Context(), hook, task)
  128. require.NotNil(t, req)
  129. require.NotNil(t, reqBody)
  130. require.NoError(t, err)
  131. assert.Equal(t, "POST", req.Method)
  132. assert.Equal(t, "https://slack.example.com/", req.URL.String())
  133. assert.Equal(t, "sha256=", req.Header.Get("X-Hub-Signature-256"))
  134. assert.Equal(t, "application/json", req.Header.Get("Content-Type"))
  135. var body SlackPayload
  136. err = json.NewDecoder(req.Body).Decode(&body)
  137. assert.NoError(t, err)
  138. assert.Equal(t, "[<http://localhost:3000/test/repo|test/repo>:<http://localhost:3000/test/repo/src/branch/test|test>] 2 new commits pushed by user1", body.Text)
  139. }
  140. func TestIsValidSlackChannel(t *testing.T) {
  141. tt := []struct {
  142. channelName string
  143. expected bool
  144. }{
  145. {"gitea", true},
  146. {"#gitea", true},
  147. {" ", false},
  148. {"#", false},
  149. {" #", false},
  150. {"gitea ", false},
  151. {" gitea", false},
  152. }
  153. for _, v := range tt {
  154. assert.Equal(t, v.expected, IsValidSlackChannel(v.channelName))
  155. }
  156. }