forked from TrueCloudLab/certificates
Fix test and change method name
This commit is contained in:
parent
f34fb80eb6
commit
272cce522e
4 changed files with 342 additions and 341 deletions
|
@ -251,337 +251,6 @@ func TestGetAccountByKeyID(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func Test_getOrderIDsByAccount(t *testing.T) {
|
|
||||||
type test struct {
|
|
||||||
id string
|
|
||||||
db nosql.DB
|
|
||||||
res []string
|
|
||||||
err *Error
|
|
||||||
}
|
|
||||||
tests := map[string]func(t *testing.T) test{
|
|
||||||
"ok/not-found": func(t *testing.T) test {
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
return nil, database.ErrNotFound
|
|
||||||
},
|
|
||||||
},
|
|
||||||
res: []string{},
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"fail/db-error": func(t *testing.T) test {
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
return nil, errors.New("force")
|
|
||||||
},
|
|
||||||
},
|
|
||||||
err: ServerInternalErr(errors.New("error loading orderIDs for account foo: force")),
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"fail/unmarshal-error": func(t *testing.T) test {
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
assert.Equals(t, bucket, ordersByAccountIDTable)
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return nil, nil
|
|
||||||
},
|
|
||||||
},
|
|
||||||
err: ServerInternalErr(errors.New("error unmarshaling orderIDs for account foo: unexpected end of JSON input")),
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"fail/error-loading-order-from-order-IDs": func(t *testing.T) test {
|
|
||||||
oids := []string{"o1", "o2", "o3"}
|
|
||||||
boids, err := json.Marshal(oids)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
dbHit := 0
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
dbHit++
|
|
||||||
switch dbHit {
|
|
||||||
case 1:
|
|
||||||
assert.Equals(t, bucket, ordersByAccountIDTable)
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return boids, nil
|
|
||||||
case 2:
|
|
||||||
assert.Equals(t, bucket, orderTable)
|
|
||||||
assert.Equals(t, key, []byte("o1"))
|
|
||||||
return nil, errors.New("force")
|
|
||||||
default:
|
|
||||||
assert.FatalError(t, errors.New("should not be here"))
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
},
|
|
||||||
},
|
|
||||||
err: ServerInternalErr(errors.New("error loading order o1 for account foo: error loading order o1: force")),
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"fail/error-updating-order-from-order-IDs": func(t *testing.T) test {
|
|
||||||
oids := []string{"o1", "o2", "o3"}
|
|
||||||
boids, err := json.Marshal(oids)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
o, err := newO()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
bo, err := json.Marshal(o)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
dbHit := 0
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
dbHit++
|
|
||||||
switch dbHit {
|
|
||||||
case 1:
|
|
||||||
assert.Equals(t, bucket, ordersByAccountIDTable)
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return boids, nil
|
|
||||||
case 2:
|
|
||||||
assert.Equals(t, bucket, orderTable)
|
|
||||||
assert.Equals(t, key, []byte("o1"))
|
|
||||||
return bo, nil
|
|
||||||
case 3:
|
|
||||||
assert.Equals(t, bucket, authzTable)
|
|
||||||
assert.Equals(t, key, []byte(o.Authorizations[0]))
|
|
||||||
return nil, errors.New("force")
|
|
||||||
default:
|
|
||||||
assert.FatalError(t, errors.New("should not be here"))
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
},
|
|
||||||
},
|
|
||||||
err: ServerInternalErr(errors.Errorf("error updating order o1 for account foo: error loading authz %s: force", o.Authorizations[0])),
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"ok/no-change-to-pending-orders": func(t *testing.T) test {
|
|
||||||
oids := []string{"o1", "o2", "o3"}
|
|
||||||
boids, err := json.Marshal(oids)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
o, err := newO()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
bo, err := json.Marshal(o)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
az, err := newAz()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
baz, err := json.Marshal(az)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
ch, err := newDNSCh()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
bch, err := json.Marshal(ch)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
switch string(bucket) {
|
|
||||||
case string(ordersByAccountIDTable):
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return boids, nil
|
|
||||||
case string(orderTable):
|
|
||||||
return bo, nil
|
|
||||||
case string(authzTable):
|
|
||||||
return baz, nil
|
|
||||||
case string(challengeTable):
|
|
||||||
return bch, nil
|
|
||||||
default:
|
|
||||||
assert.FatalError(t, errors.Errorf("did not expect query to table %s", bucket))
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
},
|
|
||||||
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
||||||
return nil, false, errors.New("should not be attempting to store anything")
|
|
||||||
},
|
|
||||||
},
|
|
||||||
res: oids,
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"fail/error-storing-new-oids": func(t *testing.T) test {
|
|
||||||
oids := []string{"o1", "o2", "o3"}
|
|
||||||
boids, err := json.Marshal(oids)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
o, err := newO()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
bo, err := json.Marshal(o)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
invalidOrder, err := newO()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
invalidOrder.Status = StatusInvalid
|
|
||||||
binvalidOrder, err := json.Marshal(invalidOrder)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
az, err := newAz()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
baz, err := json.Marshal(az)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
ch, err := newDNSCh()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
bch, err := json.Marshal(ch)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
dbGetOrder := 0
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
switch string(bucket) {
|
|
||||||
case string(ordersByAccountIDTable):
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return boids, nil
|
|
||||||
case string(orderTable):
|
|
||||||
dbGetOrder++
|
|
||||||
if dbGetOrder == 1 {
|
|
||||||
return binvalidOrder, nil
|
|
||||||
}
|
|
||||||
return bo, nil
|
|
||||||
case string(authzTable):
|
|
||||||
return baz, nil
|
|
||||||
case string(challengeTable):
|
|
||||||
return bch, nil
|
|
||||||
default:
|
|
||||||
assert.FatalError(t, errors.Errorf("did not expect query to table %s", bucket))
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
},
|
|
||||||
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
||||||
assert.Equals(t, bucket, ordersByAccountIDTable)
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return nil, false, errors.New("force")
|
|
||||||
},
|
|
||||||
},
|
|
||||||
err: ServerInternalErr(errors.New("error storing orderIDs as part of getOrderIDsByAccount logic: len(orderIDs) = 2: error storing order IDs for account foo: force")),
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"ok": func(t *testing.T) test {
|
|
||||||
oids := []string{"o1", "o2", "o3", "o4"}
|
|
||||||
boids, err := json.Marshal(oids)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
o, err := newO()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
bo, err := json.Marshal(o)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
invalidOrder, err := newO()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
invalidOrder.Status = StatusInvalid
|
|
||||||
binvalidOrder, err := json.Marshal(invalidOrder)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
az, err := newAz()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
baz, err := json.Marshal(az)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
ch, err := newDNSCh()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
bch, err := json.Marshal(ch)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
dbGetOrder := 0
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
switch string(bucket) {
|
|
||||||
case string(ordersByAccountIDTable):
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return boids, nil
|
|
||||||
case string(orderTable):
|
|
||||||
dbGetOrder++
|
|
||||||
if dbGetOrder == 1 || dbGetOrder == 3 {
|
|
||||||
return binvalidOrder, nil
|
|
||||||
}
|
|
||||||
return bo, nil
|
|
||||||
case string(authzTable):
|
|
||||||
return baz, nil
|
|
||||||
case string(challengeTable):
|
|
||||||
return bch, nil
|
|
||||||
default:
|
|
||||||
assert.FatalError(t, errors.Errorf("did not expect query to table %s", bucket))
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
},
|
|
||||||
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
||||||
assert.Equals(t, bucket, ordersByAccountIDTable)
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return nil, true, nil
|
|
||||||
},
|
|
||||||
},
|
|
||||||
res: []string{"o2", "o4"},
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"ok/no-pending-orders": func(t *testing.T) test {
|
|
||||||
oids := []string{"o1"}
|
|
||||||
boids, err := json.Marshal(oids)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
invalidOrder, err := newO()
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
invalidOrder.Status = StatusInvalid
|
|
||||||
binvalidOrder, err := json.Marshal(invalidOrder)
|
|
||||||
assert.FatalError(t, err)
|
|
||||||
|
|
||||||
return test{
|
|
||||||
id: "foo",
|
|
||||||
db: &db.MockNoSQLDB{
|
|
||||||
MGet: func(bucket, key []byte) ([]byte, error) {
|
|
||||||
switch string(bucket) {
|
|
||||||
case string(ordersByAccountIDTable):
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
return boids, nil
|
|
||||||
case string(orderTable):
|
|
||||||
return binvalidOrder, nil
|
|
||||||
default:
|
|
||||||
assert.FatalError(t, errors.Errorf("did not expect query to table %s", bucket))
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
},
|
|
||||||
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
|
||||||
assert.Equals(t, bucket, ordersByAccountIDTable)
|
|
||||||
assert.Equals(t, key, []byte("foo"))
|
|
||||||
assert.Equals(t, old, boids)
|
|
||||||
assert.Nil(t, newval)
|
|
||||||
return nil, true, nil
|
|
||||||
},
|
|
||||||
},
|
|
||||||
res: []string{},
|
|
||||||
}
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for name, run := range tests {
|
|
||||||
t.Run(name, func(t *testing.T) {
|
|
||||||
tc := run(t)
|
|
||||||
if oids, err := getOrderIDsByAccount(tc.db, tc.id); err != nil {
|
|
||||||
if assert.NotNil(t, tc.err) {
|
|
||||||
ae, ok := err.(*Error)
|
|
||||||
assert.True(t, ok)
|
|
||||||
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
|
||||||
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
|
||||||
assert.Equals(t, ae.Type, tc.err.Type)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if assert.Nil(t, tc.err) {
|
|
||||||
assert.Equals(t, tc.res, oids)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAccountToACME(t *testing.T) {
|
func TestAccountToACME(t *testing.T) {
|
||||||
dir := newDirectory("ca.smallstep.com", "acme")
|
dir := newDirectory("ca.smallstep.com", "acme")
|
||||||
prov := newProv()
|
prov := newProv()
|
||||||
|
|
|
@ -233,8 +233,11 @@ func (a *Authority) GetOrder(ctx context.Context, accID, orderID string) (*Order
|
||||||
|
|
||||||
// GetOrdersByAccount returns the list of order urls owned by the account.
|
// GetOrdersByAccount returns the list of order urls owned by the account.
|
||||||
func (a *Authority) GetOrdersByAccount(ctx context.Context, id string) ([]string, error) {
|
func (a *Authority) GetOrdersByAccount(ctx context.Context, id string) ([]string, error) {
|
||||||
|
ordersByAccountMux.Lock()
|
||||||
|
defer ordersByAccountMux.Unlock()
|
||||||
|
|
||||||
var oiba = orderIDsByAccount{}
|
var oiba = orderIDsByAccount{}
|
||||||
oids, err := oiba.getOrderIDsByAccount(a.db, id, false)
|
oids, err := oiba.unsafeGetOrderIDsByAccount(a.db, id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
|
@ -125,12 +125,15 @@ func newOrder(db nosql.DB, ops OrderOptions) (*order, error) {
|
||||||
|
|
||||||
type orderIDsByAccount struct{}
|
type orderIDsByAccount struct{}
|
||||||
|
|
||||||
|
// addOrderID adds an order ID to a users index of in progress order IDs.
|
||||||
|
// This method will also cull any orders that are no longer in the `pending`
|
||||||
|
// state from the index before returning it.
|
||||||
func (oiba orderIDsByAccount) addOrderID(db nosql.DB, accID string, oid string) ([]string, error) {
|
func (oiba orderIDsByAccount) addOrderID(db nosql.DB, accID string, oid string) ([]string, error) {
|
||||||
ordersByAccountMux.Lock()
|
ordersByAccountMux.Lock()
|
||||||
defer ordersByAccountMux.Unlock()
|
defer ordersByAccountMux.Unlock()
|
||||||
|
|
||||||
// Update the "order IDs by account ID" index
|
// Update the "order IDs by account ID" index
|
||||||
oids, err := oiba.getOrderIDsByAccount(db, accID, true)
|
oids, err := oiba.unsafeGetOrderIDsByAccount(db, accID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
@ -143,15 +146,9 @@ func (oiba orderIDsByAccount) addOrderID(db nosql.DB, accID string, oid string)
|
||||||
return newOids, nil
|
return newOids, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// getOrderIDsByAccount retrieves a list of Order IDs that were created by the
|
// unsafeGetOrderIDsByAccount retrieves a list of Order IDs that were created by the
|
||||||
// account.
|
// account.
|
||||||
func (oiba orderIDsByAccount) getOrderIDsByAccount(db nosql.DB, accID string, alreadyLocked bool) ([]string, error) {
|
func (oiba orderIDsByAccount) unsafeGetOrderIDsByAccount(db nosql.DB, accID string) ([]string, error) {
|
||||||
if !alreadyLocked {
|
|
||||||
ordersByAccountMux.Lock()
|
|
||||||
|
|
||||||
defer ordersByAccountMux.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
b, err := db.Get(ordersByAccountIDTable, []byte(accID))
|
b, err := db.Get(ordersByAccountIDTable, []byte(accID))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if nosql.IsErrNotFound(err) {
|
if nosql.IsErrNotFound(err) {
|
||||||
|
|
|
@ -1403,3 +1403,335 @@ func TestOrderFinalize(t *testing.T) {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Test_getOrderIDsByAccount(t *testing.T) {
|
||||||
|
type test struct {
|
||||||
|
id string
|
||||||
|
db nosql.DB
|
||||||
|
res []string
|
||||||
|
err *Error
|
||||||
|
}
|
||||||
|
tests := map[string]func(t *testing.T) test{
|
||||||
|
"ok/not-found": func(t *testing.T) test {
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
return nil, database.ErrNotFound
|
||||||
|
},
|
||||||
|
},
|
||||||
|
res: []string{},
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fail/db-error": func(t *testing.T) test {
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
return nil, errors.New("force")
|
||||||
|
},
|
||||||
|
},
|
||||||
|
err: ServerInternalErr(errors.New("error loading orderIDs for account foo: force")),
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fail/unmarshal-error": func(t *testing.T) test {
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
assert.Equals(t, bucket, ordersByAccountIDTable)
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return nil, nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
err: ServerInternalErr(errors.New("error unmarshaling orderIDs for account foo: unexpected end of JSON input")),
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fail/error-loading-order-from-order-IDs": func(t *testing.T) test {
|
||||||
|
oids := []string{"o1", "o2", "o3"}
|
||||||
|
boids, err := json.Marshal(oids)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
dbHit := 0
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
dbHit++
|
||||||
|
switch dbHit {
|
||||||
|
case 1:
|
||||||
|
assert.Equals(t, bucket, ordersByAccountIDTable)
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return boids, nil
|
||||||
|
case 2:
|
||||||
|
assert.Equals(t, bucket, orderTable)
|
||||||
|
assert.Equals(t, key, []byte("o1"))
|
||||||
|
return nil, errors.New("force")
|
||||||
|
default:
|
||||||
|
assert.FatalError(t, errors.New("should not be here"))
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
err: ServerInternalErr(errors.New("error loading order o1 for account foo: error loading order o1: force")),
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fail/error-updating-order-from-order-IDs": func(t *testing.T) test {
|
||||||
|
oids := []string{"o1", "o2", "o3"}
|
||||||
|
boids, err := json.Marshal(oids)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
o, err := newO()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
bo, err := json.Marshal(o)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
dbHit := 0
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
dbHit++
|
||||||
|
switch dbHit {
|
||||||
|
case 1:
|
||||||
|
assert.Equals(t, bucket, ordersByAccountIDTable)
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return boids, nil
|
||||||
|
case 2:
|
||||||
|
assert.Equals(t, bucket, orderTable)
|
||||||
|
assert.Equals(t, key, []byte("o1"))
|
||||||
|
return bo, nil
|
||||||
|
case 3:
|
||||||
|
assert.Equals(t, bucket, authzTable)
|
||||||
|
assert.Equals(t, key, []byte(o.Authorizations[0]))
|
||||||
|
return nil, errors.New("force")
|
||||||
|
default:
|
||||||
|
assert.FatalError(t, errors.New("should not be here"))
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
err: ServerInternalErr(errors.Errorf("error updating order o1 for account foo: error loading authz %s: force", o.Authorizations[0])),
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ok/no-change-to-pending-orders": func(t *testing.T) test {
|
||||||
|
oids := []string{"o1", "o2", "o3"}
|
||||||
|
boids, err := json.Marshal(oids)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
o, err := newO()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
bo, err := json.Marshal(o)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
az, err := newAz()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
baz, err := json.Marshal(az)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
ch, err := newDNSCh()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
bch, err := json.Marshal(ch)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
switch string(bucket) {
|
||||||
|
case string(ordersByAccountIDTable):
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return boids, nil
|
||||||
|
case string(orderTable):
|
||||||
|
return bo, nil
|
||||||
|
case string(authzTable):
|
||||||
|
return baz, nil
|
||||||
|
case string(challengeTable):
|
||||||
|
return bch, nil
|
||||||
|
default:
|
||||||
|
assert.FatalError(t, errors.Errorf("did not expect query to table %s", bucket))
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
},
|
||||||
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
||||||
|
return nil, false, errors.New("should not be attempting to store anything")
|
||||||
|
},
|
||||||
|
},
|
||||||
|
res: oids,
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fail/error-storing-new-oids": func(t *testing.T) test {
|
||||||
|
oids := []string{"o1", "o2", "o3"}
|
||||||
|
boids, err := json.Marshal(oids)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
o, err := newO()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
bo, err := json.Marshal(o)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
invalidOrder, err := newO()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
invalidOrder.Status = StatusInvalid
|
||||||
|
binvalidOrder, err := json.Marshal(invalidOrder)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
az, err := newAz()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
baz, err := json.Marshal(az)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
ch, err := newDNSCh()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
bch, err := json.Marshal(ch)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
dbGetOrder := 0
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
switch string(bucket) {
|
||||||
|
case string(ordersByAccountIDTable):
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return boids, nil
|
||||||
|
case string(orderTable):
|
||||||
|
dbGetOrder++
|
||||||
|
if dbGetOrder == 1 {
|
||||||
|
return binvalidOrder, nil
|
||||||
|
}
|
||||||
|
return bo, nil
|
||||||
|
case string(authzTable):
|
||||||
|
return baz, nil
|
||||||
|
case string(challengeTable):
|
||||||
|
return bch, nil
|
||||||
|
default:
|
||||||
|
assert.FatalError(t, errors.Errorf("did not expect query to table %s", bucket))
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
},
|
||||||
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
||||||
|
assert.Equals(t, bucket, ordersByAccountIDTable)
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return nil, false, errors.New("force")
|
||||||
|
},
|
||||||
|
},
|
||||||
|
err: ServerInternalErr(errors.New("error storing orderIDs as part of getOrderIDsByAccount logic: len(orderIDs) = 2: error storing order IDs for account foo: force")),
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ok": func(t *testing.T) test {
|
||||||
|
oids := []string{"o1", "o2", "o3", "o4"}
|
||||||
|
boids, err := json.Marshal(oids)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
o, err := newO()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
bo, err := json.Marshal(o)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
invalidOrder, err := newO()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
invalidOrder.Status = StatusInvalid
|
||||||
|
binvalidOrder, err := json.Marshal(invalidOrder)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
az, err := newAz()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
baz, err := json.Marshal(az)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
ch, err := newDNSCh()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
bch, err := json.Marshal(ch)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
dbGetOrder := 0
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
switch string(bucket) {
|
||||||
|
case string(ordersByAccountIDTable):
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return boids, nil
|
||||||
|
case string(orderTable):
|
||||||
|
dbGetOrder++
|
||||||
|
if dbGetOrder == 1 || dbGetOrder == 3 {
|
||||||
|
return binvalidOrder, nil
|
||||||
|
}
|
||||||
|
return bo, nil
|
||||||
|
case string(authzTable):
|
||||||
|
return baz, nil
|
||||||
|
case string(challengeTable):
|
||||||
|
return bch, nil
|
||||||
|
default:
|
||||||
|
assert.FatalError(t, errors.Errorf("did not expect query to table %s", bucket))
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
},
|
||||||
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
||||||
|
assert.Equals(t, bucket, ordersByAccountIDTable)
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return nil, true, nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
res: []string{"o2", "o4"},
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ok/no-pending-orders": func(t *testing.T) test {
|
||||||
|
oids := []string{"o1"}
|
||||||
|
boids, err := json.Marshal(oids)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
invalidOrder, err := newO()
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
invalidOrder.Status = StatusInvalid
|
||||||
|
binvalidOrder, err := json.Marshal(invalidOrder)
|
||||||
|
assert.FatalError(t, err)
|
||||||
|
|
||||||
|
return test{
|
||||||
|
id: "foo",
|
||||||
|
db: &db.MockNoSQLDB{
|
||||||
|
MGet: func(bucket, key []byte) ([]byte, error) {
|
||||||
|
switch string(bucket) {
|
||||||
|
case string(ordersByAccountIDTable):
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
return boids, nil
|
||||||
|
case string(orderTable):
|
||||||
|
return binvalidOrder, nil
|
||||||
|
default:
|
||||||
|
assert.FatalError(t, errors.Errorf("did not expect query to table %s", bucket))
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
},
|
||||||
|
MCmpAndSwap: func(bucket, key, old, newval []byte) ([]byte, bool, error) {
|
||||||
|
assert.Equals(t, bucket, ordersByAccountIDTable)
|
||||||
|
assert.Equals(t, key, []byte("foo"))
|
||||||
|
assert.Equals(t, old, boids)
|
||||||
|
assert.Nil(t, newval)
|
||||||
|
return nil, true, nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
res: []string{},
|
||||||
|
}
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for name, run := range tests {
|
||||||
|
t.Run(name, func(t *testing.T) {
|
||||||
|
tc := run(t)
|
||||||
|
var oiba = orderIDsByAccount{}
|
||||||
|
if oids, err := oiba.unsafeGetOrderIDsByAccount(tc.db, tc.id); err != nil {
|
||||||
|
if assert.NotNil(t, tc.err) {
|
||||||
|
ae, ok := err.(*Error)
|
||||||
|
assert.True(t, ok)
|
||||||
|
assert.HasPrefix(t, ae.Error(), tc.err.Error())
|
||||||
|
assert.Equals(t, ae.StatusCode(), tc.err.StatusCode())
|
||||||
|
assert.Equals(t, ae.Type, tc.err.Type)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if assert.Nil(t, tc.err) {
|
||||||
|
assert.Equals(t, tc.res, oids)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue