kurz/api/get_test.go

package api

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/gorilla/mux"

	"code.osinet.fr/fgm/kurz/domain"
)

// Use the special case of ErrUseLastResponse to ignore redirects.
func doNotFollowRedirects(_ *http.Request, _ []*http.Request) error {
	return http.ErrUseLastResponse
}

const SampleShort = "short"

func setupGet() (*httptest.Server, *http.Client) {
	router := mux.NewRouter()
	router.HandleFunc("/{short}", HandleGetShort)
	ts := httptest.NewServer(router)

	c := ts.Client()
	c.CheckRedirect = doNotFollowRedirects

	return ts, c
}

func TestHandleGetShortHappy(t *testing.T) {
	const SampleTarget = "http://example.com/"

	// Ensure storage contains the expected mapping.
	sr := make(domain.MockShortRepo)
	sr[domain.ShortURL{URL: SampleShort}] = domain.TargetURL{URL: SampleTarget}
	domain.RegisterRepositories(sr, nil)

	ts, c := setupGet()
	defer ts.Close()

	res, err := c.Get(ts.URL + "/" + SampleShort)

	if err != nil {
		t.Error("Getting an existing short URL should not fail")
	}
	res.Body.Close()

	if res.StatusCode != http.StatusTemporaryRedirect {
		t.Errorf("Existing short URL returned %d, expected %d", res.StatusCode, http.StatusTemporaryRedirect)
	}
}

func TestHandleGetShortSadNotFound(t *testing.T) {
	// Ensure empty storage.
	sr := make(domain.MockShortRepo)
	domain.RegisterRepositories(sr, nil)

	ts, c := setupGet()
	defer ts.Close()

	res, err := c.Get(ts.URL + "/" + SampleShort)

	if err != nil {
		t.Error("Getting an nonexistent short URL should not fail")
	}

	res.Body.Close()

	if res.StatusCode != http.StatusNotFound {
		t.Errorf("Existing short URL returned %d, expected %d", res.StatusCode, http.StatusNotFound)
	}
}