forked from TrueCloudLab/lego
99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package api
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/rsa"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/go-acme/lego/v3/acme"
|
|
"github.com/go-acme/lego/v3/platform/tester"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
jose "gopkg.in/square/go-jose.v2"
|
|
)
|
|
|
|
func TestOrderService_New(t *testing.T) {
|
|
mux, apiURL, tearDown := tester.SetupFakeAPI()
|
|
defer tearDown()
|
|
|
|
// small value keeps test fast
|
|
privateKey, errK := rsa.GenerateKey(rand.Reader, 512)
|
|
require.NoError(t, errK, "Could not generate test key")
|
|
|
|
mux.HandleFunc("/newOrder", func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
|
|
body, err := readSignedBody(r, privateKey)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
}
|
|
|
|
order := acme.Order{}
|
|
err = json.Unmarshal(body, &order)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
w.Header().Add("Link", `<https://example.com/acme/cert/1>;rel="alternate"`)
|
|
w.Header().Add("Link", `<https://example.com/acme/cert/2>;title="foo";rel="alternate"`)
|
|
w.Header().Add("Link", `<https://example.com/acme/cert/3>;title="foo";rel="alternate", <https://example.com/acme/cert/4>;rel="alternate"`)
|
|
|
|
err = tester.WriteJSONResponse(w, acme.Order{
|
|
Status: acme.StatusValid,
|
|
Identifiers: order.Identifiers,
|
|
})
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
})
|
|
|
|
core, err := New(http.DefaultClient, "lego-test", apiURL+"/dir", "", privateKey)
|
|
require.NoError(t, err)
|
|
|
|
order, err := core.Orders.New([]string{"example.com"})
|
|
require.NoError(t, err)
|
|
|
|
expected := acme.ExtendedOrder{
|
|
Order: acme.Order{
|
|
Status: "valid",
|
|
Identifiers: []acme.Identifier{{Type: "dns", Value: "example.com"}},
|
|
},
|
|
AlternateChainLinks: []string{
|
|
"https://example.com/acme/cert/1",
|
|
"https://example.com/acme/cert/2",
|
|
"https://example.com/acme/cert/3",
|
|
"https://example.com/acme/cert/4",
|
|
},
|
|
}
|
|
assert.Equal(t, expected, order)
|
|
}
|
|
|
|
func readSignedBody(r *http.Request, privateKey *rsa.PrivateKey) ([]byte, error) {
|
|
reqBody, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
jws, err := jose.ParseSigned(string(reqBody))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
body, err := jws.Verify(&jose.JSONWebKey{
|
|
Key: privateKey.Public(),
|
|
Algorithm: "RSA",
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return body, nil
|
|
}
|