UmputunandGitHub
75a1edcff3
* add admin akd key mongo implementations * wire admin and key stores to mongo * lint: wrong comment on keys store * typos and comments corrections * fix failed test on avatar resizer * get rid of ineffectual assignment in test * add tests for encoding fallback and mongo's keys & admin * empty encode test adjusted
61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
package admin
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/globalsign/mgo"
|
|
"github.com/go-pkgz/mongo"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestStaticStore_Get(t *testing.T) {
|
|
var ks Store = NewStaticStore([]string{"123", "xyz"}, "aa@example.com")
|
|
|
|
a := ks.Admins("any")
|
|
assert.Equal(t, []string{"123", "xyz"}, a)
|
|
|
|
email := ks.Email("blah")
|
|
assert.Equal(t, "aa@example.com", email)
|
|
}
|
|
|
|
func TestMongoStore_Get(t *testing.T) {
|
|
conn, err := mongo.MakeTestConnection(t)
|
|
require.NoError(t, err)
|
|
var ms Store = NewMongoStore(conn)
|
|
|
|
recs := []struct {
|
|
SiteID string `bson:"site"`
|
|
IDs []string `bson:"admin_ids"`
|
|
Email string `bson:"admin_email"`
|
|
}{
|
|
{"site1", []string{"i11", "i12"}, "e1"},
|
|
{"site2", []string{"i21", "i22"}, "e2"},
|
|
}
|
|
err = conn.WithCollection(func(coll *mgo.Collection) error {
|
|
if e1 := coll.Insert(recs[0]); e1 != nil {
|
|
return e1
|
|
}
|
|
if e2 := coll.Insert(recs[1]); e2 != nil {
|
|
return e2
|
|
}
|
|
return nil
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
admins := ms.Admins("site1")
|
|
assert.Equal(t, []string{"i11", "i12"}, admins)
|
|
email := ms.Email("site1")
|
|
assert.Equal(t, "e1", email)
|
|
|
|
admins = ms.Admins("site2")
|
|
assert.Equal(t, []string{"i21", "i22"}, admins)
|
|
email = ms.Email("site2")
|
|
assert.Equal(t, "e2", email)
|
|
|
|
admins = ms.Admins("no-site-in-db")
|
|
assert.Equal(t, []string{}, admins)
|
|
email = ms.Email("no-site-in-db")
|
|
assert.Equal(t, "", email)
|
|
}
|