You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

git_test.go 33KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893
  1. // Copyright 2017 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package integration
  4. import (
  5. "bytes"
  6. "encoding/hex"
  7. "fmt"
  8. "math/rand"
  9. "net/http"
  10. "net/url"
  11. "os"
  12. "path"
  13. "path/filepath"
  14. "strconv"
  15. "testing"
  16. "time"
  17. auth_model "code.gitea.io/gitea/models/auth"
  18. "code.gitea.io/gitea/models/db"
  19. issues_model "code.gitea.io/gitea/models/issues"
  20. "code.gitea.io/gitea/models/perm"
  21. repo_model "code.gitea.io/gitea/models/repo"
  22. "code.gitea.io/gitea/models/unittest"
  23. user_model "code.gitea.io/gitea/models/user"
  24. gitea_context "code.gitea.io/gitea/modules/context"
  25. "code.gitea.io/gitea/modules/git"
  26. "code.gitea.io/gitea/modules/lfs"
  27. "code.gitea.io/gitea/modules/setting"
  28. api "code.gitea.io/gitea/modules/structs"
  29. files_service "code.gitea.io/gitea/services/repository/files"
  30. "code.gitea.io/gitea/tests"
  31. "github.com/stretchr/testify/assert"
  32. )
  33. const (
  34. littleSize = 1024 // 1ko
  35. bigSize = 128 * 1024 * 1024 // 128Mo
  36. )
  37. func TestGit(t *testing.T) {
  38. onGiteaRun(t, testGit)
  39. }
  40. func testGit(t *testing.T, u *url.URL) {
  41. username := "user2"
  42. baseAPITestContext := NewAPITestContext(t, username, "repo1", auth_model.AccessTokenScopeWriteRepository, auth_model.AccessTokenScopeWriteUser)
  43. u.Path = baseAPITestContext.GitPath()
  44. forkedUserCtx := NewAPITestContext(t, "user4", "repo1", auth_model.AccessTokenScopeWriteRepository, auth_model.AccessTokenScopeWriteUser)
  45. t.Run("HTTP", func(t *testing.T) {
  46. defer tests.PrintCurrentTest(t)()
  47. ensureAnonymousClone(t, u)
  48. httpContext := baseAPITestContext
  49. httpContext.Reponame = "repo-tmp-17"
  50. forkedUserCtx.Reponame = httpContext.Reponame
  51. dstPath := t.TempDir()
  52. t.Run("CreateRepoInDifferentUser", doAPICreateRepository(forkedUserCtx, false))
  53. t.Run("AddUserAsCollaborator", doAPIAddCollaborator(forkedUserCtx, httpContext.Username, perm.AccessModeRead))
  54. t.Run("ForkFromDifferentUser", doAPIForkRepository(httpContext, forkedUserCtx.Username))
  55. u.Path = httpContext.GitPath()
  56. u.User = url.UserPassword(username, userPassword)
  57. t.Run("Clone", doGitClone(dstPath, u))
  58. dstPath2 := t.TempDir()
  59. t.Run("Partial Clone", doPartialGitClone(dstPath2, u))
  60. little, big := standardCommitAndPushTest(t, dstPath)
  61. littleLFS, bigLFS := lfsCommitAndPushTest(t, dstPath)
  62. rawTest(t, &httpContext, little, big, littleLFS, bigLFS)
  63. mediaTest(t, &httpContext, little, big, littleLFS, bigLFS)
  64. t.Run("CreateAgitFlowPull", doCreateAgitFlowPull(dstPath, &httpContext, "master", "test/head"))
  65. t.Run("BranchProtectMerge", doBranchProtectPRMerge(&httpContext, dstPath))
  66. t.Run("AutoMerge", doAutoPRMerge(&httpContext, dstPath))
  67. t.Run("CreatePRAndSetManuallyMerged", doCreatePRAndSetManuallyMerged(httpContext, httpContext, dstPath, "master", "test-manually-merge"))
  68. t.Run("MergeFork", func(t *testing.T) {
  69. defer tests.PrintCurrentTest(t)()
  70. t.Run("CreatePRAndMerge", doMergeFork(httpContext, forkedUserCtx, "master", httpContext.Username+":master"))
  71. rawTest(t, &forkedUserCtx, little, big, littleLFS, bigLFS)
  72. mediaTest(t, &forkedUserCtx, little, big, littleLFS, bigLFS)
  73. })
  74. t.Run("PushCreate", doPushCreate(httpContext, u))
  75. })
  76. t.Run("SSH", func(t *testing.T) {
  77. defer tests.PrintCurrentTest(t)()
  78. sshContext := baseAPITestContext
  79. sshContext.Reponame = "repo-tmp-18"
  80. keyname := "my-testing-key"
  81. forkedUserCtx.Reponame = sshContext.Reponame
  82. t.Run("CreateRepoInDifferentUser", doAPICreateRepository(forkedUserCtx, false))
  83. t.Run("AddUserAsCollaborator", doAPIAddCollaborator(forkedUserCtx, sshContext.Username, perm.AccessModeRead))
  84. t.Run("ForkFromDifferentUser", doAPIForkRepository(sshContext, forkedUserCtx.Username))
  85. // Setup key the user ssh key
  86. withKeyFile(t, keyname, func(keyFile string) {
  87. t.Run("CreateUserKey", doAPICreateUserKey(sshContext, "test-key", keyFile))
  88. // Setup remote link
  89. // TODO: get url from api
  90. sshURL := createSSHUrl(sshContext.GitPath(), u)
  91. // Setup clone folder
  92. dstPath := t.TempDir()
  93. t.Run("Clone", doGitClone(dstPath, sshURL))
  94. little, big := standardCommitAndPushTest(t, dstPath)
  95. littleLFS, bigLFS := lfsCommitAndPushTest(t, dstPath)
  96. rawTest(t, &sshContext, little, big, littleLFS, bigLFS)
  97. mediaTest(t, &sshContext, little, big, littleLFS, bigLFS)
  98. t.Run("CreateAgitFlowPull", doCreateAgitFlowPull(dstPath, &sshContext, "master", "test/head2"))
  99. t.Run("BranchProtectMerge", doBranchProtectPRMerge(&sshContext, dstPath))
  100. t.Run("MergeFork", func(t *testing.T) {
  101. defer tests.PrintCurrentTest(t)()
  102. t.Run("CreatePRAndMerge", doMergeFork(sshContext, forkedUserCtx, "master", sshContext.Username+":master"))
  103. rawTest(t, &forkedUserCtx, little, big, littleLFS, bigLFS)
  104. mediaTest(t, &forkedUserCtx, little, big, littleLFS, bigLFS)
  105. })
  106. t.Run("PushCreate", doPushCreate(sshContext, sshURL))
  107. })
  108. })
  109. }
  110. func ensureAnonymousClone(t *testing.T, u *url.URL) {
  111. dstLocalPath := t.TempDir()
  112. t.Run("CloneAnonymous", doGitClone(dstLocalPath, u))
  113. }
  114. func standardCommitAndPushTest(t *testing.T, dstPath string) (little, big string) {
  115. t.Run("Standard", func(t *testing.T) {
  116. defer tests.PrintCurrentTest(t)()
  117. little, big = commitAndPushTest(t, dstPath, "data-file-")
  118. })
  119. return little, big
  120. }
  121. func lfsCommitAndPushTest(t *testing.T, dstPath string) (littleLFS, bigLFS string) {
  122. t.Run("LFS", func(t *testing.T) {
  123. defer tests.PrintCurrentTest(t)()
  124. prefix := "lfs-data-file-"
  125. err := git.NewCommand(git.DefaultContext, "lfs").AddArguments("install").Run(&git.RunOpts{Dir: dstPath})
  126. assert.NoError(t, err)
  127. _, _, err = git.NewCommand(git.DefaultContext, "lfs").AddArguments("track").AddDynamicArguments(prefix + "*").RunStdString(&git.RunOpts{Dir: dstPath})
  128. assert.NoError(t, err)
  129. err = git.AddChanges(dstPath, false, ".gitattributes")
  130. assert.NoError(t, err)
  131. err = git.CommitChangesWithArgs(dstPath, git.AllowLFSFiltersArgs(), git.CommitChangesOptions{
  132. Committer: &git.Signature{
  133. Email: "user2@example.com",
  134. Name: "User Two",
  135. When: time.Now(),
  136. },
  137. Author: &git.Signature{
  138. Email: "user2@example.com",
  139. Name: "User Two",
  140. When: time.Now(),
  141. },
  142. Message: fmt.Sprintf("Testing commit @ %v", time.Now()),
  143. })
  144. assert.NoError(t, err)
  145. littleLFS, bigLFS = commitAndPushTest(t, dstPath, prefix)
  146. t.Run("Locks", func(t *testing.T) {
  147. defer tests.PrintCurrentTest(t)()
  148. lockTest(t, dstPath)
  149. })
  150. })
  151. return littleLFS, bigLFS
  152. }
  153. func commitAndPushTest(t *testing.T, dstPath, prefix string) (little, big string) {
  154. t.Run("PushCommit", func(t *testing.T) {
  155. defer tests.PrintCurrentTest(t)()
  156. t.Run("Little", func(t *testing.T) {
  157. defer tests.PrintCurrentTest(t)()
  158. little = doCommitAndPush(t, littleSize, dstPath, prefix)
  159. })
  160. t.Run("Big", func(t *testing.T) {
  161. if testing.Short() {
  162. t.Skip("Skipping test in short mode.")
  163. return
  164. }
  165. defer tests.PrintCurrentTest(t)()
  166. big = doCommitAndPush(t, bigSize, dstPath, prefix)
  167. })
  168. })
  169. return little, big
  170. }
  171. func rawTest(t *testing.T, ctx *APITestContext, little, big, littleLFS, bigLFS string) {
  172. t.Run("Raw", func(t *testing.T) {
  173. defer tests.PrintCurrentTest(t)()
  174. username := ctx.Username
  175. reponame := ctx.Reponame
  176. session := loginUser(t, username)
  177. // Request raw paths
  178. req := NewRequest(t, "GET", path.Join("/", username, reponame, "/raw/branch/master/", little))
  179. resp := session.MakeRequestNilResponseRecorder(t, req, http.StatusOK)
  180. assert.Equal(t, littleSize, resp.Length)
  181. if setting.LFS.StartServer {
  182. req = NewRequest(t, "GET", path.Join("/", username, reponame, "/raw/branch/master/", littleLFS))
  183. resp := session.MakeRequest(t, req, http.StatusOK)
  184. assert.NotEqual(t, littleSize, resp.Body.Len())
  185. assert.LessOrEqual(t, resp.Body.Len(), 1024)
  186. if resp.Body.Len() != littleSize && resp.Body.Len() <= 1024 {
  187. assert.Contains(t, resp.Body.String(), lfs.MetaFileIdentifier)
  188. }
  189. }
  190. if !testing.Short() {
  191. req = NewRequest(t, "GET", path.Join("/", username, reponame, "/raw/branch/master/", big))
  192. resp := session.MakeRequestNilResponseRecorder(t, req, http.StatusOK)
  193. assert.Equal(t, bigSize, resp.Length)
  194. if setting.LFS.StartServer {
  195. req = NewRequest(t, "GET", path.Join("/", username, reponame, "/raw/branch/master/", bigLFS))
  196. resp := session.MakeRequest(t, req, http.StatusOK)
  197. assert.NotEqual(t, bigSize, resp.Body.Len())
  198. if resp.Body.Len() != bigSize && resp.Body.Len() <= 1024 {
  199. assert.Contains(t, resp.Body.String(), lfs.MetaFileIdentifier)
  200. }
  201. }
  202. }
  203. })
  204. }
  205. func mediaTest(t *testing.T, ctx *APITestContext, little, big, littleLFS, bigLFS string) {
  206. t.Run("Media", func(t *testing.T) {
  207. defer tests.PrintCurrentTest(t)()
  208. username := ctx.Username
  209. reponame := ctx.Reponame
  210. session := loginUser(t, username)
  211. // Request media paths
  212. req := NewRequest(t, "GET", path.Join("/", username, reponame, "/media/branch/master/", little))
  213. resp := session.MakeRequestNilResponseRecorder(t, req, http.StatusOK)
  214. assert.Equal(t, littleSize, resp.Length)
  215. req = NewRequest(t, "GET", path.Join("/", username, reponame, "/media/branch/master/", littleLFS))
  216. resp = session.MakeRequestNilResponseRecorder(t, req, http.StatusOK)
  217. assert.Equal(t, littleSize, resp.Length)
  218. if !testing.Short() {
  219. req = NewRequest(t, "GET", path.Join("/", username, reponame, "/media/branch/master/", big))
  220. resp = session.MakeRequestNilResponseRecorder(t, req, http.StatusOK)
  221. assert.Equal(t, bigSize, resp.Length)
  222. if setting.LFS.StartServer {
  223. req = NewRequest(t, "GET", path.Join("/", username, reponame, "/media/branch/master/", bigLFS))
  224. resp = session.MakeRequestNilResponseRecorder(t, req, http.StatusOK)
  225. assert.Equal(t, bigSize, resp.Length)
  226. }
  227. }
  228. })
  229. }
  230. func lockTest(t *testing.T, repoPath string) {
  231. lockFileTest(t, "README.md", repoPath)
  232. }
  233. func lockFileTest(t *testing.T, filename, repoPath string) {
  234. _, _, err := git.NewCommand(git.DefaultContext, "lfs").AddArguments("locks").RunStdString(&git.RunOpts{Dir: repoPath})
  235. assert.NoError(t, err)
  236. _, _, err = git.NewCommand(git.DefaultContext, "lfs").AddArguments("lock").AddDynamicArguments(filename).RunStdString(&git.RunOpts{Dir: repoPath})
  237. assert.NoError(t, err)
  238. _, _, err = git.NewCommand(git.DefaultContext, "lfs").AddArguments("locks").RunStdString(&git.RunOpts{Dir: repoPath})
  239. assert.NoError(t, err)
  240. _, _, err = git.NewCommand(git.DefaultContext, "lfs").AddArguments("unlock").AddDynamicArguments(filename).RunStdString(&git.RunOpts{Dir: repoPath})
  241. assert.NoError(t, err)
  242. }
  243. func doCommitAndPush(t *testing.T, size int, repoPath, prefix string) string {
  244. name, err := generateCommitWithNewData(size, repoPath, "user2@example.com", "User Two", prefix)
  245. assert.NoError(t, err)
  246. _, _, err = git.NewCommand(git.DefaultContext, "push", "origin", "master").RunStdString(&git.RunOpts{Dir: repoPath}) // Push
  247. assert.NoError(t, err)
  248. return name
  249. }
  250. func generateCommitWithNewData(size int, repoPath, email, fullName, prefix string) (string, error) {
  251. // Generate random file
  252. bufSize := 4 * 1024
  253. if bufSize > size {
  254. bufSize = size
  255. }
  256. buffer := make([]byte, bufSize)
  257. tmpFile, err := os.CreateTemp(repoPath, prefix)
  258. if err != nil {
  259. return "", err
  260. }
  261. defer tmpFile.Close()
  262. written := 0
  263. for written < size {
  264. n := size - written
  265. if n > bufSize {
  266. n = bufSize
  267. }
  268. _, err := rand.Read(buffer[:n])
  269. if err != nil {
  270. return "", err
  271. }
  272. n, err = tmpFile.Write(buffer[:n])
  273. if err != nil {
  274. return "", err
  275. }
  276. written += n
  277. }
  278. if err != nil {
  279. return "", err
  280. }
  281. // Commit
  282. // Now here we should explicitly allow lfs filters to run
  283. globalArgs := git.AllowLFSFiltersArgs()
  284. err = git.AddChangesWithArgs(repoPath, globalArgs, false, filepath.Base(tmpFile.Name()))
  285. if err != nil {
  286. return "", err
  287. }
  288. err = git.CommitChangesWithArgs(repoPath, globalArgs, git.CommitChangesOptions{
  289. Committer: &git.Signature{
  290. Email: email,
  291. Name: fullName,
  292. When: time.Now(),
  293. },
  294. Author: &git.Signature{
  295. Email: email,
  296. Name: fullName,
  297. When: time.Now(),
  298. },
  299. Message: fmt.Sprintf("Testing commit @ %v", time.Now()),
  300. })
  301. return filepath.Base(tmpFile.Name()), err
  302. }
  303. func doBranchProtectPRMerge(baseCtx *APITestContext, dstPath string) func(t *testing.T) {
  304. return func(t *testing.T) {
  305. defer tests.PrintCurrentTest(t)()
  306. t.Run("CreateBranchProtected", doGitCreateBranch(dstPath, "protected"))
  307. t.Run("PushProtectedBranch", doGitPushTestRepository(dstPath, "origin", "protected"))
  308. ctx := NewAPITestContext(t, baseCtx.Username, baseCtx.Reponame, auth_model.AccessTokenScopeWriteRepository)
  309. t.Run("ProtectProtectedBranchNoWhitelist", doProtectBranch(ctx, "protected", "", ""))
  310. t.Run("GenerateCommit", func(t *testing.T) {
  311. _, err := generateCommitWithNewData(littleSize, dstPath, "user2@example.com", "User Two", "branch-data-file-")
  312. assert.NoError(t, err)
  313. })
  314. t.Run("FailToPushToProtectedBranch", doGitPushTestRepositoryFail(dstPath, "origin", "protected"))
  315. t.Run("PushToUnprotectedBranch", doGitPushTestRepository(dstPath, "origin", "protected:unprotected"))
  316. var pr api.PullRequest
  317. var err error
  318. t.Run("CreatePullRequest", func(t *testing.T) {
  319. pr, err = doAPICreatePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, "protected", "unprotected")(t)
  320. assert.NoError(t, err)
  321. })
  322. t.Run("GenerateCommit", func(t *testing.T) {
  323. _, err := generateCommitWithNewData(littleSize, dstPath, "user2@example.com", "User Two", "branch-data-file-")
  324. assert.NoError(t, err)
  325. })
  326. t.Run("PushToUnprotectedBranch", doGitPushTestRepository(dstPath, "origin", "protected:unprotected-2"))
  327. var pr2 api.PullRequest
  328. t.Run("CreatePullRequest", func(t *testing.T) {
  329. pr2, err = doAPICreatePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, "unprotected", "unprotected-2")(t)
  330. assert.NoError(t, err)
  331. })
  332. t.Run("MergePR2", doAPIMergePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr2.Index))
  333. t.Run("MergePR", doAPIMergePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index))
  334. t.Run("PullProtected", doGitPull(dstPath, "origin", "protected"))
  335. t.Run("ProtectProtectedBranchUnprotectedFilePaths", doProtectBranch(ctx, "protected", "", "unprotected-file-*"))
  336. t.Run("GenerateCommit", func(t *testing.T) {
  337. _, err := generateCommitWithNewData(littleSize, dstPath, "user2@example.com", "User Two", "unprotected-file-")
  338. assert.NoError(t, err)
  339. })
  340. t.Run("PushUnprotectedFilesToProtectedBranch", doGitPushTestRepository(dstPath, "origin", "protected"))
  341. t.Run("ProtectProtectedBranchWhitelist", doProtectBranch(ctx, "protected", baseCtx.Username, ""))
  342. t.Run("CheckoutMaster", doGitCheckoutBranch(dstPath, "master"))
  343. t.Run("CreateBranchForced", doGitCreateBranch(dstPath, "toforce"))
  344. t.Run("GenerateCommit", func(t *testing.T) {
  345. _, err := generateCommitWithNewData(littleSize, dstPath, "user2@example.com", "User Two", "branch-data-file-")
  346. assert.NoError(t, err)
  347. })
  348. t.Run("FailToForcePushToProtectedBranch", doGitPushTestRepositoryFail(dstPath, "-f", "origin", "toforce:protected"))
  349. t.Run("MergeProtectedToToforce", doGitMerge(dstPath, "protected"))
  350. t.Run("PushToProtectedBranch", doGitPushTestRepository(dstPath, "origin", "toforce:protected"))
  351. t.Run("CheckoutMasterAgain", doGitCheckoutBranch(dstPath, "master"))
  352. }
  353. }
  354. func doProtectBranch(ctx APITestContext, branch, userToWhitelist, unprotectedFilePatterns string) func(t *testing.T) {
  355. // We are going to just use the owner to set the protection.
  356. return func(t *testing.T) {
  357. csrf := GetCSRF(t, ctx.Session, fmt.Sprintf("/%s/%s/settings/branches", url.PathEscape(ctx.Username), url.PathEscape(ctx.Reponame)))
  358. if userToWhitelist == "" {
  359. // Change branch to protected
  360. req := NewRequestWithValues(t, "POST", fmt.Sprintf("/%s/%s/settings/branches/edit", url.PathEscape(ctx.Username), url.PathEscape(ctx.Reponame)), map[string]string{
  361. "_csrf": csrf,
  362. "rule_name": branch,
  363. "unprotected_file_patterns": unprotectedFilePatterns,
  364. })
  365. ctx.Session.MakeRequest(t, req, http.StatusSeeOther)
  366. } else {
  367. user, err := user_model.GetUserByName(db.DefaultContext, userToWhitelist)
  368. assert.NoError(t, err)
  369. // Change branch to protected
  370. req := NewRequestWithValues(t, "POST", fmt.Sprintf("/%s/%s/settings/branches/edit", url.PathEscape(ctx.Username), url.PathEscape(ctx.Reponame)), map[string]string{
  371. "_csrf": csrf,
  372. "rule_name": branch,
  373. "enable_push": "whitelist",
  374. "enable_whitelist": "on",
  375. "whitelist_users": strconv.FormatInt(user.ID, 10),
  376. "unprotected_file_patterns": unprotectedFilePatterns,
  377. })
  378. ctx.Session.MakeRequest(t, req, http.StatusSeeOther)
  379. }
  380. // Check if master branch has been locked successfully
  381. flashCookie := ctx.Session.GetCookie(gitea_context.CookieNameFlash)
  382. assert.NotNil(t, flashCookie)
  383. assert.EqualValues(t, "success%3DBranch%2Bprotection%2Bfor%2Brule%2B%2522"+url.QueryEscape(branch)+"%2522%2Bhas%2Bbeen%2Bupdated.", flashCookie.Value)
  384. }
  385. }
  386. func doMergeFork(ctx, baseCtx APITestContext, baseBranch, headBranch string) func(t *testing.T) {
  387. return func(t *testing.T) {
  388. defer tests.PrintCurrentTest(t)()
  389. var pr api.PullRequest
  390. var err error
  391. // Create a test pullrequest
  392. t.Run("CreatePullRequest", func(t *testing.T) {
  393. pr, err = doAPICreatePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, baseBranch, headBranch)(t)
  394. assert.NoError(t, err)
  395. })
  396. // Ensure the PR page works
  397. t.Run("EnsureCanSeePull", doEnsureCanSeePull(baseCtx, pr))
  398. // Then get the diff string
  399. var diffHash string
  400. var diffLength int
  401. t.Run("GetDiff", func(t *testing.T) {
  402. req := NewRequest(t, "GET", fmt.Sprintf("/%s/%s/pulls/%d.diff", url.PathEscape(baseCtx.Username), url.PathEscape(baseCtx.Reponame), pr.Index))
  403. resp := ctx.Session.MakeRequestNilResponseHashSumRecorder(t, req, http.StatusOK)
  404. diffHash = string(resp.Hash.Sum(nil))
  405. diffLength = resp.Length
  406. })
  407. // Now: Merge the PR & make sure that doesn't break the PR page or change its diff
  408. t.Run("MergePR", doAPIMergePullRequest(baseCtx, baseCtx.Username, baseCtx.Reponame, pr.Index))
  409. t.Run("EnsureCanSeePull", doEnsureCanSeePull(baseCtx, pr))
  410. t.Run("CheckPR", func(t *testing.T) {
  411. oldMergeBase := pr.MergeBase
  412. pr2, err := doAPIGetPullRequest(baseCtx, baseCtx.Username, baseCtx.Reponame, pr.Index)(t)
  413. assert.NoError(t, err)
  414. assert.Equal(t, oldMergeBase, pr2.MergeBase)
  415. })
  416. t.Run("EnsurDiffNoChange", doEnsureDiffNoChange(baseCtx, pr, diffHash, diffLength))
  417. // Then: Delete the head branch & make sure that doesn't break the PR page or change its diff
  418. t.Run("DeleteHeadBranch", doBranchDelete(baseCtx, baseCtx.Username, baseCtx.Reponame, headBranch))
  419. t.Run("EnsureCanSeePull", doEnsureCanSeePull(baseCtx, pr))
  420. t.Run("EnsureDiffNoChange", doEnsureDiffNoChange(baseCtx, pr, diffHash, diffLength))
  421. // Delete the head repository & make sure that doesn't break the PR page or change its diff
  422. t.Run("DeleteHeadRepository", doAPIDeleteRepository(ctx))
  423. t.Run("EnsureCanSeePull", doEnsureCanSeePull(baseCtx, pr))
  424. t.Run("EnsureDiffNoChange", doEnsureDiffNoChange(baseCtx, pr, diffHash, diffLength))
  425. }
  426. }
  427. func doCreatePRAndSetManuallyMerged(ctx, baseCtx APITestContext, dstPath, baseBranch, headBranch string) func(t *testing.T) {
  428. return func(t *testing.T) {
  429. defer tests.PrintCurrentTest(t)()
  430. var (
  431. pr api.PullRequest
  432. err error
  433. lastCommitID string
  434. )
  435. trueBool := true
  436. falseBool := false
  437. t.Run("AllowSetManuallyMergedAndSwitchOffAutodetectManualMerge", doAPIEditRepository(baseCtx, &api.EditRepoOption{
  438. HasPullRequests: &trueBool,
  439. AllowManualMerge: &trueBool,
  440. AutodetectManualMerge: &falseBool,
  441. }))
  442. t.Run("CreateHeadBranch", doGitCreateBranch(dstPath, headBranch))
  443. t.Run("PushToHeadBranch", doGitPushTestRepository(dstPath, "origin", headBranch))
  444. t.Run("CreateEmptyPullRequest", func(t *testing.T) {
  445. pr, err = doAPICreatePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, baseBranch, headBranch)(t)
  446. assert.NoError(t, err)
  447. })
  448. lastCommitID = pr.Base.Sha
  449. t.Run("ManuallyMergePR", doAPIManuallyMergePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, lastCommitID, pr.Index))
  450. }
  451. }
  452. func doEnsureCanSeePull(ctx APITestContext, pr api.PullRequest) func(t *testing.T) {
  453. return func(t *testing.T) {
  454. req := NewRequest(t, "GET", fmt.Sprintf("/%s/%s/pulls/%d", url.PathEscape(ctx.Username), url.PathEscape(ctx.Reponame), pr.Index))
  455. ctx.Session.MakeRequest(t, req, http.StatusOK)
  456. req = NewRequest(t, "GET", fmt.Sprintf("/%s/%s/pulls/%d/files", url.PathEscape(ctx.Username), url.PathEscape(ctx.Reponame), pr.Index))
  457. ctx.Session.MakeRequest(t, req, http.StatusOK)
  458. req = NewRequest(t, "GET", fmt.Sprintf("/%s/%s/pulls/%d/commits", url.PathEscape(ctx.Username), url.PathEscape(ctx.Reponame), pr.Index))
  459. ctx.Session.MakeRequest(t, req, http.StatusOK)
  460. }
  461. }
  462. func doEnsureDiffNoChange(ctx APITestContext, pr api.PullRequest, diffHash string, diffLength int) func(t *testing.T) {
  463. return func(t *testing.T) {
  464. req := NewRequest(t, "GET", fmt.Sprintf("/%s/%s/pulls/%d.diff", url.PathEscape(ctx.Username), url.PathEscape(ctx.Reponame), pr.Index))
  465. resp := ctx.Session.MakeRequestNilResponseHashSumRecorder(t, req, http.StatusOK)
  466. actual := string(resp.Hash.Sum(nil))
  467. actualLength := resp.Length
  468. equal := diffHash == actual
  469. assert.True(t, equal, "Unexpected change in the diff string: expected hash: %s size: %d but was actually: %s size: %d", hex.EncodeToString([]byte(diffHash)), diffLength, hex.EncodeToString([]byte(actual)), actualLength)
  470. }
  471. }
  472. func doPushCreate(ctx APITestContext, u *url.URL) func(t *testing.T) {
  473. return func(t *testing.T) {
  474. defer tests.PrintCurrentTest(t)()
  475. // create a context for a currently non-existent repository
  476. ctx.Reponame = fmt.Sprintf("repo-tmp-push-create-%s", u.Scheme)
  477. u.Path = ctx.GitPath()
  478. // Create a temporary directory
  479. tmpDir := t.TempDir()
  480. // Now create local repository to push as our test and set its origin
  481. t.Run("InitTestRepository", doGitInitTestRepository(tmpDir))
  482. t.Run("AddRemote", doGitAddRemote(tmpDir, "origin", u))
  483. // Disable "Push To Create" and attempt to push
  484. setting.Repository.EnablePushCreateUser = false
  485. t.Run("FailToPushAndCreateTestRepository", doGitPushTestRepositoryFail(tmpDir, "origin", "master"))
  486. // Enable "Push To Create"
  487. setting.Repository.EnablePushCreateUser = true
  488. // Assert that cloning from a non-existent repository does not create it and that it definitely wasn't create above
  489. t.Run("FailToCloneFromNonExistentRepository", doGitCloneFail(u))
  490. // Then "Push To Create"x
  491. t.Run("SuccessfullyPushAndCreateTestRepository", doGitPushTestRepository(tmpDir, "origin", "master"))
  492. // Finally, fetch repo from database and ensure the correct repository has been created
  493. repo, err := repo_model.GetRepositoryByOwnerAndName(db.DefaultContext, ctx.Username, ctx.Reponame)
  494. assert.NoError(t, err)
  495. assert.False(t, repo.IsEmpty)
  496. assert.True(t, repo.IsPrivate)
  497. // Now add a remote that is invalid to "Push To Create"
  498. invalidCtx := ctx
  499. invalidCtx.Reponame = fmt.Sprintf("invalid/repo-tmp-push-create-%s", u.Scheme)
  500. u.Path = invalidCtx.GitPath()
  501. t.Run("AddInvalidRemote", doGitAddRemote(tmpDir, "invalid", u))
  502. // Fail to "Push To Create" the invalid
  503. t.Run("FailToPushAndCreateInvalidTestRepository", doGitPushTestRepositoryFail(tmpDir, "invalid", "master"))
  504. }
  505. }
  506. func doBranchDelete(ctx APITestContext, owner, repo, branch string) func(*testing.T) {
  507. return func(t *testing.T) {
  508. csrf := GetCSRF(t, ctx.Session, fmt.Sprintf("/%s/%s/branches", url.PathEscape(owner), url.PathEscape(repo)))
  509. req := NewRequestWithValues(t, "POST", fmt.Sprintf("/%s/%s/branches/delete?name=%s", url.PathEscape(owner), url.PathEscape(repo), url.QueryEscape(branch)), map[string]string{
  510. "_csrf": csrf,
  511. })
  512. ctx.Session.MakeRequest(t, req, http.StatusOK)
  513. }
  514. }
  515. func doAutoPRMerge(baseCtx *APITestContext, dstPath string) func(t *testing.T) {
  516. return func(t *testing.T) {
  517. defer tests.PrintCurrentTest(t)()
  518. ctx := NewAPITestContext(t, baseCtx.Username, baseCtx.Reponame, auth_model.AccessTokenScopeWriteRepository)
  519. t.Run("CheckoutProtected", doGitCheckoutBranch(dstPath, "protected"))
  520. t.Run("PullProtected", doGitPull(dstPath, "origin", "protected"))
  521. t.Run("GenerateCommit", func(t *testing.T) {
  522. _, err := generateCommitWithNewData(littleSize, dstPath, "user2@example.com", "User Two", "branch-data-file-")
  523. assert.NoError(t, err)
  524. })
  525. t.Run("PushToUnprotectedBranch", doGitPushTestRepository(dstPath, "origin", "protected:unprotected3"))
  526. var pr api.PullRequest
  527. var err error
  528. t.Run("CreatePullRequest", func(t *testing.T) {
  529. pr, err = doAPICreatePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, "protected", "unprotected3")(t)
  530. assert.NoError(t, err)
  531. })
  532. // Request repository commits page
  533. req := NewRequest(t, "GET", fmt.Sprintf("/%s/%s/pulls/%d/commits", baseCtx.Username, baseCtx.Reponame, pr.Index))
  534. resp := ctx.Session.MakeRequest(t, req, http.StatusOK)
  535. doc := NewHTMLParser(t, resp.Body)
  536. // Get first commit URL
  537. commitURL, exists := doc.doc.Find("#commits-table tbody tr td.sha a").Last().Attr("href")
  538. assert.True(t, exists)
  539. assert.NotEmpty(t, commitURL)
  540. commitID := path.Base(commitURL)
  541. addCommitStatus := func(status api.CommitStatusState) func(*testing.T) {
  542. return doAPICreateCommitStatus(ctx, commitID, api.CreateStatusOption{
  543. State: status,
  544. TargetURL: "http://test.ci/",
  545. Description: "",
  546. Context: "testci",
  547. })
  548. }
  549. // Call API to add Pending status for commit
  550. t.Run("CreateStatus", addCommitStatus(api.CommitStatusPending))
  551. // Cancel not existing auto merge
  552. ctx.ExpectedCode = http.StatusNotFound
  553. t.Run("CancelAutoMergePR", doAPICancelAutoMergePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index))
  554. // Add auto merge request
  555. ctx.ExpectedCode = http.StatusCreated
  556. t.Run("AutoMergePR", doAPIAutoMergePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index))
  557. // Can not create schedule twice
  558. ctx.ExpectedCode = http.StatusConflict
  559. t.Run("AutoMergePRTwice", doAPIAutoMergePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index))
  560. // Cancel auto merge request
  561. ctx.ExpectedCode = http.StatusNoContent
  562. t.Run("CancelAutoMergePR", doAPICancelAutoMergePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index))
  563. // Add auto merge request
  564. ctx.ExpectedCode = http.StatusCreated
  565. t.Run("AutoMergePR", doAPIAutoMergePullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index))
  566. // Check pr status
  567. ctx.ExpectedCode = 0
  568. pr, err = doAPIGetPullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index)(t)
  569. assert.NoError(t, err)
  570. assert.False(t, pr.HasMerged)
  571. // Call API to add Failure status for commit
  572. t.Run("CreateStatus", addCommitStatus(api.CommitStatusFailure))
  573. // Check pr status
  574. pr, err = doAPIGetPullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index)(t)
  575. assert.NoError(t, err)
  576. assert.False(t, pr.HasMerged)
  577. // Call API to add Success status for commit
  578. t.Run("CreateStatus", addCommitStatus(api.CommitStatusSuccess))
  579. // wait to let gitea merge stuff
  580. time.Sleep(time.Second)
  581. // test pr status
  582. pr, err = doAPIGetPullRequest(ctx, baseCtx.Username, baseCtx.Reponame, pr.Index)(t)
  583. assert.NoError(t, err)
  584. assert.True(t, pr.HasMerged)
  585. }
  586. }
  587. func doCreateAgitFlowPull(dstPath string, ctx *APITestContext, baseBranch, headBranch string) func(t *testing.T) {
  588. return func(t *testing.T) {
  589. defer tests.PrintCurrentTest(t)()
  590. // skip this test if git version is low
  591. if git.CheckGitVersionAtLeast("2.29") != nil {
  592. return
  593. }
  594. gitRepo, err := git.OpenRepository(git.DefaultContext, dstPath)
  595. if !assert.NoError(t, err) {
  596. return
  597. }
  598. defer gitRepo.Close()
  599. var (
  600. pr1, pr2 *issues_model.PullRequest
  601. commit string
  602. )
  603. repo, err := repo_model.GetRepositoryByOwnerAndName(db.DefaultContext, ctx.Username, ctx.Reponame)
  604. if !assert.NoError(t, err) {
  605. return
  606. }
  607. pullNum := unittest.GetCount(t, &issues_model.PullRequest{})
  608. t.Run("CreateHeadBranch", doGitCreateBranch(dstPath, headBranch))
  609. t.Run("AddCommit", func(t *testing.T) {
  610. err := os.WriteFile(path.Join(dstPath, "test_file"), []byte("## test content"), 0o666)
  611. if !assert.NoError(t, err) {
  612. return
  613. }
  614. err = git.AddChanges(dstPath, true)
  615. assert.NoError(t, err)
  616. err = git.CommitChanges(dstPath, git.CommitChangesOptions{
  617. Committer: &git.Signature{
  618. Email: "user2@example.com",
  619. Name: "user2",
  620. When: time.Now(),
  621. },
  622. Author: &git.Signature{
  623. Email: "user2@example.com",
  624. Name: "user2",
  625. When: time.Now(),
  626. },
  627. Message: "Testing commit 1",
  628. })
  629. assert.NoError(t, err)
  630. commit, err = gitRepo.GetRefCommitID("HEAD")
  631. assert.NoError(t, err)
  632. })
  633. t.Run("Push", func(t *testing.T) {
  634. err := git.NewCommand(git.DefaultContext, "push", "origin", "HEAD:refs/for/master", "-o").AddDynamicArguments("topic=" + headBranch).Run(&git.RunOpts{Dir: dstPath})
  635. if !assert.NoError(t, err) {
  636. return
  637. }
  638. unittest.AssertCount(t, &issues_model.PullRequest{}, pullNum+1)
  639. pr1 = unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{
  640. HeadRepoID: repo.ID,
  641. Flow: issues_model.PullRequestFlowAGit,
  642. })
  643. if !assert.NotEmpty(t, pr1) {
  644. return
  645. }
  646. prMsg, err := doAPIGetPullRequest(*ctx, ctx.Username, ctx.Reponame, pr1.Index)(t)
  647. if !assert.NoError(t, err) {
  648. return
  649. }
  650. assert.Equal(t, "user2/"+headBranch, pr1.HeadBranch)
  651. assert.False(t, prMsg.HasMerged)
  652. assert.Contains(t, "Testing commit 1", prMsg.Body)
  653. assert.Equal(t, commit, prMsg.Head.Sha)
  654. _, _, err = git.NewCommand(git.DefaultContext, "push", "origin").AddDynamicArguments("HEAD:refs/for/master/test/" + headBranch).RunStdString(&git.RunOpts{Dir: dstPath})
  655. if !assert.NoError(t, err) {
  656. return
  657. }
  658. unittest.AssertCount(t, &issues_model.PullRequest{}, pullNum+2)
  659. pr2 = unittest.AssertExistsAndLoadBean(t, &issues_model.PullRequest{
  660. HeadRepoID: repo.ID,
  661. Index: pr1.Index + 1,
  662. Flow: issues_model.PullRequestFlowAGit,
  663. })
  664. if !assert.NotEmpty(t, pr2) {
  665. return
  666. }
  667. prMsg, err = doAPIGetPullRequest(*ctx, ctx.Username, ctx.Reponame, pr2.Index)(t)
  668. if !assert.NoError(t, err) {
  669. return
  670. }
  671. assert.Equal(t, "user2/test/"+headBranch, pr2.HeadBranch)
  672. assert.False(t, prMsg.HasMerged)
  673. })
  674. if pr1 == nil || pr2 == nil {
  675. return
  676. }
  677. t.Run("AddCommit2", func(t *testing.T) {
  678. err := os.WriteFile(path.Join(dstPath, "test_file"), []byte("## test content \n ## test content 2"), 0o666)
  679. if !assert.NoError(t, err) {
  680. return
  681. }
  682. err = git.AddChanges(dstPath, true)
  683. assert.NoError(t, err)
  684. err = git.CommitChanges(dstPath, git.CommitChangesOptions{
  685. Committer: &git.Signature{
  686. Email: "user2@example.com",
  687. Name: "user2",
  688. When: time.Now(),
  689. },
  690. Author: &git.Signature{
  691. Email: "user2@example.com",
  692. Name: "user2",
  693. When: time.Now(),
  694. },
  695. Message: "Testing commit 2",
  696. })
  697. assert.NoError(t, err)
  698. commit, err = gitRepo.GetRefCommitID("HEAD")
  699. assert.NoError(t, err)
  700. })
  701. t.Run("Push2", func(t *testing.T) {
  702. err := git.NewCommand(git.DefaultContext, "push", "origin", "HEAD:refs/for/master", "-o").AddDynamicArguments("topic=" + headBranch).Run(&git.RunOpts{Dir: dstPath})
  703. if !assert.NoError(t, err) {
  704. return
  705. }
  706. unittest.AssertCount(t, &issues_model.PullRequest{}, pullNum+2)
  707. prMsg, err := doAPIGetPullRequest(*ctx, ctx.Username, ctx.Reponame, pr1.Index)(t)
  708. if !assert.NoError(t, err) {
  709. return
  710. }
  711. assert.False(t, prMsg.HasMerged)
  712. assert.Equal(t, commit, prMsg.Head.Sha)
  713. _, _, err = git.NewCommand(git.DefaultContext, "push", "origin").AddDynamicArguments("HEAD:refs/for/master/test/" + headBranch).RunStdString(&git.RunOpts{Dir: dstPath})
  714. if !assert.NoError(t, err) {
  715. return
  716. }
  717. unittest.AssertCount(t, &issues_model.PullRequest{}, pullNum+2)
  718. prMsg, err = doAPIGetPullRequest(*ctx, ctx.Username, ctx.Reponame, pr2.Index)(t)
  719. if !assert.NoError(t, err) {
  720. return
  721. }
  722. assert.False(t, prMsg.HasMerged)
  723. assert.Equal(t, commit, prMsg.Head.Sha)
  724. })
  725. t.Run("Merge", doAPIMergePullRequest(*ctx, ctx.Username, ctx.Reponame, pr1.Index))
  726. t.Run("CheckoutMasterAgain", doGitCheckoutBranch(dstPath, "master"))
  727. }
  728. }
  729. func TestDataAsync_Issue29101(t *testing.T) {
  730. onGiteaRun(t, func(t *testing.T, u *url.URL) {
  731. user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
  732. repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
  733. resp, err := files_service.ChangeRepoFiles(db.DefaultContext, repo, user, &files_service.ChangeRepoFilesOptions{
  734. Files: []*files_service.ChangeRepoFile{
  735. {
  736. Operation: "create",
  737. TreePath: "test.txt",
  738. ContentReader: bytes.NewReader(make([]byte, 10000)),
  739. },
  740. },
  741. OldBranch: repo.DefaultBranch,
  742. NewBranch: repo.DefaultBranch,
  743. })
  744. assert.NoError(t, err)
  745. sha := resp.Commit.SHA
  746. gitRepo, err := git.OpenRepository(db.DefaultContext, repo.RepoPath())
  747. assert.NoError(t, err)
  748. commit, err := gitRepo.GetCommit(sha)
  749. assert.NoError(t, err)
  750. entry, err := commit.GetTreeEntryByPath("test.txt")
  751. assert.NoError(t, err)
  752. b := entry.Blob()
  753. r, err := b.DataAsync()
  754. assert.NoError(t, err)
  755. defer r.Close()
  756. r2, err := b.DataAsync()
  757. assert.NoError(t, err)
  758. defer r2.Close()
  759. })
  760. }