logging_test.go

package duckdb

import (
	"database/sql"
	"database/sql/driver"
	"os"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
)

type logStore struct {
	store []string
}

var myLogStore logStore

func (s *logStore) Contains(entry string) bool {
	for _, e := range s.store {
		if strings.Contains(e, entry) {
			return true
		}
	}
	return false
}

func WriteLogEntry(level, logType, logMsg string) {
	myLogStore.store = append(myLogStore.store, level+", "+logType+", "+logMsg)
}

func TestLogStorage(t *testing.T) {
	defer func() {
		err := os.Remove("test_logging.db")
		require.NoError(t, err)
		err = os.Remove("test_logging.db.wal")
		require.NoError(t, err)
	}()

	c := newConnectorWrapper(t, ``, func(execer driver.ExecerContext) error {
		return nil
	})
	defer closeConnectorWrapper(t, c)

	callbacks := LoggerCallbacks{DefaultLoggerCallback: WriteLogEntry}

	err := RegisterLogStorage(c, "MyCustomStorage", callbacks)
	require.NoError(t, err)

	db := sql.OpenDB(c)
	defer closeDbWrapper(t, db)

	// Configure our new log storage.
	_, err = db.Exec("CALL enable_logging(level = 'info');")
	require.NoError(t, err)
	_, err = db.Exec("SET logging_storage = 'MyCustomStorage';")
	require.NoError(t, err)

	// ATTACH a database and silently fail a CHECKPOINT.
	_, err = db.Exec("ATTACH 'test_logging.db'")
	require.NoError(t, err)
	_, err = db.Exec("PRAGMA wal_autocheckpoint = '1TB';")
	require.NoError(t, err)
	_, err = db.Exec("PRAGMA debug_checkpoint_abort = 'before_header';")
	require.NoError(t, err)
	_, err = db.Exec("CREATE TABLE test_logging.integers AS SELECT * FROM range(100) tbl(i);")
	require.NoError(t, err)

	// Ensure that our log storage contains the logs.
	require.Len(t, myLogStore.store, 4)
	require.True(t, myLogStore.Contains("ATTACH 'test_logging.db'"))
	require.True(t, myLogStore.Contains("PRAGMA wal_autocheckpoint = '1TB';"))
	require.True(t, myLogStore.Contains("PRAGMA debug_checkpoint_abort = 'before_header';"))
	require.True(t, myLogStore.Contains("CREATE TABLE test_logging.integers AS SELECT * FROM range(100) tbl(i);"))
}