package scraper_test

import (
	"testing"

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

	"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()

	result := scraper.ExpandShortUrl(srvr.URL)
	if result != redirecting_to {
		t.Errorf("Expected %q, got %q", redirecting_to, result)
	}
}