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
|
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 {
|
func status(c *gin.Context, downloader youtube.Downloader) error {
|
||||||
id, err := downloader.ExtractVideoID()
|
id, err := downloader.ExtractVideoID()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -79,11 +89,7 @@ func main() {
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
r.GET("/download", func(c *gin.Context) {
|
r.POST("/task", createTask)
|
||||||
if err := download(c, youtube_dl.NewYoutubeDlDownloader(c.Query("url"), "yt-dlp")); err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
r.GET("/status", func(c *gin.Context) {
|
r.GET("/status", func(c *gin.Context) {
|
||||||
if err := status(c, youtube_dl.NewYoutubeDlDownloader(c.Query("url"), "yt-dlp")); err != nil {
|
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) {
|
func (d *YoutubeDlDownloader) getVideoJson(url string) (*youtube.Video, error) {
|
||||||
cmd := exec.CommandContext(context.Background(), d.CommandName, "-j", url)
|
cmd := exec.CommandContext(context.Background(), d.CommandName, "-j", url)
|
||||||
output, err := cmd.Output()
|
output, err := cmd.CombinedOutput()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, fmt.Errorf("Command terminated with %w: %s", err, output)
|
||||||
}
|
}
|
||||||
result := new(youtube.Video)
|
result := new(youtube.Video)
|
||||||
err = json.Unmarshal(output, &result)
|
err = json.Unmarshal(output, &result)
|
||||||
|
|
Loading…
Reference in a new issue