Change REST api verbs
This commit is contained in:
parent
5b2b356ddd
commit
aecbffff46
3 changed files with 74 additions and 7 deletions
|
@ -33,6 +33,16 @@ func download(c *gin.Context, downloader youtube.Downloader) error {
|
|||
return nil
|
||||
}
|
||||
|
||||
func createTask(c *gin.Context) {
|
||||
err := c.Request.ParseForm()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if err := download(c, youtube_dl.NewYoutubeDlDownloader(c.PostForm("url"), "yt-dlp")); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
func status(c *gin.Context, downloader youtube.Downloader) error {
|
||||
id, err := downloader.ExtractVideoID()
|
||||
if err != nil {
|
||||
|
@ -79,11 +89,7 @@ func main() {
|
|||
}
|
||||
})
|
||||
|
||||
r.GET("/download", func(c *gin.Context) {
|
||||
if err := download(c, youtube_dl.NewYoutubeDlDownloader(c.Query("url"), "yt-dlp")); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
})
|
||||
r.POST("/task", createTask)
|
||||
|
||||
r.GET("/status", func(c *gin.Context) {
|
||||
if err := status(c, youtube_dl.NewYoutubeDlDownloader(c.Query("url"), "yt-dlp")); err != nil {
|
||||
|
|
61
backend/main_test.go
Normal file
61
backend/main_test.go
Normal file
|
@ -0,0 +1,61 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"net/url"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/remogatto/prettytest"
|
||||
)
|
||||
|
||||
// Start of setup
|
||||
type testSuite struct {
|
||||
prettytest.Suite
|
||||
}
|
||||
|
||||
func TestRunner(t *testing.T) {
|
||||
prettytest.Run(
|
||||
t,
|
||||
new(testSuite),
|
||||
)
|
||||
}
|
||||
|
||||
// Test the creation of a new task. A new task is created with a POST
|
||||
// request to the endpoint.
|
||||
func (t *testSuite) TestCreateTask() {
|
||||
// Set form values
|
||||
form := url.Values{}
|
||||
form.Add("url", "https://www.youtube.com/watch?v=AVIBLFl28vo")
|
||||
|
||||
// Create a new request
|
||||
req, err := http.NewRequest("POST", "/task", strings.NewReader(form.Encode()))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||
|
||||
// Create an http recorder
|
||||
rr := httptest.NewRecorder()
|
||||
|
||||
// Set the context and call the handler
|
||||
c, _ := gin.CreateTestContext(rr)
|
||||
c.Request = req
|
||||
|
||||
createTask(c)
|
||||
|
||||
// Read the response and assert it.
|
||||
var data struct {
|
||||
Id string
|
||||
}
|
||||
|
||||
if err := json.Unmarshal(rr.Body.Bytes(), &data); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
t.Equal("AVIBLFl28vo", data.Id)
|
||||
}
|
|
@ -63,9 +63,9 @@ func (d *YoutubeDlDownloader) ExtractVideoID() (string, error) {
|
|||
|
||||
func (d *YoutubeDlDownloader) getVideoJson(url string) (*youtube.Video, error) {
|
||||
cmd := exec.CommandContext(context.Background(), d.CommandName, "-j", url)
|
||||
output, err := cmd.Output()
|
||||
output, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
return nil, fmt.Errorf("Command terminated with %w: %s", err, output)
|
||||
}
|
||||
result := new(youtube.Video)
|
||||
err = json.Unmarshal(output, &result)
|
||||
|
|
Loading…
Reference in a new issue