client io architecture

server/internal/repo/fs/fs_test.go

@@ -1,26 +1,45 @@
-package fs
+package fs_test
 
 import (
 	"os"
 	"path/filepath"
+	"slices"
 	"testing"
 
+	"galaxy/server/internal/repo/fs"
 	"galaxy/util"
 
 	"github.com/stretchr/testify/assert"
 )
 
+const (
+	lockFile = ".lock"
+)
+
+type sampleData struct {
+	data []byte
+}
+
+func (sd *sampleData) UnmarshalBinary(data []byte) error {
+	sd.data = slices.Clone(data)
+	return nil
+}
+
+func (sd sampleData) MarshalBinary() (data []byte, err error) {
+	return sd.data, nil
+}
+
 func TestNewFileStorageSuccess(t *testing.T) {
 	root, cleanup := util.CreateWorkDir(t)
 	defer cleanup()
-	_, err := NewFileStorage(root)
+	_, err := fs.NewFileStorage(root)
 	assert.NoError(t, err)
 }
 
 func TestLock(t *testing.T) {
 	root, cleanup := util.CreateWorkDir(t)
 	defer cleanup()
-	fs, err := NewFileStorage(root)
+	fs, err := fs.NewFileStorage(root)
 	assert.NoError(t, err, "create file storage")
 	unlock, err := fs.Lock()
 	assert.NoError(t, err, "acquire lock")
@@ -40,7 +59,7 @@ func TestExist(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	fs, err := NewFileStorage(root)
+	fs, err := fs.NewFileStorage(root)
 	assert.NoError(t, err, "create file storage")
 
 	exist, err := fs.Exists(fileName)
@@ -56,7 +75,7 @@ func TestWrite(t *testing.T) {
 	root, cleanup := util.CreateWorkDir(t)
 	defer cleanup()
 
-	fs, err := NewFileStorage(root)
+	fs, err := fs.NewFileStorage(root)
 	assert.NoError(t, err, "create file storage: %s", err)
 
 	unlock, err := fs.Lock()
@@ -99,7 +118,7 @@ func TestRead(t *testing.T) {
 
 	sd := new(sampleData)
 
-	fs, err := NewFileStorage(root)
+	fs, err := fs.NewFileStorage(root)
 	assert.NoError(t, err, "create file storage: %s", err)
 
 	assert.EqualError(t, fs.Read("some.file", sd), "lock must be acquired before read")
@@ -144,7 +163,7 @@ func TestRead(t *testing.T) {
 func TestWriteErrorWithoutLock(t *testing.T) {
 	root, cleanup := util.CreateWorkDir(t)
 	defer cleanup()
-	fs, err := NewFileStorage(root)
+	fs, err := fs.NewFileStorage(root)
 	assert.NoError(t, err, "create file storage")
 	sd := &sampleData{[]byte{0, 1, 2, 3}}
 	err = fs.Write("some/path", sd)
@@ -153,7 +172,7 @@ func TestWriteErrorWithoutLock(t *testing.T) {
 }
 
 func TestNewFileStorageErrorNotExists(t *testing.T) {
-	_, err := NewFileStorage(filepath.Join(os.TempDir(), "non-existent-dir"))
+	_, err := fs.NewFileStorage(filepath.Join(os.TempDir(), "non-existent-dir"))
 	assert.Error(t, err)
 }
 
@@ -165,7 +184,7 @@ func TestNewFileStorageErrorNotADirectory(t *testing.T) {
 	if err := f.Close(); err != nil {
 		t.Fatal(err)
 	}
-	_, err = NewFileStorage(f.Name())
+	_, err = fs.NewFileStorage(f.Name())
 	assert.Error(t, err)
 	if err := os.Remove(f.Name()); err != nil {
 		t.Fatal(err)
@@ -173,6 +192,6 @@ func TestNewFileStorageErrorNotADirectory(t *testing.T) {
 }
 
 func TestNewFileStorageErrorNoAccess(t *testing.T) {
-	_, err := NewFileStorage(nonWritableDir)
+	_, err := fs.NewFileStorage("/some/random/dir")
 	assert.Error(t, err)
 }