32 lines
800 B
Go
32 lines
800 B
Go
package scraper_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
. "gitlab.com/offline-twitter/twitter_offline_engine/pkg/scraper"
|
|
)
|
|
|
|
func TestParseAPIMedia(t *testing.T) {
|
|
assert := assert.New(t)
|
|
data, err := os.ReadFile("test_responses/tweet_content/image.json")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var apimedia APIMedia
|
|
err = json.Unmarshal(data, &apimedia)
|
|
require.NoError(t, err)
|
|
|
|
image := ParseAPIMedia(apimedia)
|
|
assert.Equal(ImageID(1395882862289772553), image.ID)
|
|
assert.Equal("https://pbs.twimg.com/media/E18sEUrWYAk8dBl.jpg", image.RemoteURL)
|
|
assert.Equal(593, image.Width)
|
|
assert.Equal(239, image.Height)
|
|
assert.Equal("E1/E18sEUrWYAk8dBl.jpg", image.LocalFilename)
|
|
assert.False(image.IsDownloaded)
|
|
}
|