// GoToSocial // Copyright (C) GoToSocial Authors admin@gotosocial.org // SPDX-License-Identifier: AGPL-3.0-or-later // // 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 testrig import ( "context" "fmt" "os" "path" "codeberg.org/gruf/go-storage/memory" gtsstorage "github.com/superseriousbusiness/gotosocial/internal/storage" ) // NewInMemoryStorage returns a new in memory storage with the default test config func NewInMemoryStorage() *gtsstorage.Driver { storage := memory.Open(200, false) return >sstorage.Driver{ Storage: storage, } } // StandardStorageSetup populates the storage with standard test entries from the given directory. func StandardStorageSetup(storage *gtsstorage.Driver, relativePath string) { storedA := newTestStoredAttachments() a := NewTestAttachments() for k, paths := range storedA { attachmentInfo, ok := a[k] if !ok { panic(fmt.Errorf("key %s not found in test attachments", k)) } filenameOriginal := paths.Original filenameSmall := paths.Small pathOriginal := attachmentInfo.File.Path pathSmall := attachmentInfo.Thumbnail.Path bOriginal, err := os.ReadFile(fmt.Sprintf("%s/%s", relativePath, filenameOriginal)) if err != nil { panic(err) } if _, err := storage.Put(context.TODO(), pathOriginal, bOriginal); err != nil { panic(err) } bSmall, err := os.ReadFile(fmt.Sprintf("%s/%s", relativePath, filenameSmall)) if err != nil { panic(err) } if _, err := storage.Put(context.TODO(), pathSmall, bSmall); err != nil { panic(err) } } storedE := newTestStoredEmoji() e := NewTestEmojis() for k, paths := range storedE { emojiInfo, ok := e[k] if !ok { panic(fmt.Errorf("key %s not found in test emojis", k)) } filenameOriginal := paths.Original filenameStatic := paths.Static pathOriginal := emojiInfo.ImagePath pathStatic := emojiInfo.ImageStaticPath bOriginal, err := os.ReadFile(fmt.Sprintf("%s/%s", relativePath, filenameOriginal)) if err != nil { panic(err) } if _, err := storage.Put(context.TODO(), pathOriginal, bOriginal); err != nil { panic(err) } bStatic, err := os.ReadFile(fmt.Sprintf("%s/%s", relativePath, filenameStatic)) if err != nil { panic(err) } if _, err := storage.Put(context.TODO(), pathStatic, bStatic); err != nil { panic(err) } } } // StandardStorageTeardown deletes everything in storage so that it's clean for the next test. func StandardStorageTeardown(storage *gtsstorage.Driver) { defer os.RemoveAll(path.Join(os.TempDir(), "gotosocial")) var keys []string _ = storage.WalkKeys(context.Background(), func(key string) error { keys = append(keys, key) return nil }) for _, key := range keys { _ = storage.Delete(context.Background(), key) } }