track_entity_test.go

package tests

import (
	"strings"
	"testing"

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

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

	{
		session := openSessionMust(t, store)
		err = session.Delete(&User{})
		assert.Error(t, err)
		_ = err.(*ravendb.IllegalStateError)
		msg := err.Error()
		assert.True(t, strings.HasSuffix(msg, "is not associated with the session, cannot delete unknown entity instance"))
		session.Close()
	}
}

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

	{
		session := openSessionMust(t, store)
		user1 := &User{}
		user1.setName("John")
		user1.ID = "users/1"

		user2 := &User{}
		user2.setName("Jonathan")
		user2.ID = "users/2"

		err = session.Store(user1)
		assert.NoError(t, err)
		err = session.Store(user2)
		assert.NoError(t, err)

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

	{
		session := openSessionMust(t, store)
		err = session.DeleteByID("users/1", "")
		assert.NoError(t, err)
		err = session.DeleteByID("users/2", "")
		assert.NoError(t, err)
		err = session.SaveChanges()
		assert.NoError(t, err)
		session.Close()
	}

	{
		session := openSessionMust(t, store)
		var user *User
		err = session.Load(&user, "users/1")
		assert.NoError(t, err)
		err = session.Load(&user, "users/2")
		assert.NoError(t, err)
		session.Close()
	}
}

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

	{
		session := openSessionMust(t, store)
		user := &User{}
		user.ID = "users/1"
		user.setName("User1")

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

		newUser := &User{}
		newUser.setName("User2")
		newUser.ID = "users/1"

		err = session.Store(newUser)
		_ = err.(*ravendb.NonUniqueObjectError)
		msg := err.Error()
		assert.True(t, strings.HasPrefix(msg, "Attempted to associate a different object with id 'users/1'"))
		session.Close()
	}
}

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

	// matches order of java tests
	trackEntityTestLoadingDeletedDocumentShouldReturnNull(t, driver)
	trackEntityTestDeletingEntityThatIsNotTrackedShouldThrow(t, driver)
	trackEntityTestStoringDocumentWithTheSameIdInTheSameSessionShouldThrow(t, driver)
}