mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-03 12:26:03 +02:00
aedfbc4c71
Since when storage backend like redis can be fault in many cases, the interface should return error for the caller to handle.
77 lines
2.1 KiB
Go
77 lines
2.1 KiB
Go
package inmemory
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/protobuf/types/known/anypb"
|
|
)
|
|
|
|
func TestDB(t *testing.T) {
|
|
ctx := context.Background()
|
|
db := NewDB("example", 2)
|
|
t.Run("get missing record", func(t *testing.T) {
|
|
record, err := db.Get(ctx, "abcd")
|
|
require.Error(t, err)
|
|
assert.Nil(t, record)
|
|
})
|
|
t.Run("get record", func(t *testing.T) {
|
|
data := new(anypb.Any)
|
|
assert.NoError(t, db.Put(ctx, "abcd", data))
|
|
record, err := db.Get(ctx, "abcd")
|
|
require.NoError(t, err)
|
|
if assert.NotNil(t, record) {
|
|
assert.NotNil(t, record.CreatedAt)
|
|
assert.Equal(t, data, record.Data)
|
|
assert.Nil(t, record.DeletedAt)
|
|
assert.Equal(t, "abcd", record.Id)
|
|
assert.NotNil(t, record.ModifiedAt)
|
|
assert.Equal(t, "example", record.Type)
|
|
assert.Equal(t, "000000000001", record.Version)
|
|
}
|
|
})
|
|
t.Run("delete record", func(t *testing.T) {
|
|
assert.NoError(t, db.Delete(ctx, "abcd"))
|
|
record, err := db.Get(ctx, "abcd")
|
|
require.NoError(t, err)
|
|
if assert.NotNil(t, record) {
|
|
assert.NotNil(t, record.DeletedAt)
|
|
}
|
|
})
|
|
t.Run("clear deleted", func(t *testing.T) {
|
|
db.ClearDeleted(ctx, time.Now().Add(time.Second))
|
|
record, err := db.Get(ctx, "abcd")
|
|
require.Error(t, err)
|
|
assert.Nil(t, record)
|
|
})
|
|
t.Run("keep remaining", func(t *testing.T) {
|
|
data := new(anypb.Any)
|
|
assert.NoError(t, db.Put(ctx, "abcd", data))
|
|
assert.NoError(t, db.Delete(ctx, "abcd"))
|
|
db.ClearDeleted(ctx, time.Now().Add(-10*time.Second))
|
|
record, err := db.Get(ctx, "abcd")
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, record)
|
|
db.ClearDeleted(ctx, time.Now().Add(time.Second))
|
|
})
|
|
t.Run("list", func(t *testing.T) {
|
|
for i := 0; i < 10; i++ {
|
|
data := new(anypb.Any)
|
|
assert.NoError(t, db.Put(ctx, fmt.Sprintf("%02d", i), data))
|
|
}
|
|
|
|
records, err := db.List(ctx, "")
|
|
require.NoError(t, err)
|
|
assert.Len(t, records, 10)
|
|
records, err = db.List(ctx, "00000000000A")
|
|
require.NoError(t, err)
|
|
assert.Len(t, records, 4)
|
|
records, err = db.List(ctx, "00000000000F")
|
|
require.NoError(t, err)
|
|
assert.Len(t, records, 0)
|
|
})
|
|
}
|