51 lines
1 KiB
Go
51 lines
1 KiB
Go
package cwhub
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/jarcoal/httpmock"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDownloadFile(t *testing.T) {
|
|
examplePath := "./example.txt"
|
|
defer os.Remove(examplePath)
|
|
|
|
httpmock.Activate()
|
|
defer httpmock.DeactivateAndReset()
|
|
|
|
// OK
|
|
httpmock.RegisterResponder(
|
|
"GET",
|
|
"https://example.com/xx",
|
|
httpmock.NewStringResponder(200, "example content oneoneone"),
|
|
)
|
|
|
|
httpmock.RegisterResponder(
|
|
"GET",
|
|
"https://example.com/x",
|
|
httpmock.NewStringResponder(404, "not found"),
|
|
)
|
|
|
|
err := downloadFile("https://example.com/xx", examplePath)
|
|
require.NoError(t, err)
|
|
|
|
content, err := os.ReadFile(examplePath)
|
|
assert.Equal(t, "example content oneoneone", string(content))
|
|
require.NoError(t, err)
|
|
|
|
// bad uri
|
|
err = downloadFile("https://zz.com", examplePath)
|
|
require.Error(t, err)
|
|
|
|
// 404
|
|
err = downloadFile("https://example.com/x", examplePath)
|
|
require.Error(t, err)
|
|
|
|
// bad target
|
|
err = downloadFile("https://example.com/xx", "")
|
|
require.Error(t, err)
|
|
}
|