4e8e71e4e2
This is not yet working, it's just a direct copy of the Manager of Flamenco 2, with Logrus replaced by Zerolog. The API has been documented in flamenco-manager.yaml as a starting point for the integration.
117 lines
3.2 KiB
Go
117 lines
3.2 KiB
Go
/* (c) 2019, Blender Foundation - Sybren A. Stüvel
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
* the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be
|
|
* included in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
|
* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
package filestore
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"runtime"
|
|
"time"
|
|
|
|
"git.blender.org/flamenco/pkg/shaman/config"
|
|
)
|
|
|
|
// CreateTestStore returns a Store that can be used for unit testing.
|
|
func CreateTestStore() *Store {
|
|
tempDir, err := ioutil.TempDir("", "shaman-filestore-test-")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
conf := config.Config{
|
|
FileStorePath: tempDir,
|
|
}
|
|
storage := New(conf)
|
|
store, ok := storage.(*Store)
|
|
if !ok {
|
|
panic("storage should be *Store")
|
|
}
|
|
|
|
return store
|
|
}
|
|
|
|
// CleanupTestStore deletes a store returned by CreateTestStore()
|
|
func CleanupTestStore(store *Store) {
|
|
if err := os.RemoveAll(store.baseDir); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// MustStoreFileForTest allows a unit test to store some file in the 'stored' storage bin.
|
|
// Any error will cause a panic.
|
|
func (s *Store) MustStoreFileForTest(checksum string, filesize int64, contents []byte) {
|
|
file, err := s.OpenForUpload(checksum, filesize)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer file.Close()
|
|
|
|
written, err := file.Write(contents)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if written != len(contents) {
|
|
panic("short write")
|
|
}
|
|
|
|
err = s.MoveToStored(checksum, filesize, file.Name())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// LinkTestFileStore creates a copy of _test_file_store by hard-linking files into a temporary directory.
|
|
// Panics if there are any errors.
|
|
func LinkTestFileStore(cloneTo string) {
|
|
_, myFilename, _, _ := runtime.Caller(0)
|
|
fileStorePath := path.Join(path.Dir(path.Dir(myFilename)), "_test_file_store")
|
|
now := time.Now()
|
|
|
|
visit := func(visitPath string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
relpath, err := filepath.Rel(fileStorePath, visitPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
targetPath := path.Join(cloneTo, relpath)
|
|
if info.IsDir() {
|
|
return os.MkdirAll(targetPath, 0755)
|
|
}
|
|
err = os.Link(visitPath, targetPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Make sure we always test with fresh files by default.
|
|
return os.Chtimes(targetPath, now, now)
|
|
}
|
|
if err := filepath.Walk(fileStorePath, visit); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|