package shrty

import (
	"testing"
	"github.com/stretchr/testify/assert"
)

func TestCheckURL(t *testing.T) {
	data := Data{
		URL: "https://t.co/kA45uWnKkg",
	}

	expected := Data{
		URL: "https://t.co/kA45uWnKkg",
		URLFetched: "https://www.youtube.com/watch?v=Qg0pO9VG1J8&utm_content=buffera4269&utm_medium=social&utm_source=twitter.com&utm_campaign=buffer",
		OriginalURL: "https://www.youtube.com/watch?v=Qg0pO9VG1J8&utm_content=buffera4269&utm_medium=social&utm_source=twitter.com&utm_campaign=buffer",
		HTTPStatusCode: "200",
		Domain: "www.youtube.com",
	}

	err := CheckURL(&data)
	assert.NoError(t, err)
	assert.Equal(t, expected, data)
}