@@ -10,11 +10,13 @@ import (
10
10
organization_model "code.gitea.io/gitea/models/organization"
11
11
"code.gitea.io/gitea/models/unittest"
12
12
user_model "code.gitea.io/gitea/models/user"
13
+ "code.gitea.io/gitea/modules/setting"
13
14
15
+ "github.com/gobwas/glob"
14
16
"github.com/stretchr/testify/assert"
15
17
)
16
18
17
- func TestAddOrSetPrimaryEmailAddress (t * testing.T ) {
19
+ func TestAdminAddOrSetPrimaryEmailAddress (t * testing.T ) {
18
20
assert .NoError (t , unittest .PrepareTestDatabase ())
19
21
20
22
user := unittest .AssertExistsAndLoadBean (t , & user_model.User {ID : 27 })
@@ -28,7 +30,7 @@ func TestAddOrSetPrimaryEmailAddress(t *testing.T) {
28
30
assert .
NotEqual (
t ,
"[email protected] " ,
primary .
Email )
29
31
assert .Equal (t , user .Email , primary .Email )
30
32
31
- assert .
NoError (
t ,
AddOrSetPrimaryEmailAddress (
db .
DefaultContext ,
user ,
"[email protected] " ))
33
+ assert .
NoError (
t ,
AdminAddOrSetPrimaryEmailAddress (
db .
DefaultContext ,
user ,
"[email protected] " ))
32
34
33
35
primary , err = user_model .GetPrimaryEmailAddressOfUser (db .DefaultContext , user .ID )
34
36
assert .NoError (t , err )
@@ -39,7 +41,19 @@ func TestAddOrSetPrimaryEmailAddress(t *testing.T) {
39
41
assert .NoError (t , err )
40
42
assert .Len (t , emails , 2 )
41
43
42
- assert .
NoError (
t ,
AddOrSetPrimaryEmailAddress (
db .
DefaultContext ,
user ,
"[email protected] " ))
44
+ setting .Service .EmailDomainAllowList = []glob.Glob {glob .MustCompile ("example.org" )}
45
+ defer func () {
46
+ setting .Service .EmailDomainAllowList = []glob.Glob {}
47
+ }()
48
+
49
+ assert .
NoError (
t ,
AdminAddOrSetPrimaryEmailAddress (
db .
DefaultContext ,
user ,
"[email protected] " ))
50
+
51
+ primary , err = user_model .GetPrimaryEmailAddressOfUser (db .DefaultContext , user .ID )
52
+ assert .NoError (t , err )
53
+ assert .
Equal (
t ,
"[email protected] " ,
primary .
Email )
54
+ assert .Equal (t , user .Email , primary .Email )
55
+
56
+ assert .
NoError (
t ,
AdminAddOrSetPrimaryEmailAddress (
db .
DefaultContext ,
user ,
"[email protected] " ))
43
57
44
58
primary , err = user_model .GetPrimaryEmailAddressOfUser (db .DefaultContext , user .ID )
45
59
assert .NoError (t , err )
@@ -48,7 +62,7 @@ func TestAddOrSetPrimaryEmailAddress(t *testing.T) {
48
62
49
63
emails , err = user_model .GetEmailAddresses (db .DefaultContext , user .ID )
50
64
assert .NoError (t , err )
51
- assert .Len (t , emails , 2 )
65
+ assert .Len (t , emails , 3 )
52
66
}
53
67
54
68
func TestReplacePrimaryEmailAddress (t * testing.T ) {
0 commit comments