mirror of
https://github.com/dolthub/dolt.git
synced 2026-02-20 10:09:03 -06:00
03b7221c36
stretchr has fixed a bug with the -count flag. I could merge these changes into attic-labs, but it's easier to just use strechr. We forked stretchr a long time ago so that we didn't link in the HTTP testing libraries into the noms binaries (because we were using d.Chk in production code). The HTTP issue doesn't seem to happen anymore, even though we're still using d.Chk.
123 lines
3.3 KiB
Go
123 lines
3.3 KiB
Go
// Copyright 2017 Attic Labs, Inc. All rights reserved.
|
|
// Licensed under the Apache License, version 2.0:
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
package nbs
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/attic-labs/noms/go/util/sizecache"
|
|
"github.com/aws/aws-sdk-go/service/dynamodb"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestDynamoTableReaderAt(t *testing.T) {
|
|
ddb := makeFakeDDB(t)
|
|
|
|
chunks := [][]byte{
|
|
[]byte("hello2"),
|
|
[]byte("goodbye2"),
|
|
[]byte("badbye2"),
|
|
}
|
|
|
|
tableData, h := buildTable(chunks)
|
|
ddb.putData(fmtTableName(h), tableData)
|
|
|
|
t.Run("ddbTableStore", func(t *testing.T) {
|
|
t.Run("ReadTable", func(t *testing.T) {
|
|
test := func(dts *ddbTableStore) {
|
|
assert := assert.New(t)
|
|
data, err := dts.ReadTable(h, &Stats{})
|
|
assert.NoError(err)
|
|
assert.Equal(tableData, data)
|
|
|
|
data, err = dts.ReadTable(computeAddr([]byte{}), &Stats{})
|
|
assert.Error(err)
|
|
assert.IsType(tableNotInDynamoErr{}, err)
|
|
assert.Nil(data)
|
|
}
|
|
|
|
t.Run("EventuallyConsistentSuccess", func(t *testing.T) {
|
|
test(&ddbTableStore{ddb, "table", nil, nil})
|
|
})
|
|
|
|
t.Run("EventuallyConsistentFailure", func(t *testing.T) {
|
|
test(&ddbTableStore{&eventuallyConsistentDDB{ddb}, "table", nil, nil})
|
|
})
|
|
|
|
t.Run("WithCache", func(t *testing.T) {
|
|
tc := sizecache.New(uint64(2 * len(tableData)))
|
|
dts := &ddbTableStore{ddb, "table", nil, tc}
|
|
test(dts)
|
|
|
|
// Table should have been cached on read
|
|
baseline := ddb.numGets
|
|
_, err := dts.ReadTable(h, &Stats{})
|
|
assert.NoError(t, err)
|
|
assert.Zero(t, ddb.numGets-baseline)
|
|
})
|
|
})
|
|
|
|
t.Run("WriteTable", func(t *testing.T) {
|
|
t.Run("WithoutCache", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
dts := &ddbTableStore{makeFakeDDB(t), "table", nil, nil}
|
|
assert.NoError(dts.Write(h, tableData))
|
|
|
|
data, err := dts.ReadTable(h, &Stats{})
|
|
assert.NoError(err)
|
|
assert.Equal(tableData, data)
|
|
})
|
|
|
|
t.Run("WithCache", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
tc := sizecache.New(uint64(2 * len(tableData)))
|
|
dts := &ddbTableStore{makeFakeDDB(t), "table", nil, tc}
|
|
assert.NoError(dts.Write(h, tableData))
|
|
|
|
// Table should have been cached on write
|
|
baseline := ddb.numGets
|
|
data, err := dts.ReadTable(h, &Stats{})
|
|
assert.NoError(err)
|
|
assert.Equal(tableData, data)
|
|
assert.Zero(ddb.numGets - baseline)
|
|
})
|
|
})
|
|
})
|
|
|
|
t.Run("ReadAtWithCache", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
stats := &Stats{}
|
|
|
|
tc := sizecache.New(uint64(2 * len(tableData)))
|
|
tra := &dynamoTableReaderAt{&ddbTableStore{ddb, "table", nil, tc}, h}
|
|
|
|
// First, read when table is not yet cached
|
|
scratch := make([]byte, len(tableData)/4)
|
|
baseline := ddb.numGets
|
|
_, err := tra.ReadAtWithStats(scratch, 0, stats)
|
|
assert.NoError(err)
|
|
assert.True(ddb.numGets > baseline)
|
|
|
|
// Table should have been cached on read so read again, a different slice this time
|
|
baseline = ddb.numGets
|
|
_, err = tra.ReadAtWithStats(scratch, int64(len(scratch)), stats)
|
|
assert.NoError(err)
|
|
assert.Zero(ddb.numGets - baseline)
|
|
})
|
|
}
|
|
|
|
type eventuallyConsistentDDB struct {
|
|
ddbsvc
|
|
}
|
|
|
|
func (ec *eventuallyConsistentDDB) GetItem(input *dynamodb.GetItemInput) (*dynamodb.GetItemOutput, error) {
|
|
if input.ConsistentRead != nil && *(input.ConsistentRead) {
|
|
return ec.ddbsvc.GetItem(input)
|
|
}
|
|
return &dynamodb.GetItemOutput{}, nil
|
|
}
|