mirror of
https://github.com/dolthub/dolt.git
synced 2026-02-05 02:59:44 -06:00
133 lines
3.5 KiB
Go
133 lines
3.5 KiB
Go
package datas
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/attic-labs/noms/Godeps/_workspace/src/github.com/stretchr/testify/assert"
|
|
"github.com/attic-labs/noms/chunks"
|
|
"github.com/attic-labs/noms/types"
|
|
)
|
|
|
|
func TestDataStoreAccess(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
ds := NewDataStore(cs)
|
|
input := "abc"
|
|
|
|
c := chunks.NewChunk([]byte(input))
|
|
c1 := ds.Get(c.Ref())
|
|
assert.True(c1.IsEmpty())
|
|
|
|
has := ds.Has(c.Ref())
|
|
assert.False(has)
|
|
|
|
ds.Put(c)
|
|
c1 = ds.Get(c.Ref())
|
|
assert.False(c1.IsEmpty())
|
|
|
|
has = ds.Has(c.Ref())
|
|
assert.True(has)
|
|
}
|
|
|
|
func TestDataStoreCommit(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
ds := NewDataStore(cs)
|
|
datasetID := "ds1"
|
|
|
|
datasets := ds.Datasets()
|
|
assert.Zero(datasets.Len())
|
|
|
|
// |a|
|
|
a := types.NewString("a")
|
|
aCommit := NewCommit(cs).SetValue(a)
|
|
ds2, err := ds.Commit(datasetID, aCommit)
|
|
assert.NoError(err)
|
|
|
|
// The old datastore still still has no head.
|
|
_, ok := ds.MaybeHead(datasetID)
|
|
assert.False(ok)
|
|
|
|
// The new datastore has |a|.
|
|
aCommit1 := ds2.Head(datasetID)
|
|
assert.True(aCommit1.Value().Equals(a))
|
|
ds = ds2
|
|
|
|
// |a| <- |b|
|
|
b := types.NewString("b")
|
|
bCommit := NewCommit(cs).SetValue(b).SetParents(NewSetOfRefOfCommit(cs).Insert(NewRefOfCommit(aCommit.Ref())))
|
|
ds, err = ds.Commit(datasetID, bCommit)
|
|
assert.NoError(err)
|
|
assert.True(ds.Head(datasetID).Value().Equals(b))
|
|
|
|
// |a| <- |b|
|
|
// \----|c|
|
|
// Should be disallowed.
|
|
c := types.NewString("c")
|
|
cCommit := NewCommit(cs).SetValue(c)
|
|
ds, err = ds.Commit(datasetID, cCommit)
|
|
assert.Error(err)
|
|
assert.True(ds.Head(datasetID).Value().Equals(b))
|
|
|
|
// |a| <- |b| <- |d|
|
|
d := types.NewString("d")
|
|
dCommit := NewCommit(cs).SetValue(d).SetParents(NewSetOfRefOfCommit(cs).Insert(NewRefOfCommit(bCommit.Ref())))
|
|
ds, err = ds.Commit(datasetID, dCommit)
|
|
assert.NoError(err)
|
|
assert.True(ds.Head(datasetID).Value().Equals(d))
|
|
|
|
// Attempt to recommit |b| with |a| as parent.
|
|
// Should be disallowed.
|
|
ds, err = ds.Commit(datasetID, bCommit)
|
|
assert.Error(err)
|
|
assert.True(ds.Head(datasetID).Value().Equals(d))
|
|
|
|
// Add a commit to a different datasetId
|
|
_, err = ds.Commit("otherDs", aCommit)
|
|
assert.NoError(err)
|
|
|
|
// Get a fresh datastore, and verify that both datasets are present
|
|
newDs := NewDataStore(cs)
|
|
datasets2 := newDs.Datasets()
|
|
assert.Equal(uint64(2), datasets2.Len())
|
|
}
|
|
|
|
func TestDataStoreConcurrency(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
cs := chunks.NewMemoryStore()
|
|
ds := NewDataStore(cs)
|
|
datasetID := "ds1"
|
|
|
|
// Setup:
|
|
// |a| <- |b|
|
|
a := types.NewString("a")
|
|
aCommit := NewCommit(cs).SetValue(a)
|
|
ds, err := ds.Commit(datasetID, aCommit)
|
|
b := types.NewString("b")
|
|
bCommit := NewCommit(cs).SetValue(b).SetParents(NewSetOfRefOfCommit(cs).Insert(NewRefOfCommit(aCommit.Ref())))
|
|
ds, err = ds.Commit(datasetID, bCommit)
|
|
assert.NoError(err)
|
|
assert.True(ds.Head(datasetID).Value().Equals(b))
|
|
|
|
// Important to create this here.
|
|
ds2 := NewDataStore(cs)
|
|
|
|
// Change 1:
|
|
// |a| <- |b| <- |c|
|
|
c := types.NewString("c")
|
|
cCommit := NewCommit(cs).SetValue(c).SetParents(NewSetOfRefOfCommit(cs).Insert(NewRefOfCommit(bCommit.Ref())))
|
|
ds, err = ds.Commit(datasetID, cCommit)
|
|
assert.NoError(err)
|
|
assert.True(ds.Head(datasetID).Value().Equals(c))
|
|
|
|
// Change 2:
|
|
// |a| <- |b| <- |e|
|
|
// Should be disallowed, DataStore returned by Commit() should have |c| as Head.
|
|
e := types.NewString("e")
|
|
eCommit := NewCommit(cs).SetValue(e).SetParents(NewSetOfRefOfCommit(cs).Insert(NewRefOfCommit(bCommit.Ref())))
|
|
ds2, err = ds2.Commit(datasetID, eCommit)
|
|
assert.Error(err)
|
|
assert.True(ds.Head(datasetID).Value().Equals(c))
|
|
}
|