diff options
Diffstat (limited to 'modules/git/blob_test.go')
-rw-r--r-- | modules/git/blob_test.go | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/modules/git/blob_test.go b/modules/git/blob_test.go new file mode 100644 index 00000000..810964b3 --- /dev/null +++ b/modules/git/blob_test.go @@ -0,0 +1,59 @@ +// Copyright 2015 The Gogs Authors. All rights reserved. +// Copyright 2019 The Gitea Authors. All rights reserved. +// SPDX-License-Identifier: MIT + +package git + +import ( + "io" + "path/filepath" + "testing" + + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" +) + +func TestBlob_Data(t *testing.T) { + output := "file2\n" + bareRepo1Path := filepath.Join(testReposDir, "repo1_bare") + repo, err := openRepositoryWithDefaultContext(bareRepo1Path) + require.NoError(t, err) + + defer repo.Close() + + testBlob, err := repo.GetBlob("6c493ff740f9380390d5c9ddef4af18697ac9375") + require.NoError(t, err) + + r, err := testBlob.DataAsync() + require.NoError(t, err) + require.NotNil(t, r) + + data, err := io.ReadAll(r) + require.NoError(t, r.Close()) + + require.NoError(t, err) + assert.Equal(t, output, string(data)) +} + +func Benchmark_Blob_Data(b *testing.B) { + bareRepo1Path := filepath.Join(testReposDir, "repo1_bare") + repo, err := openRepositoryWithDefaultContext(bareRepo1Path) + if err != nil { + b.Fatal(err) + } + defer repo.Close() + + testBlob, err := repo.GetBlob("6c493ff740f9380390d5c9ddef4af18697ac9375") + if err != nil { + b.Fatal(err) + } + + for i := 0; i < b.N; i++ { + r, err := testBlob.DataAsync() + if err != nil { + b.Fatal(err) + } + io.ReadAll(r) + _ = r.Close() + } +} |