mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
9a07b4d089
2978 files. The fork is 2026-07-15 (rebrand: gitea.dev -> github.com/hanzoai/git), so 2026 is the year, and "Copyright 2026 Hanzo AI, Inc." was already the form used by the handful of files that had ours. Shape is now: // Copyright 2026 Hanzo AI, Inc. All rights reserved. // Copyright 2015 The Gogs Authors. All rights reserved. // Copyright 2019 The Gitea Authors. All rights reserved. // SPDX-License-Identifier: MIT Ours on top, theirs retained verbatim. Retained because MIT requires it, and separately because they are true — those authors did write that code. Not renamed to "The Git Authors": Git is a different project under a different license, and putting its name here would attribute this code to people who have nothing to do with it. Adding our line is the fix; overwriting theirs would have been both a licence violation and a false statement. gofmt still reports import-order drift in some of these files. It predates this change (verified against a stash) and is left alone rather than burying a one-line-per-file legal change under 3000 files of unrelated reordering. Build clean. Co-authored-by: Hanzo Dev <[email protected]>
232 lines
6.0 KiB
Go
232 lines
6.0 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2022 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package dbfs
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/git/modules/test"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDbfsBasic(t *testing.T) {
|
|
defer test.MockVariableValue(&defaultFileBlockSize, 4)()
|
|
|
|
// test basic write/read
|
|
f, err := OpenFile(t.Context(), "test.txt", os.O_RDWR|os.O_CREATE)
|
|
assert.NoError(t, err)
|
|
|
|
n, err := f.Write([]byte("0123456789")) // blocks: 0123 4567 89
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 10, n)
|
|
|
|
_, err = f.Seek(0, io.SeekStart)
|
|
assert.NoError(t, err)
|
|
|
|
buf, err := io.ReadAll(f)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 10, n)
|
|
assert.Equal(t, "0123456789", string(buf))
|
|
|
|
// write some new data
|
|
_, err = f.Seek(1, io.SeekStart)
|
|
assert.NoError(t, err)
|
|
_, err = f.Write([]byte("bcdefghi")) // blocks: 0bcd efgh i9
|
|
assert.NoError(t, err)
|
|
|
|
// read from offset
|
|
buf, err = io.ReadAll(f)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "9", string(buf))
|
|
|
|
// read all
|
|
_, err = f.Seek(0, io.SeekStart)
|
|
assert.NoError(t, err)
|
|
buf, err = io.ReadAll(f)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "0bcdefghi9", string(buf))
|
|
|
|
// write to new size
|
|
_, err = f.Seek(-1, io.SeekEnd)
|
|
assert.NoError(t, err)
|
|
_, err = f.Write([]byte("JKLMNOP")) // blocks: 0bcd efgh iJKL MNOP
|
|
assert.NoError(t, err)
|
|
_, err = f.Seek(0, io.SeekStart)
|
|
assert.NoError(t, err)
|
|
buf, err = io.ReadAll(f)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "0bcdefghiJKLMNOP", string(buf))
|
|
|
|
// write beyond EOF and fill with zero
|
|
_, err = f.Seek(5, io.SeekCurrent)
|
|
assert.NoError(t, err)
|
|
_, err = f.Write([]byte("xyzu")) // blocks: 0bcd efgh iJKL MNOP 0000 0xyz u
|
|
assert.NoError(t, err)
|
|
_, err = f.Seek(0, io.SeekStart)
|
|
assert.NoError(t, err)
|
|
buf, err = io.ReadAll(f)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "0bcdefghiJKLMNOP\x00\x00\x00\x00\x00xyzu", string(buf))
|
|
|
|
// write to the block with zeros
|
|
_, err = f.Seek(-6, io.SeekCurrent)
|
|
assert.NoError(t, err)
|
|
_, err = f.Write([]byte("ABCD")) // blocks: 0bcd efgh iJKL MNOP 000A BCDz u
|
|
assert.NoError(t, err)
|
|
_, err = f.Seek(0, io.SeekStart)
|
|
assert.NoError(t, err)
|
|
buf, err = io.ReadAll(f)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "0bcdefghiJKLMNOP\x00\x00\x00ABCDzu", string(buf))
|
|
|
|
assert.NoError(t, f.Close())
|
|
|
|
// test rename
|
|
err = Rename(t.Context(), "test.txt", "test2.txt")
|
|
assert.NoError(t, err)
|
|
|
|
_, err = OpenFile(t.Context(), "test.txt", os.O_RDONLY)
|
|
assert.Error(t, err)
|
|
|
|
f, err = OpenFile(t.Context(), "test2.txt", os.O_RDONLY)
|
|
assert.NoError(t, err)
|
|
assert.NoError(t, f.Close())
|
|
|
|
// test remove
|
|
err = Remove(t.Context(), "test2.txt")
|
|
assert.NoError(t, err)
|
|
|
|
_, err = OpenFile(t.Context(), "test2.txt", os.O_RDONLY)
|
|
assert.Error(t, err)
|
|
|
|
// test stat
|
|
f, err = OpenFile(t.Context(), "test/test.txt", os.O_RDWR|os.O_CREATE)
|
|
assert.NoError(t, err)
|
|
stat, err := f.Stat()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "test.txt", stat.Name())
|
|
assert.EqualValues(t, 0, stat.Size())
|
|
_, err = f.Write([]byte("0123456789"))
|
|
assert.NoError(t, err)
|
|
stat, err = f.Stat()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 10, stat.Size())
|
|
|
|
t.Run("NonExisting", func(t *testing.T) {
|
|
f, err := OpenFile(t.Context(), "non-existing.txt", os.O_RDONLY)
|
|
assert.ErrorIs(t, err, os.ErrNotExist)
|
|
assert.Nil(t, f)
|
|
|
|
f, err = OpenFile(t.Context(), "non-existing.txt", os.O_WRONLY)
|
|
assert.ErrorIs(t, err, os.ErrNotExist)
|
|
assert.Nil(t, f)
|
|
|
|
f, err = OpenFile(t.Context(), "non-existing.txt", os.O_WRONLY|os.O_APPEND|os.O_TRUNC)
|
|
assert.ErrorIs(t, err, os.ErrNotExist)
|
|
assert.Nil(t, f)
|
|
})
|
|
|
|
t.Run("Existing", func(t *testing.T) {
|
|
assertFileContent := func(f File, expected string) {
|
|
_, err := f.Seek(0, io.SeekStart)
|
|
require.NoError(t, err)
|
|
buf, err := io.ReadAll(f)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, expected, string(buf))
|
|
}
|
|
|
|
f, err := OpenFile(t.Context(), "existing.txt", os.O_RDWR|os.O_CREATE)
|
|
require.NoError(t, err)
|
|
_, _ = f.Write([]byte("test"))
|
|
assertFileContent(f, "test")
|
|
assert.NoError(t, f.Close())
|
|
|
|
f, err = OpenFile(t.Context(), "existing.txt", os.O_RDWR|os.O_CREATE|os.O_APPEND)
|
|
require.NoError(t, err)
|
|
_, _ = f.Write([]byte("\nnew"))
|
|
assertFileContent(f, "test\nnew")
|
|
assert.NoError(t, f.Close())
|
|
|
|
f, err = OpenFile(t.Context(), "existing.txt", os.O_RDWR|os.O_TRUNC)
|
|
require.NoError(t, err)
|
|
assertFileContent(f, "")
|
|
assert.NoError(t, f.Close())
|
|
|
|
f, err = OpenFile(t.Context(), "existing.txt", os.O_RDWR|os.O_CREATE|os.O_EXCL)
|
|
assert.ErrorIs(t, err, os.ErrExist)
|
|
assert.Nil(t, f)
|
|
})
|
|
}
|
|
|
|
func TestDbfsReadWrite(t *testing.T) {
|
|
defer test.MockVariableValue(&defaultFileBlockSize, 4)()
|
|
|
|
f1, err := OpenFile(t.Context(), "test.log", os.O_RDWR|os.O_CREATE)
|
|
assert.NoError(t, err)
|
|
defer f1.Close()
|
|
|
|
f2, err := OpenFile(t.Context(), "test.log", os.O_RDONLY)
|
|
assert.NoError(t, err)
|
|
defer f2.Close()
|
|
|
|
_, err = f1.Write([]byte("line 1\n"))
|
|
assert.NoError(t, err)
|
|
|
|
f2r := bufio.NewReader(f2)
|
|
|
|
line, err := f2r.ReadString('\n')
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "line 1\n", line)
|
|
_, err = f2r.ReadString('\n')
|
|
assert.ErrorIs(t, err, io.EOF)
|
|
|
|
_, err = f1.Write([]byte("line 2\n"))
|
|
assert.NoError(t, err)
|
|
|
|
line, err = f2r.ReadString('\n')
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "line 2\n", line)
|
|
_, err = f2r.ReadString('\n')
|
|
assert.ErrorIs(t, err, io.EOF)
|
|
}
|
|
|
|
func TestDbfsSeekWrite(t *testing.T) {
|
|
defer test.MockVariableValue(&defaultFileBlockSize, 4)()
|
|
|
|
// write something
|
|
fw, err := OpenFile(t.Context(), "test2.log", os.O_RDWR|os.O_CREATE)
|
|
require.NoError(t, err)
|
|
defer fw.Close()
|
|
|
|
n, err := fw.Write([]byte("111"))
|
|
assert.NoError(t, err)
|
|
|
|
_, err = fw.Seek(int64(n), io.SeekStart)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = fw.Write([]byte("222"))
|
|
assert.NoError(t, err)
|
|
|
|
_, err = fw.Seek(int64(n), io.SeekStart)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = fw.Write([]byte("333"))
|
|
assert.NoError(t, err)
|
|
|
|
// then read it
|
|
fr, err := OpenFile(t.Context(), "test2.log", os.O_RDONLY)
|
|
require.NoError(t, err)
|
|
defer fr.Close()
|
|
|
|
buf, err := io.ReadAll(fr)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "111333", string(buf))
|
|
}
|