diff options
Diffstat (limited to 'internal/storage')
-rw-r--r-- | internal/storage/inmem.go | 31 | ||||
-rw-r--r-- | internal/storage/local.go | 21 | ||||
-rw-r--r-- | internal/storage/mock_Storage.go | 47 | ||||
-rw-r--r-- | internal/storage/storage.go | 24 |
4 files changed, 123 insertions, 0 deletions
diff --git a/internal/storage/inmem.go b/internal/storage/inmem.go new file mode 100644 index 000000000..25432fbaa --- /dev/null +++ b/internal/storage/inmem.go @@ -0,0 +1,31 @@ +package storage + +import ( + "fmt" + + "github.com/sirupsen/logrus" + "github.com/superseriousbusiness/gotosocial/internal/config" +) + +func NewInMem(c *config.Config, log *logrus.Logger) (Storage, error) { + return &inMemStorage{ + stored: make(map[string][]byte), + }, nil +} + +type inMemStorage struct { + stored map[string][]byte +} + +func (s *inMemStorage) StoreFileAt(path string, data []byte) error { + s.stored[path] = data + return nil +} + +func (s *inMemStorage) RetrieveFileFrom(path string) ([]byte, error) { + d, ok := s.stored[path] + if !ok { + return nil, fmt.Errorf("no data found at path %s", path) + } + return d, nil +} diff --git a/internal/storage/local.go b/internal/storage/local.go new file mode 100644 index 000000000..29461d5d4 --- /dev/null +++ b/internal/storage/local.go @@ -0,0 +1,21 @@ +package storage + +import ( + "github.com/sirupsen/logrus" + "github.com/superseriousbusiness/gotosocial/internal/config" +) + +func NewLocal(c *config.Config, log *logrus.Logger) (Storage, error) { + return &localStorage{}, nil +} + +type localStorage struct { +} + +func (s *localStorage) StoreFileAt(path string, data []byte) error { + return nil +} + +func (s *localStorage) RetrieveFileFrom(path string) ([]byte, error) { + return nil, nil +} diff --git a/internal/storage/mock_Storage.go b/internal/storage/mock_Storage.go new file mode 100644 index 000000000..865d52205 --- /dev/null +++ b/internal/storage/mock_Storage.go @@ -0,0 +1,47 @@ +// Code generated by mockery v2.7.4. DO NOT EDIT. + +package storage + +import mock "github.com/stretchr/testify/mock" + +// MockStorage is an autogenerated mock type for the Storage type +type MockStorage struct { + mock.Mock +} + +// RetrieveFileFrom provides a mock function with given fields: path +func (_m *MockStorage) RetrieveFileFrom(path string) ([]byte, error) { + ret := _m.Called(path) + + var r0 []byte + if rf, ok := ret.Get(0).(func(string) []byte); ok { + r0 = rf(path) + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).([]byte) + } + } + + var r1 error + if rf, ok := ret.Get(1).(func(string) error); ok { + r1 = rf(path) + } else { + r1 = ret.Error(1) + } + + return r0, r1 +} + +// StoreFileAt provides a mock function with given fields: path, data +func (_m *MockStorage) StoreFileAt(path string, data []byte) error { + ret := _m.Called(path, data) + + var r0 error + if rf, ok := ret.Get(0).(func(string, []byte) error); ok { + r0 = rf(path, data) + } else { + r0 = ret.Error(0) + } + + return r0 +} diff --git a/internal/storage/storage.go b/internal/storage/storage.go new file mode 100644 index 000000000..fa884ed07 --- /dev/null +++ b/internal/storage/storage.go @@ -0,0 +1,24 @@ +/* + GoToSocial + Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see <http://www.gnu.org/licenses/>. +*/ + +package storage + +type Storage interface { + StoreFileAt(path string, data []byte) error + RetrieveFileFrom(path string) ([]byte, error) +} |