offline-twitter/scraper/link_expander_test.go

package scraper_test

import (
	"testing"

	"net/http"
	"net/http/httptest"

	"github.com/stretchr/testify/assert"

	. "offline_twitter/scraper"
)

func TestExpandShortUrl(t *testing.T) {
	redirecting_to := "redirect target"
	srvr := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
		w.Header().Set("Location", redirecting_to)
		w.WriteHeader(301)
	}))
	defer srvr.Close()

	assert.Equal(t, redirecting_to, ExpandShortUrl(srvr.URL))
}