mirror of
				https://github.com/go-gitea/gitea
				synced 2025-10-31 19:38:23 +00:00 
			
		
		
		
	A testing cleanup. This pull request replaces `os.MkdirTemp` with `t.TempDir`. We can use the `T.TempDir` function from the `testing` package to create temporary directory. The directory created by `T.TempDir` is automatically removed when the test and all its subtests complete. This saves us at least 2 lines (error check, and cleanup) on every instance, or in some cases adds cleanup that we forgot. Reference: https://pkg.go.dev/testing#T.TempDir ```go func TestFoo(t *testing.T) { // before tmpDir, err := os.MkdirTemp("", "") require.NoError(t, err) defer os.RemoveAll(tmpDir) // now tmpDir := t.TempDir() } ``` Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
		
			
				
	
	
		
			123 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			123 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2018 The Gitea Authors. All rights reserved.
 | |
| // Use of this source code is governed by a MIT-style
 | |
| // license that can be found in the LICENSE file.
 | |
| 
 | |
| package git
 | |
| 
 | |
| import (
 | |
| 	"bytes"
 | |
| 	"io"
 | |
| 	"path/filepath"
 | |
| 	"testing"
 | |
| 
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| )
 | |
| 
 | |
| func TestGetFormatPatch(t *testing.T) {
 | |
| 	bareRepo1Path := filepath.Join(testReposDir, "repo1_bare")
 | |
| 	clonedPath, err := cloneRepo(t, bareRepo1Path)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	repo, err := openRepositoryWithDefaultContext(clonedPath)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 	defer repo.Close()
 | |
| 
 | |
| 	rd := &bytes.Buffer{}
 | |
| 	err = repo.GetPatch("8d92fc95^", "8d92fc95", rd)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	patchb, err := io.ReadAll(rd)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	patch := string(patchb)
 | |
| 	assert.Regexp(t, "^From 8d92fc95", patch)
 | |
| 	assert.Contains(t, patch, "Subject: [PATCH] Add file2.txt")
 | |
| }
 | |
| 
 | |
| func TestReadPatch(t *testing.T) {
 | |
| 	// Ensure we can read the patch files
 | |
| 	bareRepo1Path := filepath.Join(testReposDir, "repo1_bare")
 | |
| 	repo, err := openRepositoryWithDefaultContext(bareRepo1Path)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 	defer repo.Close()
 | |
| 	// This patch doesn't exist
 | |
| 	noFile, err := repo.ReadPatchCommit(0)
 | |
| 	assert.Error(t, err)
 | |
| 
 | |
| 	// This patch is an empty one (sometimes it's a 404)
 | |
| 	noCommit, err := repo.ReadPatchCommit(1)
 | |
| 	assert.Error(t, err)
 | |
| 
 | |
| 	// This patch is legit and should return a commit
 | |
| 	oldCommit, err := repo.ReadPatchCommit(2)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	assert.Empty(t, noFile)
 | |
| 	assert.Empty(t, noCommit)
 | |
| 	assert.Len(t, oldCommit, 40)
 | |
| 	assert.True(t, oldCommit == "6e8e2a6f9efd71dbe6917816343ed8415ad696c3")
 | |
| }
 | |
| 
 | |
| func TestReadWritePullHead(t *testing.T) {
 | |
| 	// Ensure we can write SHA1 head corresponding to PR and open them
 | |
| 	bareRepo1Path := filepath.Join(testReposDir, "repo1_bare")
 | |
| 
 | |
| 	// As we are writing we should clone the repository first
 | |
| 	clonedPath, err := cloneRepo(t, bareRepo1Path)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	repo, err := openRepositoryWithDefaultContext(clonedPath)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 	defer repo.Close()
 | |
| 
 | |
| 	// Try to open non-existing Pull
 | |
| 	_, err = repo.GetRefCommitID(PullPrefix + "0/head")
 | |
| 	assert.Error(t, err)
 | |
| 
 | |
| 	// Write a fake sha1 with only 40 zeros
 | |
| 	newCommit := "feaf4ba6bc635fec442f46ddd4512416ec43c2c2"
 | |
| 	err = repo.SetReference(PullPrefix+"1/head", newCommit)
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	// Read the file created
 | |
| 	headContents, err := repo.GetRefCommitID(PullPrefix + "1/head")
 | |
| 	if err != nil {
 | |
| 		assert.NoError(t, err)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	assert.Len(t, headContents, 40)
 | |
| 	assert.True(t, headContents == newCommit)
 | |
| 
 | |
| 	// Remove file after the test
 | |
| 	err = repo.RemoveReference(PullPrefix + "1/head")
 | |
| 	assert.NoError(t, err)
 | |
| }
 |