Function TestUpdateBalanceIdentity_NotFound

database/balance_test.go:1688–1713

(t *testing.T)

}

func TestUpdateBalanceIdentity_NotFound(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.NoError(t, err)
	defer func() { _ = db.Close() }()

	ds := Datasource{Conn: db}

	query := `
		UPDATE ledgerforge.balances
		SET identity_id = $2
		WHERE balance_id = $1
	`

	mock.ExpectExec(regexp.QuoteMeta(query)).
		WithArgs("bln_nonexistent", "idt_user_456").
		WillReturnResult(sqlmock.NewResult(0, 0))

	err = ds.UpdateBalanceIdentity("bln_nonexistent", "idt_user_456")
	assert.Error(t, err)
	apiErr, ok := err.(apierror.APIError)
	assert.True(t, ok)
	assert.Equal(t, apierror.ErrNotFound, apiErr.Code)

	err = mock.ExpectationsWereMet()
	assert.NoError(t, err)
}

func TestUpdateBalanceIdentity_QueryError(t *testing.T) {
	db, mock, err := sqlmock.New()