forked from Shiloh/githaven
722a7c902d
In investigating #7947 it has become clear that the storage component of go-git repositories needs closing. This PR adds this Close function and adds the Close functions as necessary. In TransferOwnership the ctx.Repo.GitRepo is closed if it is open to help prevent the risk of multiple open files. Fixes #7947
71 lines
1.7 KiB
Go
71 lines
1.7 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 (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRepository_GetBlob_Found(t *testing.T) {
|
|
repoPath := filepath.Join(testReposDir, "repo1_bare")
|
|
r, err := OpenRepository(repoPath)
|
|
assert.NoError(t, err)
|
|
defer r.Close()
|
|
|
|
testCases := []struct {
|
|
OID string
|
|
Data []byte
|
|
}{
|
|
{"e2129701f1a4d54dc44f03c93bca0a2aec7c5449", []byte("file1\n")},
|
|
{"6c493ff740f9380390d5c9ddef4af18697ac9375", []byte("file2\n")},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
blob, err := r.GetBlob(testCase.OID)
|
|
assert.NoError(t, err)
|
|
|
|
dataReader, err := blob.DataAsync()
|
|
assert.NoError(t, err)
|
|
defer dataReader.Close()
|
|
|
|
data, err := ioutil.ReadAll(dataReader)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, testCase.Data, data)
|
|
}
|
|
}
|
|
|
|
func TestRepository_GetBlob_NotExist(t *testing.T) {
|
|
repoPath := filepath.Join(testReposDir, "repo1_bare")
|
|
r, err := OpenRepository(repoPath)
|
|
assert.NoError(t, err)
|
|
defer r.Close()
|
|
|
|
testCase := "0000000000000000000000000000000000000000"
|
|
testError := ErrNotExist{testCase, ""}
|
|
|
|
blob, err := r.GetBlob(testCase)
|
|
assert.Nil(t, blob)
|
|
assert.EqualError(t, err, testError.Error())
|
|
}
|
|
|
|
func TestRepository_GetBlob_NoId(t *testing.T) {
|
|
repoPath := filepath.Join(testReposDir, "repo1_bare")
|
|
r, err := OpenRepository(repoPath)
|
|
assert.NoError(t, err)
|
|
defer r.Close()
|
|
|
|
testCase := ""
|
|
testError := fmt.Errorf("Length must be 40: %s", testCase)
|
|
|
|
blob, err := r.GetBlob(testCase)
|
|
assert.Nil(t, blob)
|
|
assert.EqualError(t, err, testError.Error())
|
|
}
|