count_test.go

package reltest

import (
	"context"
	"database/sql"
	"testing"

	"github.com/go-rel/rel"
	"github.com/stretchr/testify/assert"
)

func TestCount(t *testing.T) {
	var (
		repo = New()
	)

	repo.ExpectCount("books").Result(2)
	count, err := repo.Count(context.TODO(), "books")
	assert.Nil(t, err)
	assert.Equal(t, 2, count)
	repo.AssertExpectations(t)

	repo.ExpectCount("books").Result(2)
	assert.NotPanics(t, func() {
		count := repo.MustCount(context.TODO(), "books")
		assert.Equal(t, 2, count)
	})
	repo.AssertExpectations(t)
}

func TestCount_error(t *testing.T) {
	var (
		repo = New()
	)

	repo.ExpectCount("books").ConnectionClosed()
	count, err := repo.Count(context.TODO(), "books")
	assert.Equal(t, sql.ErrConnDone, err)
	assert.Equal(t, 0, count)
	repo.AssertExpectations(t)

	repo.ExpectCount("books").ConnectionClosed()
	assert.Panics(t, func() {
		count := repo.MustCount(context.TODO(), "books")
		assert.Equal(t, 0, count)
	})
	repo.AssertExpectations(t)
}

func TestCount_assert(t *testing.T) {
	var (
		repo = New()
	)

	repo.ExpectCount("users")

	assert.Panics(t, func() {
		repo.Count(context.TODO(), "books")
	})
	assert.False(t, repo.AssertExpectations(nt))
	assert.Equal(t, "FAIL: Mock defined but not called:\nCount(ctx, \"users\", rel.From(\"users\"))", nt.lastLog)
}

func TestCount_assert_transaction(t *testing.T) {
	var (
		repo = New()
	)

	repo.ExpectTransaction(func(repo *Repository) {
		repo.ExpectCount("users")
	})

	assert.False(t, repo.AssertExpectations(nt))
	assert.Equal(t, "FAIL: Mock defined but not called:\n<Transaction: 1> Count(ctx, \"users\", rel.From(\"users\"))", nt.lastLog)
}

func TestCount_String(t *testing.T) {
	var (
		mockCount = MockCount{assert: &Assert{}, argCollection: "users", argQuery: rel.From("users")}
	)

	assert.Equal(t, "Count(ctx, \"users\", rel.From(\"users\"))", mockCount.String())
	assert.Equal(t, "ExpectCount(\"users\", rel.From(\"users\"))", mockCount.ExpectString())
}