github.com/altipla-consulting/ravendb-go-client@v0.1.3/tests/hi_lo_test.go

package tests

import (
	"testing"

	ravendb "github.com/altipla-consulting/ravendb-go-client"
	"github.com/stretchr/testify/assert"
)

type HiLoDoc struct {
	Max int64 `json:"Max"`
}

type Product struct {
	ProductName string `json:"ProductName"`
}

func hiloTestCapacityShouldDouble(t *testing.T, driver *RavenTestDriver) {
	var err error
	store := driver.getDocumentStoreMust(t)
	defer store.Close()

	hiLoIdGenerator := ravendb.NewHiLoIDGenerator("users", store, store.GetDatabase(), store.GetConventions().GetIdentityPartsSeparator())

	{
		session := openSessionMust(t, store)
		hiloDoc := &HiLoDoc{
			Max: 64,
		}

		err = session.StoreWithID(hiloDoc, "Raven/Hilo/users")
		assert.NoError(t, err)
		err = session.SaveChanges()
		assert.NoError(t, err)

		for i := 0; i < 32; i++ {
			hiLoIdGenerator.GenerateDocumentID(&User{})
		}
		session.Close()
	}

	{
		session := openSessionMust(t, store)

		var hiloDoc *HiLoDoc
		err = session.Load(&hiloDoc, "Raven/Hilo/users")
		assert.NoError(t, err)
		max := hiloDoc.Max
		assert.Equal(t, max, int64(96))

		//we should be receiving a range of 64 now
		hiLoIdGenerator.GenerateDocumentID(&User{})
		session.Close()
	}

	{
		session := openSessionMust(t, store)

		var hiloDoc *HiLoDoc
		err = session.Load(&hiloDoc, "Raven/Hilo/users")
		assert.NoError(t, err)
		max := hiloDoc.Max

		// TODO: in Java it's 160. On Travis CI (linux) it's 160
		// On my mac, it's 128.
		// It's strange because the requests sent for
		// /databases/test_db_1/hilo/next are exactly the
		// same but in Java case the server sends back "High": 160
		// and in Go case it's "High": 128
		// Maybe it's KeepAlive difference?
		valid := (max == 96+64) || (max == 96+32)
		assert.True(t, valid)
		session.Close()
	}
}

func hiloTestReturnUnusedRangeOnClose(t *testing.T, driver *RavenTestDriver) {
	var err error
	store := driver.getDocumentStoreMust(t)
	defer store.Close()

	urls := store.GetUrls()
	newStore := ravendb.NewDocumentStore(urls, "")
	newStore.SetDatabase(store.GetDatabase())

	err = newStore.Initialize()
	assert.NoError(t, err)

	{
		session := openSessionMust(t, newStore)
		assert.NoError(t, err)
		assert.NotNil(t, session)

		hiloDoc := &HiLoDoc{
			Max: 32,
		}
		err = session.StoreWithID(hiloDoc, "Raven/Hilo/users")
		assert.NoError(t, err)
		err = session.SaveChanges()
		assert.NoError(t, err)

		err = session.Store(&User{})
		assert.NoError(t, err)
		err = session.Store(&User{})
		assert.NoError(t, err)
		err = session.SaveChanges()
		assert.NoError(t, err)
		session.Close()
	}

	newStore.Close() //on document Store close, hilo-return should be called

	newStore = ravendb.NewDocumentStore(nil, store.GetDatabase())
	newStore.SetUrls(store.GetUrls())

	err = newStore.Initialize()
	assert.NoError(t, err)

	{
		session := openSessionMust(t, newStore)
		assert.NoError(t, err)
		assert.NotNil(t, session)

		var hiloDoc *HiLoDoc
		err = session.Load(&hiloDoc, "Raven/Hilo/users")
		assert.NoError(t, err)
		max := hiloDoc.Max
		assert.Equal(t, max, int64(34))
		session.Close()
	}

	newStore.Close() //on document Store close, hilo-return should be called
}

func hiloTestCanNotGoDown(t *testing.T, driver *RavenTestDriver) {
	var err error
	store := driver.getDocumentStoreMust(t)
	defer store.Close()

	session := openSessionMust(t, store)

	hiloDoc := &HiLoDoc{
		Max: 32,
	}

	err = session.StoreWithID(hiloDoc, "Raven/Hilo/users")
	assert.NoError(t, err)
	err = session.SaveChanges()
	assert.NoError(t, err)
	assert.Nil(t, err)

	hiLoKeyGenerator := ravendb.NewHiLoIDGenerator("users", store, store.GetDatabase(), store.GetConventions().GetIdentityPartsSeparator())

	nextID, err := hiLoKeyGenerator.NextID()
	assert.Nil(t, err)
	ids := []int64{nextID}

	hiloDoc.Max = 12
	session.StoreWithChangeVectorAndID(hiloDoc, "", "Raven/Hilo/users")
	err = session.SaveChanges()
	assert.Nil(t, err)

	for i := 0; i < 128; i++ {
		nextID, err = hiLoKeyGenerator.NextID()
		assert.NoError(t, err)
		contains := int64ArrayContains(ids, nextID)
		assert.False(t, contains)
		ids = append(ids, nextID)
	}
	assert.False(t, int64ArrayHasDuplicates(ids))
	session.Close()
}

func int64ArrayContains(a []int64, n int64) bool {
	for _, el := range a {
		if el == n {
			return true
		}
	}
	return false
}

func hiloTestMultiDb(t *testing.T, driver *RavenTestDriver) {
	var err error
	store := driver.getDocumentStoreMust(t)
	defer store.Close()

	session := openSessionMust(t, store)

	hiloDoc := &HiLoDoc{
		Max: 64,
	}
	err = session.StoreWithID(hiloDoc, "Raven/Hilo/users")
	assert.NoError(t, err)

	productsHilo := &HiLoDoc{
		Max: 128,
	}
	err = session.StoreWithID(productsHilo, "Raven/Hilo/products")
	assert.NoError(t, err)

	err = session.SaveChanges()
	assert.NoError(t, err)

	multiDbHilo := ravendb.NewMultiDatabaseHiLoIDGenerator(store, store.GetConventions())
	generateDocumentKey, err := multiDbHilo.GenerateDocumentID("", &User{})
	assert.NoError(t, err)
	assert.Equal(t, generateDocumentKey, "users/65-A")
	generateDocumentKey, err = multiDbHilo.GenerateDocumentID("", &Product{})
	assert.NoError(t, err)
	assert.Equal(t, generateDocumentKey, "products/129-A")
	session.Close()
}

func hiloTestDoesNotGetAnotherRangeWhenDoingParallelRequests(t *testing.T, driver *RavenTestDriver) {
	// Note: not applicable to Go as we doesn't have Executor to limit concurrency
}

func TestHiLo(t *testing.T) {
	driver := createTestDriver(t)
	destroy := func() { destroyDriver(t, driver) }
	defer recoverTest(t, destroy)

	// matches order of java tests
	hiloTestCapacityShouldDouble(t, driver)
	hiloTestReturnUnusedRangeOnClose(t, driver)
	hiloTestCanNotGoDown(t, driver)
	hiloTestMultiDb(t, driver)

	hiloTestDoesNotGetAnotherRangeWhenDoingParallelRequests(t, driver)
}