updater_test.go

package wgm_test

import (
	"github.com/oeasenet/wgm"
	"github.com/stretchr/testify/require"
	"testing"
)

func TestGetUpdater(t *testing.T) {
	doc := Doc{}
	u := wgm.MustUpdater(doc)
	require.NotNil(t, u)

}

func TestUpdater_Find(t *testing.T) {
	DBdoc := NewDoc("Alice", 12)
	SetupDefaultConnection()
	objectID := InsertDoc(DBdoc)
	defer DelDoc(DBdoc)

	doc := &Doc{}
	doc.Id = objectID
	u, hasResult := wgm.MustUpdater(doc).Find()

	require.True(t, hasResult)
	require.NotNil(t, u)
	require.Equal(t, doc.Age, DBdoc.Age)
	require.Equal(t, doc.Name, DBdoc.Name)
}

func TestUpdater_Update(t *testing.T) {
	DBdoc := NewDoc("Alice", 12)
	SetupDefaultConnection()
	objectID := InsertDoc(DBdoc)
	defer DelDoc(DBdoc)

	doc := &Doc{}
	doc.Id = objectID
	u, _ := wgm.MustUpdater(doc).Find()
	doc.Name = "dada"
	doc.Age = 99
	err := u.Update()
	require.Nil(t, err)

	DBdoc = &Doc{}
	wgm.FindOne(DBdoc, nil)
	require.Equal(t, DBdoc.Age, 99)
	require.Equal(t, DBdoc.Name, "dada")
}

func BenchmarkUpdater_Find(b *testing.B) {
	DBdoc := NewDoc("Alice", 12)
	SetupDefaultConnection()
	objectID := InsertDoc(DBdoc)
	defer DelDoc(DBdoc)

	doc := &Doc{}
	doc.Id = objectID
	u, hasResult := wgm.MustUpdater(doc).Find()

	require.True(b, hasResult)
	require.NotNil(b, u)
	require.Equal(b, doc.Age, DBdoc.Age)
	require.Equal(b, doc.Name, DBdoc.Name)
}

func BenchmarkUpdater_Update(b *testing.B) {
	DBdoc := NewDoc("Alice", 12)
	SetupDefaultConnection()
	objectID := InsertDoc(DBdoc)
	defer DelDoc(DBdoc)

	doc := &Doc{}
	doc.Id = objectID
	u, _ := wgm.MustUpdater(doc).Find()
	doc.Name = "dada"
	doc.Age = 99
	err := u.Update()
	require.Nil(b, err)

	DBdoc = &Doc{}
	wgm.FindOne(DBdoc, nil)
	require.Equal(b, DBdoc.Age, 99)
	require.Equal(b, DBdoc.Name, "dada")
}