Expose getStreamURL for use elsewhere

kkdai · Aug 22, 2020 · 0dcd24a · 0dcd24a
1 parent e074941
commit 0dcd24a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/client.go b/client.go
@@ -57,15 +57,19 @@ func (c *Client) GetStream(video *Video, format *Format) (*http.Response, error)
 
 // GetStreamContext returns the HTTP response for a specific format with a context
 func (c *Client) GetStreamContext(ctx context.Context, video *Video, format *Format) (*http.Response, error) {
-	url, err := c.getStreamURL(ctx, video, format)
+	url, err := c.GetStreamURLContext(ctx, video, format)
 	if err != nil {
 		return nil, err
 	}
 
 	return c.httpGet(ctx, url)
 }
 
-func (c *Client) getStreamURL(ctx context.Context, video *Video, format *Format) (string, error) {
+func (c *Client) GetStreamURL(video *Video, format *Format) (string, error) {
+	return c.GetStreamURLContext(context.Background(), video, format)
+}
+
+func (c *Client) GetStreamURLContext(ctx context.Context, video *Video, format *Format) (string, error) {
 	if format.URL != "" {
 		return format.URL, nil
 	}

diff --git a/video_test.go b/video_test.go
@@ -69,7 +69,7 @@ func TestDownload_Regular(t *testing.T) {
 				format = &video.Formats[0]
 			}
 
-			url, err := testClient.getStreamURL(ctx, video, format)
+			url, err := testClient.GetStreamURLContext(ctx, video, format)
 			require.NoError(err)
 			require.NotEmpty(url)
 		})