marcobaobao
4c35b0b41f
introduced pipelines and abstracted download process.go in Downloader interface
40 lines
707 B
Go
40 lines
707 B
Go
package livestream
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/marcopiovanello/yt-dlp-web-ui/v3/server/config"
|
|
"github.com/marcopiovanello/yt-dlp-web-ui/v3/server/internal/kv"
|
|
"github.com/marcopiovanello/yt-dlp-web-ui/v3/server/internal/queue"
|
|
)
|
|
|
|
func setupTest() {
|
|
config.Instance().DownloaderPath = "build/yt-dlp"
|
|
}
|
|
|
|
const URL = "https://www.youtube.com/watch?v=pwoAyLGOysU"
|
|
|
|
func TestLivestream(t *testing.T) {
|
|
setupTest()
|
|
|
|
done := make(chan *LiveStream)
|
|
|
|
ls := New(URL, done, &queue.MessageQueue{}, &kv.Store{})
|
|
go ls.Start()
|
|
|
|
time.AfterFunc(time.Second*20, func() {
|
|
ls.Kill()
|
|
})
|
|
|
|
for {
|
|
select {
|
|
case wt := <-ls.WaitTime():
|
|
t.Log(wt)
|
|
case <-done:
|
|
t.Log("done")
|
|
return
|
|
}
|
|
}
|
|
}
|