mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-10-31 02:46:04 +01:00 
			
		
		
		
	Fix #33914 before:  after:  --------- Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
		
			
				
	
	
		
			115 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			115 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2019 The Gitea Authors. All rights reserved.
 | |
| // SPDX-License-Identifier: MIT
 | |
| 
 | |
| package files
 | |
| 
 | |
| import (
 | |
| 	"html/template"
 | |
| 	"testing"
 | |
| 
 | |
| 	"code.gitea.io/gitea/models/unittest"
 | |
| 	"code.gitea.io/gitea/modules/fileicon"
 | |
| 	"code.gitea.io/gitea/modules/git"
 | |
| 	api "code.gitea.io/gitea/modules/structs"
 | |
| 	"code.gitea.io/gitea/services/contexttest"
 | |
| 
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| )
 | |
| 
 | |
| func TestGetTreeBySHA(t *testing.T) {
 | |
| 	unittest.PrepareTestEnv(t)
 | |
| 	ctx, _ := contexttest.MockContext(t, "user2/repo1")
 | |
| 	contexttest.LoadRepo(t, ctx, 1)
 | |
| 	contexttest.LoadRepoCommit(t, ctx)
 | |
| 	contexttest.LoadUser(t, ctx, 2)
 | |
| 	contexttest.LoadGitRepo(t, ctx)
 | |
| 	defer ctx.Repo.GitRepo.Close()
 | |
| 
 | |
| 	sha := ctx.Repo.Repository.DefaultBranch
 | |
| 	page := 1
 | |
| 	perPage := 10
 | |
| 	ctx.SetPathParam("id", "1")
 | |
| 	ctx.SetPathParam("sha", sha)
 | |
| 
 | |
| 	tree, err := GetTreeBySHA(ctx, ctx.Repo.Repository, ctx.Repo.GitRepo, ctx.PathParam("sha"), page, perPage, true)
 | |
| 	assert.NoError(t, err)
 | |
| 	expectedTree := &api.GitTreeResponse{
 | |
| 		SHA: "65f1bf27bc3bf70f64657658635e66094edbcb4d",
 | |
| 		URL: "https://try.gitea.io/api/v1/repos/user2/repo1/git/trees/65f1bf27bc3bf70f64657658635e66094edbcb4d",
 | |
| 		Entries: []api.GitEntry{
 | |
| 			{
 | |
| 				Path: "README.md",
 | |
| 				Mode: "100644",
 | |
| 				Type: "blob",
 | |
| 				Size: 30,
 | |
| 				SHA:  "4b4851ad51df6a7d9f25c979345979eaeb5b349f",
 | |
| 				URL:  "https://try.gitea.io/api/v1/repos/user2/repo1/git/blobs/4b4851ad51df6a7d9f25c979345979eaeb5b349f",
 | |
| 			},
 | |
| 		},
 | |
| 		Truncated:  false,
 | |
| 		Page:       1,
 | |
| 		TotalCount: 1,
 | |
| 	}
 | |
| 
 | |
| 	assert.Equal(t, expectedTree, tree)
 | |
| }
 | |
| 
 | |
| func TestGetTreeViewNodes(t *testing.T) {
 | |
| 	unittest.PrepareTestEnv(t)
 | |
| 	ctx, _ := contexttest.MockContext(t, "user2/repo1")
 | |
| 	ctx.Repo.RefFullName = git.RefNameFromBranch("sub-home-md-img-check")
 | |
| 	contexttest.LoadRepo(t, ctx, 1)
 | |
| 	contexttest.LoadRepoCommit(t, ctx)
 | |
| 	contexttest.LoadUser(t, ctx, 2)
 | |
| 	contexttest.LoadGitRepo(t, ctx)
 | |
| 	defer ctx.Repo.GitRepo.Close()
 | |
| 
 | |
| 	renderedIconPool := fileicon.NewRenderedIconPool()
 | |
| 	mockIconForFile := func(id string) template.HTML {
 | |
| 		return template.HTML(`<svg class="svg git-entry-icon octicon-file" width="16" height="16" aria-hidden="true"><use xlink:href="#` + id + `"></use></svg>`)
 | |
| 	}
 | |
| 	mockIconForFolder := func(id string) template.HTML {
 | |
| 		return template.HTML(`<svg class="svg git-entry-icon octicon-file-directory-fill" width="16" height="16" aria-hidden="true"><use xlink:href="#` + id + `"></use></svg>`)
 | |
| 	}
 | |
| 	treeNodes, err := GetTreeViewNodes(ctx, renderedIconPool, ctx.Repo.Commit, "", "")
 | |
| 	assert.NoError(t, err)
 | |
| 	assert.Equal(t, []*TreeViewNode{
 | |
| 		{
 | |
| 			EntryName: "docs",
 | |
| 			EntryMode: "tree",
 | |
| 			FullPath:  "docs",
 | |
| 			EntryIcon: mockIconForFolder(`svg-mfi-folder-docs`),
 | |
| 		},
 | |
| 	}, treeNodes)
 | |
| 
 | |
| 	treeNodes, err = GetTreeViewNodes(ctx, renderedIconPool, ctx.Repo.Commit, "", "docs/README.md")
 | |
| 	assert.NoError(t, err)
 | |
| 	assert.Equal(t, []*TreeViewNode{
 | |
| 		{
 | |
| 			EntryName: "docs",
 | |
| 			EntryMode: "tree",
 | |
| 			FullPath:  "docs",
 | |
| 			EntryIcon: mockIconForFolder(`svg-mfi-folder-docs`),
 | |
| 			Children: []*TreeViewNode{
 | |
| 				{
 | |
| 					EntryName: "README.md",
 | |
| 					EntryMode: "blob",
 | |
| 					FullPath:  "docs/README.md",
 | |
| 					EntryIcon: mockIconForFile(`svg-mfi-readme`),
 | |
| 				},
 | |
| 			},
 | |
| 		},
 | |
| 	}, treeNodes)
 | |
| 
 | |
| 	treeNodes, err = GetTreeViewNodes(ctx, renderedIconPool, ctx.Repo.Commit, "docs", "README.md")
 | |
| 	assert.NoError(t, err)
 | |
| 	assert.Equal(t, []*TreeViewNode{
 | |
| 		{
 | |
| 			EntryName: "README.md",
 | |
| 			EntryMode: "blob",
 | |
| 			FullPath:  "docs/README.md",
 | |
| 			EntryIcon: mockIconForFile(`svg-mfi-readme`),
 | |
| 		},
 | |
| 	}, treeNodes)
 | |
| }
 |