58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
|
// Copyright (C) 2019 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package testidentity
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/alicebob/miniredis"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"storj.io/storj/internal/testcontext"
|
||
|
|
"storj.io/storj/pkg/identity"
|
||
|
|
"storj.io/storj/pkg/peertls/extensions"
|
||
|
|
"storj.io/storj/storage"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RevocationDBsTest runs the passed test function with each type of revocation database.
|
||
|
|
func RevocationDBsTest(t *testing.T, test func(*testing.T, extensions.RevocationDB, storage.KeyValueStore)) {
|
||
|
|
|
||
|
|
t.Run("Redis-backed revocation DB", func(t *testing.T) {
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
defer ctx.Cleanup()
|
||
|
|
|
||
|
|
redisServer, err := miniredis.Run()
|
||
|
|
require.NoError(t, err)
|
||
|
|
defer redisServer.Close()
|
||
|
|
|
||
|
|
{
|
||
|
|
// Test using redis-backed revocation DB
|
||
|
|
dbURL := "redis://" + redisServer.Addr() + "?db=0"
|
||
|
|
redisRevDB, err := identity.NewRevocationDB(dbURL)
|
||
|
|
require.NoError(t, err)
|
||
|
|
defer ctx.Check(redisRevDB.Close)
|
||
|
|
|
||
|
|
test(t, redisRevDB, redisRevDB.DB)
|
||
|
|
}
|
||
|
|
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("Bolt-backed revocation DB", func(t *testing.T) {
|
||
|
|
{
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
defer ctx.Cleanup()
|
||
|
|
|
||
|
|
// Test using bolt-backed revocation DB
|
||
|
|
revocationDBPath := ctx.File("revocations.db")
|
||
|
|
|
||
|
|
dbURL := "bolt://" + revocationDBPath
|
||
|
|
boltRevDB, err := identity.NewRevocationDB(dbURL)
|
||
|
|
require.NoError(t, err)
|
||
|
|
defer ctx.Check(boltRevDB.Close)
|
||
|
|
|
||
|
|
test(t, boltRevDB, boltRevDB.DB)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|