query_sql_and_execute_test.go

package lore

import (
	"fmt"
	"testing"

	"github.com/Masterminds/squirrel"
)

/*
TestSelectModelByPrimaryKey tests the SelectModelByPrimaryKey function.
*/
func TestSelectModelByPrimaryKey(t *testing.T) {
	// Test invalid ModelInterface instance.
	tmi := createTestModelInvalidInstance()

	// Test empty primary key resilience.
	pk := tmi.DbPrimaryFieldKey()
	if pk != "" {
		t.Error("Expected empty primary field key for invalid instance")
		return
	}
	found, err := SelectModelByPrimaryKey(tmi, nil, nil)
	if found != false || err == nil {
		t.Error("Expected not found and non-nil err for invalid instance with empty primary key")
		return
	}

	tm := createTestModelInstance()
	db, dbMock := getTestSqlxDb(t)
	dbMock.ExpectQuery(fmt.Sprintf(
		"^SELECT \\* FROM %s WHERE %s =", _TEST_DB_TABLENAME, _TEST_DB_FIELDNAME_ID,
	)).WithArgs(_TEST_MODEL_ID)
	SelectModelByPrimaryKey(tm, db, newTestModelEmpty())
	err = dbMock.ExpectationsWereMet()
	if err != nil {
		t.Error(err)
		return
	}
}

/*
TestSelectModelWhere tests the SelectModelWhere function.
*/
func TestSelectModelWhere(t *testing.T) {
	tm := createTestModelInstance()
	db, dbMock := getTestSqlxDb(t)
	dbMock.ExpectQuery(fmt.Sprintf(
		"^SELECT \\* FROM %s WHERE %s", _TEST_DB_TABLENAME, _TEST_DB_FIELDNAME_FIELD,
	)).WithArgs(_TEST_MODEL_FIELD)
	SelectModelWhere(tm, db, newSqlPart(squirrel.Eq{
		_TEST_DB_FIELDNAME_FIELD: _TEST_MODEL_FIELD,
	}), newTestModelEmpty())
	err := dbMock.ExpectationsWereMet()
	if err != nil {
		t.Error(err)
		return
	}
}

/*
TestSelectModelsWhere tests the SelectModelsWhere function.
*/
func TestSelectModelsWhere(t *testing.T) {
	tm := createTestModelInstance()
	db, dbMock := getTestSqlxDb(t)
	dbMock.ExpectQuery(fmt.Sprintf(
		"^SELECT \\* FROM %s WHERE %s.*LIMIT 3", _TEST_DB_TABLENAME, _TEST_DB_FIELDNAME_FIELD,
	)).WithArgs(_TEST_MODEL_FIELD)
	var limit uint64
	limit = 3
	SelectModelsWhere(tm, db, newSqlPart(squirrel.Eq{
		_TEST_DB_FIELDNAME_FIELD: _TEST_MODEL_FIELD,
	}), &limit, newTestModelEmptyList())
	err := dbMock.ExpectationsWereMet()
	if err != nil {
		t.Error(err)
		return
	}
}

/*
TestInsertNewModel tests the InsertNewModel function.

TODO: Find way to test args in a way that is robust to random changes in argument order injected via
Squirrel.
*/
func TestInsertNewModel(t *testing.T) {
	tm := createTestModelInstance()
	db, dbMock := getTestSqlxDb(t)
	dbMock.ExpectQuery(fmt.Sprintf(
		"^INSERT INTO %s.*RETURNING \\*", _TEST_DB_TABLENAME,
	))
	InsertNewModel(tm, db, newTestModelEmpty())
	err := dbMock.ExpectationsWereMet()
	if err != nil {
		t.Error(err)
		return
	}
}

/*
TestUpdateModelByPrimaryKey tests the UpdateModelByPrimaryKey function.

TODO: Find way to test args in a way that is robust to random changes in argument order injected via
Squirrel.
*/
func TestUpdateModelByPrimaryKey(t *testing.T) {
	tm := createTestModelInstance()
	db, dbMock := getTestSqlxDb(t)
	dbMock.ExpectQuery(fmt.Sprintf(
		"^UPDATE %s SET .* WHERE %s.*RETURNING \\*", _TEST_DB_TABLENAME, _TEST_DB_FIELDNAME_ID,
	))
	UpdateModelByPrimaryKey(tm, db, newTestModelEmpty())
	err := dbMock.ExpectationsWereMet()
	if err != nil {
		t.Error(err)
		return
	}
}

/*
TestUpdateSetMapWhere tests the UpdateSetMapWhere function.
*/
func TestUpdateSetMapWhere(t *testing.T) {
	tm := createTestModelInstance()
	db, dbMock := getTestSqlxDb(t)
	dbMock.ExpectQuery(fmt.Sprintf(
		"^UPDATE %s SET %s =.* WHERE %s =.*RETURNING \\*",
		_TEST_DB_TABLENAME,
		_TEST_DB_FIELDNAME_FIELD,
		_TEST_DB_FIELDNAME_ID,
	)).WithArgs(_TEST_MODEL_FIELD+1, _TEST_MODEL_ID)
	UpdateSetMapWhere(tm, db, map[string]interface{}{
		_TEST_DB_FIELDNAME_FIELD: _TEST_MODEL_FIELD + 1,
	}, newSqlPart(squirrel.Eq{
		_TEST_DB_FIELDNAME_ID: _TEST_MODEL_ID,
	}), newTestModelEmptyList())
	err := dbMock.ExpectationsWereMet()
	if err != nil {
		t.Error(err)
		return
	}
}

/*
TestDeleteModelByPrimaryKey tests the DeleteModelByPrimaryKey function.
*/
func TestDeleteModelByPrimaryKey(t *testing.T) {
	tm := createTestModelInstance()
	db, dbMock := getTestSqlxDb(t)
	dbMock.ExpectQuery(fmt.Sprintf(
		"^DELETE FROM %s WHERE %s =.*RETURNING \\*", _TEST_DB_TABLENAME, _TEST_DB_FIELDNAME_ID,
	)).WithArgs(_TEST_MODEL_ID)
	DeleteModelByPrimaryKey(tm, db, newTestModelEmpty())
	err := dbMock.ExpectationsWereMet()
	if err != nil {
		t.Error(err)
		return
	}
}

/*
TestDeleteModelsWhere tests the DeleteModelsWhere function.

TODO: Find way to test multiple WHERE args in a way that is robust to random changes in argument
order injected via Squirrel.
*/
func TestDeleteModelsWhere(t *testing.T) {
	tm := createTestModelInstance()
	db, dbMock := getTestSqlxDb(t)
	dbMock.ExpectQuery(fmt.Sprintf(
		"^DELETE FROM %s WHERE %s = .* RETURNING \\*",
		_TEST_DB_TABLENAME,
		_TEST_DB_FIELDNAME_FIELD,
	)).WithArgs(_TEST_MODEL_FIELD)
	DeleteModelsWhere(tm, db, newSqlPart(squirrel.Eq{
		_TEST_DB_FIELDNAME_FIELD: _TEST_MODEL_FIELD,
	}), newTestModelEmptyList())
	err := dbMock.ExpectationsWereMet()
	if err != nil {
		t.Error(err)
		return
	}
}