8b5a80f766
* fix(server/files): fix Content-Disposition is not set * test(header): check content disposition header
49 lines
715 B
Go
49 lines
715 B
Go
package server
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/ihexxa/gocfg"
|
|
"github.com/ihexxa/quickshare/src/client"
|
|
)
|
|
|
|
func startTestServer(config string) *Server {
|
|
defaultCfg, err := DefaultConfig()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
cfg, err := gocfg.New(NewConfig()).
|
|
Load(
|
|
gocfg.JSONStr(defaultCfg),
|
|
gocfg.JSONStr(config),
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
srv, err := NewServer(cfg)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
go srv.Start()
|
|
return srv
|
|
}
|
|
|
|
func waitForReady(addr string) bool {
|
|
retry := 20
|
|
setCl := client.NewSettingsClient(addr)
|
|
|
|
for retry > 0 {
|
|
_, _, errs := setCl.Health()
|
|
if len(errs) > 0 {
|
|
time.Sleep(100 * time.Millisecond)
|
|
} else {
|
|
return true
|
|
}
|
|
retry--
|
|
}
|
|
|
|
return false
|
|
}
|