gocryptotrader/engine/database_connection_test.go

package engine

import (
	"log"
	"sync"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/thrasher-corp/gocryptotrader/database"
	"github.com/thrasher-corp/gocryptotrader/database/drivers"
)

func CreateDatabase(t *testing.T) {
	t.Helper()
	// fun workarounds to globals ruining testing
	database.DB.DataPath = t.TempDir()

	t.Cleanup(func() {
		if database.DB.IsConnected() {
			err := database.DB.CloseConnection()
			if err != nil {
				log.Fatal(err)
			}
		}
	})
}

func TestSetupDatabaseConnectionManager(t *testing.T) {
	_, err := SetupDatabaseConnectionManager(nil)
	assert.ErrorIs(t, err, errNilConfig)

	m, err := SetupDatabaseConnectionManager(&database.Config{})
	assert.NoError(t, err)

	if m == nil {
		t.Error("expected manager")
	}
}

func TestStartSQLite(t *testing.T) {
	CreateDatabase(t)
	m, err := SetupDatabaseConnectionManager(&database.Config{})
	assert.NoError(t, err)

	var wg sync.WaitGroup
	err = m.Start(&wg)
	assert.ErrorIs(t, err, database.ErrDatabaseSupportDisabled)

	m, err = SetupDatabaseConnectionManager(&database.Config{Enabled: true})
	assert.NoError(t, err)

	err = m.Start(&wg)
	assert.ErrorIs(t, err, database.ErrNoDatabaseProvided)

	m.cfg = database.Config{Driver: database.DBSQLite}
	err = m.Start(&wg)
	assert.ErrorIs(t, err, database.ErrDatabaseSupportDisabled)

	_, err = SetupDatabaseConnectionManager(&database.Config{
		Enabled: true,
		Driver:  database.DBSQLite,
		ConnectionDetails: drivers.ConnectionDetails{
			Host:     "localhost",
			Database: "test.db",
		},
	})
	assert.NoError(t, err)
}

// This test does not care for a successful connection
func TestStartPostgres(t *testing.T) {
	m, err := SetupDatabaseConnectionManager(&database.Config{})
	assert.NoError(t, err)

	var wg sync.WaitGroup
	err = m.Start(&wg)
	assert.ErrorIs(t, err, database.ErrDatabaseSupportDisabled)

	m.cfg.Enabled = true
	err = m.Start(&wg)
	assert.ErrorIs(t, err, database.ErrNoDatabaseProvided)

	m.cfg.Driver = database.DBPostgreSQL
	err = m.Start(&wg)
	assert.ErrorIs(t, err, database.ErrFailedToConnect)
}

func TestDatabaseConnectionManagerIsRunning(t *testing.T) {
	CreateDatabase(t)
	m, err := SetupDatabaseConnectionManager(&database.Config{
		Enabled: true,
		Driver:  database.DBSQLite,
		ConnectionDetails: drivers.ConnectionDetails{
			Host:     "localhost",
			Database: "test.db",
		},
	})
	assert.NoError(t, err)

	if m.IsRunning() {
		t.Error("expected false")
	}
	var wg sync.WaitGroup
	err = m.Start(&wg)
	assert.NoError(t, err)

	if !m.IsRunning() {
		t.Error("expected true")
	}
	m = nil
	if m.IsRunning() {
		t.Error("expected false")
	}
}

func TestDatabaseConnectionManagerStop(t *testing.T) {
	CreateDatabase(t)
	m, err := SetupDatabaseConnectionManager(&database.Config{
		Enabled: true,
		Driver:  database.DBSQLite,
		ConnectionDetails: drivers.ConnectionDetails{
			Host:     "localhost",
			Database: "test.db",
		},
	})
	assert.NoError(t, err)

	err = m.Stop()
	assert.ErrorIs(t, err, ErrSubSystemNotStarted)

	var wg sync.WaitGroup
	err = m.Start(&wg)
	assert.NoError(t, err)

	err = m.Stop()
	assert.NoError(t, err)

	m = nil
	err = m.Stop()
	assert.ErrorIs(t, err, ErrNilSubsystem)
}

func TestCheckConnection(t *testing.T) {
	CreateDatabase(t)
	var m *DatabaseConnectionManager
	err := m.checkConnection()
	assert.ErrorIs(t, err, ErrNilSubsystem)

	m, err = SetupDatabaseConnectionManager(&database.Config{
		Enabled: true,
		Driver:  database.DBSQLite,
		ConnectionDetails: drivers.ConnectionDetails{
			Host:     "localhost",
			Database: "test.db",
		},
	})
	assert.NoError(t, err)

	err = m.checkConnection()
	assert.ErrorIs(t, err, ErrSubSystemNotStarted)

	var wg sync.WaitGroup
	err = m.Start(&wg)
	assert.NoError(t, err)

	err = m.checkConnection()
	assert.NoError(t, err)

	err = m.Stop()
	assert.NoError(t, err)

	err = m.checkConnection()
	assert.ErrorIs(t, err, ErrSubSystemNotStarted)

	err = m.Start(&wg)
	assert.NoError(t, err)

	err = m.checkConnection()
	assert.NoError(t, err)

	m.dbConn.SetConnected(false)
	err = m.checkConnection()
	assert.ErrorIs(t, err, database.ErrDatabaseNotConnected)

	err = m.Stop()
	assert.NoError(t, err)
}

func TestGetInstance(t *testing.T) {
	CreateDatabase(t)
	m, err := SetupDatabaseConnectionManager(&database.Config{
		Enabled: true,
		Driver:  database.DBSQLite,
		ConnectionDetails: drivers.ConnectionDetails{
			Host:     "localhost",
			Database: "test.db",
		},
	})
	assert.NoError(t, err)

	db := m.GetInstance()
	if db != nil {
		t.Error("expected nil")
	}
	var wg sync.WaitGroup
	err = m.Start(&wg)
	assert.NoError(t, err)

	db = m.GetInstance()
	if db == nil {
		t.Error("expected not nil")
	}

	m = nil
	db = m.GetInstance()
	if db != nil {
		t.Error("expected nil")
	}
}