smallstep-certificates/acme/db/nosql/nosql_test.go

package nosql

import (
	"context"
	"testing"

	"github.com/pkg/errors"
	"github.com/smallstep/assert"
	"github.com/smallstep/certificates/db"
	"github.com/smallstep/nosql"
)

func TestNew(t *testing.T) {
	type test struct {
		db  nosql.DB
		err error
	}
	var tests = map[string]test{
		"fail/db.CreateTable-error": {
			db: &db.MockNoSQLDB{
				MCreateTable: func(bucket []byte) error {
					assert.Equals(t, string(bucket), string(accountTable))
					return errors.New("force")
				},
			},
			err: errors.Errorf("error creating table %s: force", string(accountTable)),
		},
		"ok": {
			db: &db.MockNoSQLDB{
				MCreateTable: func(bucket []byte) error {
					return nil
				},
			},
		},
	}
	for name, tc := range tests {
		t.Run(name, func(t *testing.T) {
			if _, err := New(tc.db); err != nil {
				if assert.NotNil(t, tc.err) {
					assert.HasPrefix(t, err.Error(), tc.err.Error())
				}
			} else {
				assert.Nil(t, tc.err)
			}
		})
	}
}

type errorThrower string

func (et errorThrower) MarshalJSON() ([]byte, error) {
	return nil, errors.New("force")
}

func TestDB_save(t *testing.T) {
	type test struct {
		db  nosql.DB
		nu  interface{}
		old interface{}
		err error
	}
	var tests = map[string]test{
		"fail/error-marshaling-new": {
			nu:  errorThrower("foo"),
			err: errors.New("error marshaling acme type: challenge"),
		},
		"fail/error-marshaling-old": {
			nu:  "new",
			old: errorThrower("foo"),
			err: errors.New("error marshaling acme type: challenge"),
		},
		"fail/db.CmpAndSwap-error": {
			nu:  "new",
			old: "old",
			db: &db.MockNoSQLDB{
				MCmpAndSwap: func(bucket, key, old, nu []byte) ([]byte, bool, error) {
					assert.Equals(t, bucket, challengeTable)
					assert.Equals(t, string(key), "id")
					assert.Equals(t, string(old), "\"old\"")
					assert.Equals(t, string(nu), "\"new\"")
					return nil, false, errors.New("force")
				},
			},
			err: errors.New("error saving acme challenge: force"),
		},
		"fail/db.CmpAndSwap-false-marshaling-old": {
			nu:  "new",
			old: "old",
			db: &db.MockNoSQLDB{
				MCmpAndSwap: func(bucket, key, old, nu []byte) ([]byte, bool, error) {
					assert.Equals(t, bucket, challengeTable)
					assert.Equals(t, string(key), "id")
					assert.Equals(t, string(old), "\"old\"")
					assert.Equals(t, string(nu), "\"new\"")
					return nil, false, nil
				},
			},
			err: errors.New("error saving acme challenge; changed since last read"),
		},
		"ok": {
			nu:  "new",
			old: "old",
			db: &db.MockNoSQLDB{
				MCmpAndSwap: func(bucket, key, old, nu []byte) ([]byte, bool, error) {
					assert.Equals(t, bucket, challengeTable)
					assert.Equals(t, string(key), "id")
					assert.Equals(t, string(old), "\"old\"")
					assert.Equals(t, string(nu), "\"new\"")
					return nu, true, nil
				},
			},
		},
		"ok/nils": {
			nu:  nil,
			old: nil,
			db: &db.MockNoSQLDB{
				MCmpAndSwap: func(bucket, key, old, nu []byte) ([]byte, bool, error) {
					assert.Equals(t, bucket, challengeTable)
					assert.Equals(t, string(key), "id")
					assert.Equals(t, old, nil)
					assert.Equals(t, nu, nil)
					return nu, true, nil
				},
			},
		},
	}
	for name, tc := range tests {
		t.Run(name, func(t *testing.T) {
			db := &DB{db: tc.db}
			if err := db.save(context.Background(), "id", tc.nu, tc.old, "challenge", challengeTable); err != nil {
				if assert.NotNil(t, tc.err) {
					assert.HasPrefix(t, err.Error(), tc.err.Error())
				}
			} else {
				assert.Nil(t, tc.err)
			}
		})
	}
}