1
0
mirror of https://github.com/go-acme/lego.git synced 2024-12-23 09:15:11 +02:00
lego/registration/registar_test.go

57 lines
1.4 KiB
Go
Raw Normal View History

2019-03-11 18:56:48 +02:00
package registration
import (
"crypto/rand"
"crypto/rsa"
"net/http"
"testing"
2019-03-11 18:56:48 +02:00
"github.com/go-acme/lego/acme"
"github.com/go-acme/lego/acme/api"
"github.com/go-acme/lego/platform/tester"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestRegistrar_ResolveAccountByKey(t *testing.T) {
mux, apiURL, tearDown := tester.SetupFakeAPI()
defer tearDown()
2019-02-06 19:15:53 +02:00
mux.HandleFunc("/account", func(w http.ResponseWriter, _ *http.Request) {
w.Header().Set("Location", apiURL+"/account_recovery")
_, err := w.Write([]byte("{}"))
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
})
2019-02-06 19:15:53 +02:00
mux.HandleFunc("/account_recovery", func(w http.ResponseWriter, _ *http.Request) {
err := tester.WriteJSONResponse(w, acme.Account{
Status: "valid",
})
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
})
key, err := rsa.GenerateKey(rand.Reader, 512)
require.NoError(t, err, "Could not generate test key")
user := mockUser{
email: "test@test.com",
regres: &Resource{},
privatekey: key,
}
core, err := api.New(http.DefaultClient, "lego-test", apiURL+"/dir", "", key)
require.NoError(t, err)
registrar := NewRegistrar(core, user)
res, err := registrar.ResolveAccountByKey()
require.NoError(t, err, "Unexpected error resolving account by key")
assert.Equal(t, "valid", res.Body.Status, "Unexpected account status")
}