mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-21 17:21:09 +01:00
57 lines
778 B
Go
57 lines
778 B
Go
package test
|
|
|
|
import (
|
|
fs "github.com/crawlab-team/crawlab/fs"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
T, err = NewTest()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
var T *Test
|
|
|
|
type Test struct {
|
|
m fs.Manager
|
|
}
|
|
|
|
func (t *Test) Setup(t2 *testing.T) {
|
|
t.Cleanup()
|
|
t2.Cleanup(t.Cleanup)
|
|
}
|
|
|
|
func (t *Test) Cleanup() {
|
|
_ = T.m.DeleteDir("/test")
|
|
|
|
// wait to avoid caching
|
|
time.Sleep(200 * time.Millisecond)
|
|
}
|
|
|
|
func NewTest() (res *Test, err error) {
|
|
// test
|
|
t := &Test{}
|
|
|
|
// filer url
|
|
filerUrl := os.Getenv("CRAWLAB_FILER_URL")
|
|
if filerUrl == "" {
|
|
filerUrl = "http://localhost:8888"
|
|
}
|
|
|
|
// manager
|
|
t.m, err = fs.NewSeaweedFsManager(
|
|
fs.WithFilerUrl(filerUrl),
|
|
fs.WithTimeout(10*time.Second),
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return t, nil
|
|
}
|