127 lines
2.9 KiB
Go
127 lines
2.9 KiB
Go
|
package authorization_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/influxdata/influxdb/v2"
|
||
|
|
"github.com/influxdata/influxdb/v2/authorization"
|
||
|
|
"github.com/influxdata/influxdb/v2/inmem"
|
||
|
|
"github.com/influxdata/influxdb/v2/kv"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAuth(t *testing.T) {
|
||
|
|
s := func() kv.Store {
|
||
|
|
return inmem.NewKVStore()
|
||
|
|
}
|
||
|
|
|
||
|
|
setup := func(t *testing.T, store *authorization.Store, tx kv.Tx) {
|
||
|
|
for i := 1; i <= 10; i++ {
|
||
|
|
err := store.CreateAuthorization(context.Background(), tx, &influxdb.Authorization{
|
||
|
|
ID: influxdb.ID(i),
|
||
|
|
Token: fmt.Sprintf("randomtoken%d", i),
|
||
|
|
OrgID: influxdb.ID(i),
|
||
|
|
UserID: influxdb.ID(i),
|
||
|
|
})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
tt := []struct {
|
||
|
|
name string
|
||
|
|
setup func(*testing.T, *authorization.Store, kv.Tx)
|
||
|
|
update func(*testing.T, *authorization.Store, kv.Tx)
|
||
|
|
results func(*testing.T, *authorization.Store, kv.Tx)
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "create",
|
||
|
|
setup: setup,
|
||
|
|
results: func(t *testing.T, store *authorization.Store, tx kv.Tx) {
|
||
|
|
auths, err := store.ListAuthorizations(context.Background(), tx, influxdb.AuthorizationFilter{})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(auths) != 10 {
|
||
|
|
t.Fatalf("expected 10 authorizations, got: %d", len(auths))
|
||
|
|
}
|
||
|
|
|
||
|
|
expected := []*influxdb.Authorization{}
|
||
|
|
for i := 1; i <= 10; i++ {
|
||
|
|
expected = append(expected, &influxdb.Authorization{
|
||
|
|
ID: influxdb.ID(i),
|
||
|
|
Token: fmt.Sprintf("randomtoken%d", i),
|
||
|
|
OrgID: influxdb.ID(i),
|
||
|
|
UserID: influxdb.ID(i),
|
||
|
|
Status: "active",
|
||
|
|
})
|
||
|
|
}
|
||
|
|
if !reflect.DeepEqual(auths, expected) {
|
||
|
|
t.Fatalf("expected identical authorizations: \n%+v\n%+v", auths, expected)
|
||
|
|
}
|
||
|
|
|
||
|
|
// should not be able to create two authorizations with identical tokens
|
||
|
|
err = store.CreateAuthorization(context.Background(), tx, &influxdb.Authorization{
|
||
|
|
ID: influxdb.ID(1),
|
||
|
|
Token: fmt.Sprintf("randomtoken%d", 1),
|
||
|
|
OrgID: influxdb.ID(1),
|
||
|
|
UserID: influxdb.ID(1),
|
||
|
|
})
|
||
|
|
if err == nil {
|
||
|
|
t.Fatalf("expected to be unable to create authorizations with identical tokens")
|
||
|
|
}
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, testScenario := range tt {
|
||
|
|
t.Run(testScenario.name, func(t *testing.T) {
|
||
|
|
ts, err := authorization.NewStore(s())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// setup
|
||
|
|
if testScenario.setup != nil {
|
||
|
|
err := ts.Update(context.Background(), func(tx kv.Tx) error {
|
||
|
|
testScenario.setup(t, ts, tx)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// update
|
||
|
|
if testScenario.update != nil {
|
||
|
|
err := ts.Update(context.Background(), func(tx kv.Tx) error {
|
||
|
|
testScenario.update(t, ts, tx)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// results
|
||
|
|
if testScenario.results != nil {
|
||
|
|
err := ts.View(context.Background(), func(tx kv.Tx) error {
|
||
|
|
testScenario.results(t, ts, tx)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|