dbcleaner_test.go

package dbcleaner

import (
	"errors"
	"testing"
	"time"

	"github.com/khaiql/dbcleaner/engine"
	"github.com/stretchr/testify/mock"
)

func setupMock() *engine.MockEngine {
	mockEngine := &engine.MockEngine{}
	mockEngine.On("Truncate", mock.AnythingOfType("string")).Return(nil)
	mockEngine.On("Close").Return(nil)

	return mockEngine
}

func TestClean(t *testing.T) {
	cleaner := New(SetNumberOfRetry(2), SetLockTimeout(1*time.Second), SetRetryInterval(1*time.Second))

	mockEngine := setupMock()
	cleaner.SetEngine(mockEngine)

	t.Run("TestNothingLock", func(t *testing.T) {
		cleaner.Clean("table1", "table2")
		mockEngine.AssertNumberOfCalls(t, "Truncate", 2)
		mockEngine.AssertCalled(t, "Truncate", "table1")
		mockEngine.AssertCalled(t, "Truncate", "table2")
	})

	t.Run("TestLockAndThenUnlock", func(t *testing.T) {
		tbName := "lock_table"
		cleaner.Acquire(tbName)
		cleaner.Clean(tbName)

		mockEngine.AssertCalled(t, "Truncate", tbName)
	})

	t.Run("TestClose", func(t *testing.T) {
		cleaner.Close()
		mockEngine.AssertCalled(t, "Close")
	})

	t.Run("TestTruncateError", func(t *testing.T) {
		defer func() {
			if r := recover(); r == nil {
				t.Error("The code did not panic")
			}
		}()

		errorTruncateMock := &engine.MockEngine{}
		errorTruncateMock.On("Truncate", mock.AnythingOfType("string")).Return(errors.New("Truncate error"))

		cleaner.SetEngine(errorTruncateMock)
		cleaner.Clean("error_table")
	})

	t.Run("TestCleanWithoutLock", func(t *testing.T) {
		e := setupMock()
		cleaner.SetEngine(e)
		cleaner.Acquire("table_1")
		cleaner.Acquire("table_1")

		go func() {
			cleaner.Clean("table_1")
			e.AssertNumberOfCalls(t, "Truncate", 1)
		}()

		time.Sleep(2 * time.Second)
		cleaner.Clean("table_1")
		e.AssertNumberOfCalls(t, "Truncate", 2)
	})

	t.Run("TestAcquire", func(t *testing.T) {
		t.Run("FailedToAcquireBecauseOfDeadLock", func(t *testing.T) {
			cleaner.Acquire("table_1")

			defer func() {
				if r := recover(); r == nil {
					t.Error("It should panic because of not able to acquire the lock")
				}
			}()
			other := New(SetNumberOfRetry(2), SetLockTimeout(1*time.Second), SetRetryInterval(1*time.Second))
			other.Acquire("table_1")
		})
	})
}