Skip to content

Commit 4b23e6a

Browse files
ethantkoeniglunny
authored andcommitted
Unit tests for models/action (go-gitea#619)
1 parent f4feeec commit 4b23e6a

File tree

5 files changed

+397
-19
lines changed

5 files changed

+397
-19
lines changed

models/access_test.go

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,10 @@ var accessModes = []AccessMode{
1919
func TestAccessLevel(t *testing.T) {
2020
assert.NoError(t, PrepareTestDatabase())
2121

22-
user1 := &User{ID: 2}; LoadTestFixture(t, user1)
23-
user2 := &User{ID: 4}; LoadTestFixture(t, user2)
24-
repo1 := &Repository{OwnerID: 2, IsPrivate: false}; LoadTestFixture(t, repo1)
25-
repo2 := &Repository{OwnerID: 3, IsPrivate: true}; LoadTestFixture(t, repo2)
22+
user1 := &User{ID: 2}; AssertExistsAndLoadBean(t, user1)
23+
user2 := &User{ID: 4}; AssertExistsAndLoadBean(t, user2)
24+
repo1 := &Repository{OwnerID: 2, IsPrivate: false}; AssertExistsAndLoadBean(t, repo1)
25+
repo2 := &Repository{OwnerID: 3, IsPrivate: true}; AssertExistsAndLoadBean(t, repo2)
2626

2727
level, err := AccessLevel(user1, repo1)
2828
assert.NoError(t, err)
@@ -44,10 +44,10 @@ func TestAccessLevel(t *testing.T) {
4444
func TestHasAccess(t *testing.T) {
4545
assert.NoError(t, PrepareTestDatabase())
4646

47-
user1 := &User{ID: 2}; LoadTestFixture(t, user1)
48-
user2 := &User{ID: 4}; LoadTestFixture(t, user2)
49-
repo1 := &Repository{OwnerID: 2, IsPrivate: false}; LoadTestFixture(t, repo1)
50-
repo2 := &Repository{OwnerID: 3, IsPrivate: true}; LoadTestFixture(t, repo2)
47+
user1 := &User{ID: 2}; AssertExistsAndLoadBean(t, user1)
48+
user2 := &User{ID: 4}; AssertExistsAndLoadBean(t, user2)
49+
repo1 := &Repository{OwnerID: 2, IsPrivate: false}; AssertExistsAndLoadBean(t, repo1)
50+
repo2 := &Repository{OwnerID: 3, IsPrivate: true}; AssertExistsAndLoadBean(t, repo2)
5151

5252
for _, accessMode := range accessModes {
5353
has, err := HasAccess(user1, repo1, accessMode)
@@ -71,8 +71,8 @@ func TestHasAccess(t *testing.T) {
7171
func TestUser_GetRepositoryAccesses(t *testing.T) {
7272
assert.NoError(t, PrepareTestDatabase())
7373

74-
user1 := &User{ID: 1}; LoadTestFixture(t, user1)
75-
user2 := &User{ID: 2}; LoadTestFixture(t, user2)
74+
user1 := &User{ID: 1}; AssertExistsAndLoadBean(t, user1)
75+
user2 := &User{ID: 2}; AssertExistsAndLoadBean(t, user2)
7676

7777
accesses, err := user1.GetRepositoryAccesses()
7878
assert.NoError(t, err)
@@ -82,8 +82,8 @@ func TestUser_GetRepositoryAccesses(t *testing.T) {
8282
func TestUser_GetAccessibleRepositories(t *testing.T) {
8383
assert.NoError(t, PrepareTestDatabase())
8484

85-
user1 := &User{ID: 1}; LoadTestFixture(t, user1)
86-
user2 := &User{ID: 2}; LoadTestFixture(t, user2)
85+
user1 := &User{ID: 1}; AssertExistsAndLoadBean(t, user1)
86+
user2 := &User{ID: 2}; AssertExistsAndLoadBean(t, user2)
8787

8888
repos, err := user1.GetAccessibleRepositories(0)
8989
assert.NoError(t, err)
@@ -98,7 +98,7 @@ func TestUser_GetAccessibleRepositories(t *testing.T) {
9898
func TestRepository_RecalculateAccesses(t *testing.T) {
9999
// test with organization repo
100100
assert.NoError(t, PrepareTestDatabase())
101-
repo1 := &Repository{ID: 3}; LoadTestFixture(t, repo1)
101+
repo1 := &Repository{ID: 3}; AssertExistsAndLoadBean(t, repo1)
102102
assert.NoError(t, repo1.GetOwner())
103103

104104
sess := x.NewSession()
@@ -119,7 +119,7 @@ func TestRepository_RecalculateAccesses(t *testing.T) {
119119
func TestRepository_RecalculateAccesses2(t *testing.T) {
120120
// test with non-organization repo
121121
assert.NoError(t, PrepareTestDatabase())
122-
repo1 := &Repository{ID: 4}; LoadTestFixture(t, repo1)
122+
repo1 := &Repository{ID: 4}; AssertExistsAndLoadBean(t, repo1)
123123
assert.NoError(t, repo1.GetOwner())
124124

125125
sess := x.NewSession()

0 commit comments

Comments
 (0)