package prxy

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

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/xlgmokha/x/pkg/x"
	"gitlab.com/mokhax/spike/pkg/test"
)

func TestProxy(t *testing.T) {
	t.Run("http://idp.test", func(t *testing.T) {
		var lastIdPRequest *http.Request
		var lastUiRequest *http.Request

		idp := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			lastIdPRequest = r
			w.WriteHeader(http.StatusOK)
		}))
		defer idp.Close()

		ui := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			lastUiRequest = r
			w.WriteHeader(http.StatusTeapot)
		}))
		defer ui.Close()

		subject := New(map[string]string{
			"idp.test": idp.URL,
			"ui.test":  ui.URL,
		})

		r, w := test.RequestResponse("GET", "http://idp.test:8080/saml/new")

		subject.ServeHTTP(w, r)

		url := x.Must(url.Parse(idp.URL))

		assert.Nil(t, lastUiRequest)
		assert.Equal(t, http.StatusOK, w.Code)

		require.NotNil(t, lastIdPRequest)
		assert.Equal(t, url.Host, lastIdPRequest.Host)
	})
}