source_test.go

package main

import (
	"net/http"
	"net/url"
	"testing"
)

func TestMatchSource(t *testing.T) {
	u, _ := url.Parse("http://foo?url=http://bar/image.jpg")
	req := &http.Request{Method: http.MethodGet, URL: u}

	source := MatchSource(req)
	if source == nil {
		t.Error("Cannot match image source")
	}
}