package handlers

import (
	"encoding/json"
	"testing"

	"github.com/stretchr/testify/assert"

	"edge-infra.dev/pkg/edge/auth-proxy/faker"
)

func TestGoodTokenExchange(t *testing.T) {
	req, err := json.Marshal(faker.MockGoodTokenExchangeResponse())
	assert.NoError(t, err)
	_, body, err := proxyHandler.TokenExchange(nil, req)
	assert.NoError(t, err)
	assert.Equal(t, req, body)
}

func TestBadTokenExchange(t *testing.T) {
	req, err := json.Marshal(faker.MockBadTokenExchangeResponse())
	assert.NoError(t, err)
	_, body, err := proxyHandler.TokenExchange(nil, req)
	assert.Error(t, err)
	assert.Equal(t, req, body)
}