diff options
| author |  tobi <31960611+tsmethurst@users.noreply.github.com> | 2022-10-01 17:48:38 +0200 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2022-10-01 17:48:38 +0200 |
| commit | 651303d97002a38c32f929a327e77d66fb439e99 (patch) | |
| tree | bad47bfce768cffb6fcb4eb59b1fe73ee75e1c8b /internal/db/bundb/domain_test.go | |
| parent | [chore] small logging cleanup (#869) (diff) | |
| download | gotosocial-651303d97002a38c32f929a327e77d66fb439e99.tar.xz | |
[chore] Normalize domain blocks to punycode (#873)
* normalize domain blocks to punycode
* normalize -> normalizeDomain
Diffstat (limited to 'internal/db/bundb/domain_test.go')
| -rw-r--r-- | internal/db/bundb/domain_test.go | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/internal/db/bundb/domain_test.go b/internal/db/bundb/domain_test.go index b326236ad..48c4a7798 100644 --- a/internal/db/bundb/domain_test.go +++ b/internal/db/bundb/domain_test.go @@ -59,6 +59,78 @@ func (suite *DomainTestSuite) TestIsDomainBlocked() { suite.True(blocked) } +func (suite *DomainTestSuite) TestIsDomainBlockedNonASCII() { + ctx := context.Background() + + now := time.Now() + + domainBlock := >smodel.DomainBlock{ + ID: "01G204214Y9TNJEBX39C7G88SW", + Domain: "xn--80aaa1bbb1h.com", + CreatedAt: now, + UpdatedAt: now, + CreatedByAccountID: suite.testAccounts["admin_account"].ID, + CreatedByAccount: suite.testAccounts["admin_account"], + } + + // no domain block exists for the given domain yet + blocked, err := suite.db.IsDomainBlocked(ctx, "какашка.com") + suite.NoError(err) + suite.False(blocked) + + blocked, err = suite.db.IsDomainBlocked(ctx, "xn--80aaa1bbb1h.com") + suite.NoError(err) + suite.False(blocked) + + err = suite.db.CreateDomainBlock(ctx, *domainBlock) + suite.NoError(err) + + // domain block now exists + blocked, err = suite.db.IsDomainBlocked(ctx, "какашка.com") + suite.NoError(err) + suite.True(blocked) + + blocked, err = suite.db.IsDomainBlocked(ctx, "xn--80aaa1bbb1h.com") + suite.NoError(err) + suite.True(blocked) +} + +func (suite *DomainTestSuite) TestIsDomainBlockedNonASCII2() { + ctx := context.Background() + + now := time.Now() + + domainBlock := >smodel.DomainBlock{ + ID: "01G204214Y9TNJEBX39C7G88SW", + Domain: "какашка.com", + CreatedAt: now, + UpdatedAt: now, + CreatedByAccountID: suite.testAccounts["admin_account"].ID, + CreatedByAccount: suite.testAccounts["admin_account"], + } + + // no domain block exists for the given domain yet + blocked, err := suite.db.IsDomainBlocked(ctx, "какашка.com") + suite.NoError(err) + suite.False(blocked) + + blocked, err = suite.db.IsDomainBlocked(ctx, "xn--80aaa1bbb1h.com") + suite.NoError(err) + suite.False(blocked) + + err = suite.db.CreateDomainBlock(ctx, *domainBlock) + suite.NoError(err) + + // domain block now exists + blocked, err = suite.db.IsDomainBlocked(ctx, "какашка.com") + suite.NoError(err) + suite.True(blocked) + + blocked, err = suite.db.IsDomainBlocked(ctx, "xn--80aaa1bbb1h.com") + suite.NoError(err) + suite.True(blocked) +} + func TestDomainTestSuite(t *testing.T) { suite.Run(t, new(DomainTestSuite)) } |