diff options
| author | 2022-12-01 16:06:09 +0100 | |
|---|---|---|
| committer | 2022-12-01 16:06:09 +0100 | |
| commit | cf20397f261becaf84d4d3e3f6620d1366b34131 (patch) | |
| tree | dc46e3a337b2606b01856211d40c196cb03c41d4 /internal | |
| parent | [feature] Support PKCS1 "RSA PUBLIC KEY" pem block type (#1179) (diff) | |
| download | gotosocial-cf20397f261becaf84d4d3e3f6620d1366b34131.tar.xz | |
[bugfix] Use case-insensitive selects when getting remote accounts by username/domain (#1191)v0.6.0-rc2
* [bugfix] Case-insensitive account selection
* don't lowercase cache key
Diffstat (limited to 'internal')
| -rw-r--r-- | internal/db/bundb/account.go | 11 | ||||
| -rw-r--r-- | internal/db/bundb/account_test.go | 17 | 
2 files changed, 24 insertions, 4 deletions
| diff --git a/internal/db/bundb/account.go b/internal/db/bundb/account.go index 712e0c1c7..ea0852d77 100644 --- a/internal/db/bundb/account.go +++ b/internal/db/bundb/account.go @@ -22,6 +22,7 @@ import (  	"context"  	"errors"  	"fmt" +	"strings"  	"time"  	"codeberg.org/gruf/go-cache/v3/result" @@ -108,11 +109,13 @@ func (a *accountDB) GetAccountByUsernameDomain(ctx context.Context, username str  			q := a.newAccountQ(account)  			if domain != "" { -				q = q.Where("? = ?", bun.Ident("account.username"), username) -				q = q.Where("? = ?", bun.Ident("account.domain"), domain) +				q = q. +					Where("LOWER(?) = ?", bun.Ident("account.username"), strings.ToLower(username)). +					Where("? = ?", bun.Ident("account.domain"), domain)  			} else { -				q = q.Where("? = ?", bun.Ident("account.username"), username) -				q = q.Where("? IS NULL", bun.Ident("account.domain")) +				q = q. +					Where("? = ?", bun.Ident("account.username"), strings.ToLower(username)). // usernames on our instance are always lowercase +					Where("? IS NULL", bun.Ident("account.domain"))  			}  			return q.Scan(ctx) diff --git a/internal/db/bundb/account_test.go b/internal/db/bundb/account_test.go index 50603623f..bf85f14f4 100644 --- a/internal/db/bundb/account_test.go +++ b/internal/db/bundb/account_test.go @@ -22,6 +22,7 @@ import (  	"context"  	"crypto/rand"  	"crypto/rsa" +	"strings"  	"testing"  	"time" @@ -84,6 +85,22 @@ func (suite *AccountTestSuite) TestGetAccountByUsernameDomain() {  	suite.NotNil(account2)  } +func (suite *AccountTestSuite) TestGetAccountByUsernameDomainMixedCase() { +	testAccount := suite.testAccounts["remote_account_2"] + +	account1, err := suite.db.GetAccountByUsernameDomain(context.Background(), testAccount.Username, testAccount.Domain) +	suite.NoError(err) +	suite.NotNil(account1) + +	account2, err := suite.db.GetAccountByUsernameDomain(context.Background(), strings.ToUpper(testAccount.Username), testAccount.Domain) +	suite.NoError(err) +	suite.NotNil(account2) + +	account3, err := suite.db.GetAccountByUsernameDomain(context.Background(), strings.ToLower(testAccount.Username), testAccount.Domain) +	suite.NoError(err) +	suite.NotNil(account3) +} +  func (suite *AccountTestSuite) TestUpdateAccount() {  	ctx := context.Background() | 
