code.gitea.io/gitea@v1.22.3/services/org/repo_test.go (about) 1 // Copyright 2021 The Gitea Authors. All rights reserved. 2 // SPDX-License-Identifier: MIT 3 4 package org 5 6 import ( 7 "testing" 8 9 "code.gitea.io/gitea/models/db" 10 "code.gitea.io/gitea/models/organization" 11 repo_model "code.gitea.io/gitea/models/repo" 12 "code.gitea.io/gitea/models/unittest" 13 14 "github.com/stretchr/testify/assert" 15 ) 16 17 func TestTeam_AddRepository(t *testing.T) { 18 assert.NoError(t, unittest.PrepareTestDatabase()) 19 20 testSuccess := func(teamID, repoID int64) { 21 team := unittest.AssertExistsAndLoadBean(t, &organization.Team{ID: teamID}) 22 repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: repoID}) 23 assert.NoError(t, TeamAddRepository(db.DefaultContext, team, repo)) 24 unittest.AssertExistsAndLoadBean(t, &organization.TeamRepo{TeamID: teamID, RepoID: repoID}) 25 unittest.CheckConsistencyFor(t, &organization.Team{ID: teamID}, &repo_model.Repository{ID: repoID}) 26 } 27 testSuccess(2, 3) 28 testSuccess(2, 5) 29 30 team := unittest.AssertExistsAndLoadBean(t, &organization.Team{ID: 1}) 31 repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1}) 32 assert.Error(t, TeamAddRepository(db.DefaultContext, team, repo)) 33 unittest.CheckConsistencyFor(t, &organization.Team{ID: 1}, &repo_model.Repository{ID: 1}) 34 }