diff --git a/storage/storage.go b/storage/storage.go index bc77f19..102bd27 100644 --- a/storage/storage.go +++ b/storage/storage.go @@ -8,8 +8,8 @@ import ( ) type IStorageProvider interface { - storeRaw(bucketName string, objectName string, data []byte) string - storeExisting(bucketName string, objectName string, existingFilePath string) string + StoreRaw(bucketName string, objectName string, data []byte) string + StoreExisting(bucketName string, objectName string, existingFilePath string) string } type FileSystemStorageProvider struct { @@ -17,7 +17,7 @@ type FileSystemStorageProvider struct { basePath string } -func (sp FileSystemStorageProvider) storeRaw(bucketName string, objectName string, data []byte) string { +func (sp FileSystemStorageProvider) StoreRaw(bucketName string, objectName string, data []byte) string { directoryPath := filepath.Join(sp.basePath, bucketName) sp.fileSystem.MkdirAll(directoryPath, os.ModePerm) @@ -29,8 +29,10 @@ func (sp FileSystemStorageProvider) storeRaw(bucketName string, objectName strin return filePath } -func (sp FileSystemStorageProvider) storeExisting(bucketName string, objectName string, existingFilePath string) string { +func (sp FileSystemStorageProvider) StoreExisting(bucketName string, objectName string, existingFilePath string) string { bytesRead, _ := afero.ReadFile(sp.fileSystem, existingFilePath) - return sp.storeRaw(bucketName, objectName, bytesRead) + return sp.StoreRaw(bucketName, objectName, bytesRead) +} + } diff --git a/storage/storage_test.go b/storage/storage_test.go index bf2c515..d0aa620 100644 --- a/storage/storage_test.go +++ b/storage/storage_test.go @@ -19,7 +19,7 @@ func TestFileSystemStorageProvider(t *testing.T) { basePath: "/tmp/foo/bar", } - finalPath := provider.storeRaw("test", "test.bin", dummyData) + finalPath := provider.StoreRaw("test", "test.bin", dummyData) assert.Equal(t, "/tmp/foo/bar/test/test.bin", finalPath) exists, _ := afero.Exists(fileSystem, "/tmp/foo/bar/test/test.bin") @@ -39,7 +39,7 @@ func TestFileSystemStorageProvider(t *testing.T) { basePath: "/tmp/foo/bar", } - finalPath := provider.storeExisting("test", "test.bin", "/tmp/existing.bin") + finalPath := provider.StoreExisting("test", "test.bin", "/tmp/existing.bin") assert.Equal(t, "/tmp/foo/bar/test/test.bin", finalPath) exists, _ := afero.Exists(fileSystem, "/tmp/foo/bar/test/test.bin")