organizations_test.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687
  1. // Copyright 2022 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package db
  5. import (
  6. "context"
  7. "os"
  8. "path/filepath"
  9. "testing"
  10. "time"
  11. "github.com/stretchr/testify/assert"
  12. "github.com/stretchr/testify/require"
  13. "gogs.io/gogs/internal/conf"
  14. "gogs.io/gogs/internal/dbtest"
  15. "gogs.io/gogs/internal/errutil"
  16. )
  17. func TestOrganizations(t *testing.T) {
  18. if testing.Short() {
  19. t.Skip()
  20. }
  21. t.Parallel()
  22. ctx := context.Background()
  23. tables := []any{
  24. new(User), new(EmailAddress), new(OrgUser), new(Team), new(TeamUser), new(Repository), new(Watch), new(Star),
  25. new(Follow), new(Issue), new(PublicKey), new(AccessToken), new(Collaboration), new(Access), new(Action),
  26. new(IssueUser), new(TeamRepo),
  27. }
  28. db := &organizations{
  29. DB: dbtest.NewDB(t, "orgs", tables...),
  30. }
  31. for _, tc := range []struct {
  32. name string
  33. test func(t *testing.T, ctx context.Context, db *organizations)
  34. }{
  35. {"Create", orgsCreate},
  36. {"GetByName", orgsGetByName},
  37. {"SearchByName", orgsSearchByName},
  38. {"List", orgsList},
  39. {"CountByUser", orgsCountByUser},
  40. {"Count", orgsCount},
  41. {"DeleteByID", orgsDeleteByID},
  42. {"AddMember", orgsAddMember},
  43. {"RemoveMember", orgsRemoveMember},
  44. {"HasMember", orgsHasMember},
  45. {"ListMembers", orgsListMembers},
  46. {"IsOwnedBy", orgsIsOwnedBy},
  47. {"SetMemberVisibility", orgsSetMemberVisibility},
  48. {"GetTeamByName", orgsGetTeamByName},
  49. {"AccessibleRepositoriesByUser", orgsAccessibleRepositoriesByUser},
  50. {"MirrorRepositoriesByUser", orgsMirrorRepositoriesByUser},
  51. } {
  52. t.Run(tc.name, func(t *testing.T) {
  53. t.Cleanup(func() {
  54. err := clearTables(t, db.DB, tables...)
  55. require.NoError(t, err)
  56. })
  57. tc.test(t, ctx, db)
  58. })
  59. if t.Failed() {
  60. break
  61. }
  62. }
  63. }
  64. func orgsCreate(t *testing.T, ctx context.Context, db *organizations) {
  65. usersStore := NewUsersStore(db.DB)
  66. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  67. require.NoError(t, err)
  68. t.Run("name not allowed", func(t *testing.T) {
  69. _, err := db.Create(ctx, "-", alice.ID, CreateOrganizationOptions{})
  70. wantErr := ErrNameNotAllowed{
  71. args: errutil.Args{
  72. "reason": "reserved",
  73. "name": "-",
  74. },
  75. }
  76. assert.Equal(t, wantErr, err)
  77. })
  78. // Users and organizations share the same namespace for names.
  79. t.Run("name already exists", func(t *testing.T) {
  80. _, err := db.Create(ctx, alice.Name, alice.ID, CreateOrganizationOptions{})
  81. wantErr := ErrOrganizationAlreadyExist{
  82. args: errutil.Args{
  83. "name": alice.Name,
  84. },
  85. }
  86. assert.Equal(t, wantErr, err)
  87. })
  88. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsCreate-tempPictureAvatarUploadPath")
  89. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  90. org, err := db.Create(
  91. ctx,
  92. "acme",
  93. alice.ID,
  94. CreateOrganizationOptions{
  95. FullName: "Acme Corp",
  96. Email: "admin@acme.com",
  97. Location: "Earth",
  98. Website: "acme.com",
  99. Description: "A popcorn company",
  100. },
  101. )
  102. require.NoError(t, err)
  103. got, err := db.GetByName(ctx, org.Name)
  104. require.NoError(t, err)
  105. assert.Equal(t, org.Name, got.Name)
  106. assert.Equal(t, org.FullName, got.FullName)
  107. assert.Equal(t, org.Email, got.Email)
  108. assert.Equal(t, org.Location, got.Location)
  109. assert.Equal(t, org.Website, got.Website)
  110. assert.Equal(t, org.Description, got.Description)
  111. assert.Equal(t, -1, got.MaxRepoCreation)
  112. assert.Equal(t, 1, got.NumTeams)
  113. assert.Equal(t, 1, got.NumMembers)
  114. assert.Equal(t, db.NowFunc().Format(time.RFC3339), got.Created.UTC().Format(time.RFC3339))
  115. assert.Equal(t, db.NowFunc().Format(time.RFC3339), got.Updated.UTC().Format(time.RFC3339))
  116. }
  117. func orgsGetByName(t *testing.T, ctx context.Context, db *organizations) {
  118. t.Run("correct user type", func(t *testing.T) {
  119. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "usersGetByUsername-tempPictureAvatarUploadPath")
  120. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  121. org1, err := db.Create(ctx, "org1", 1, CreateOrganizationOptions{})
  122. require.NoError(t, err)
  123. got, err := db.GetByName(ctx, org1.Name)
  124. require.NoError(t, err)
  125. assert.Equal(t, org1.Name, got.Name)
  126. _, err = db.GetByName(ctx, "bad_name")
  127. wantErr := ErrOrganizationNotExist{args: errutil.Args{"name": "bad_name"}}
  128. assert.Equal(t, wantErr, err)
  129. })
  130. t.Run("wrong user type", func(t *testing.T) {
  131. alice, err := NewUsersStore(db.DB).Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  132. require.NoError(t, err)
  133. _, err = db.GetByName(ctx, alice.Name)
  134. wantErr := ErrOrganizationNotExist{args: errutil.Args{"name": alice.Name}}
  135. assert.Equal(t, wantErr, err)
  136. })
  137. }
  138. func orgsList(t *testing.T, ctx context.Context, db *organizations) {
  139. usersStore := NewUsersStore(db.DB)
  140. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  141. require.NoError(t, err)
  142. bob, err := usersStore.Create(ctx, "bob", "bob@example.com", CreateUserOptions{})
  143. require.NoError(t, err)
  144. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsList-tempPictureAvatarUploadPath")
  145. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  146. org1, err := db.Create(ctx, "org1-alice-owned", alice.ID, CreateOrganizationOptions{})
  147. require.NoError(t, err)
  148. org2, err := db.Create(ctx, "org2-alice-owned", alice.ID, CreateOrganizationOptions{})
  149. require.NoError(t, err)
  150. err = db.SetMemberVisibility(ctx, org2.ID, alice.ID, true)
  151. require.NoError(t, err)
  152. err = db.AddMember(ctx, org2.ID, bob.ID)
  153. require.NoError(t, err)
  154. org3, err := db.Create(ctx, "org3-bob-owned", bob.ID, CreateOrganizationOptions{})
  155. require.NoError(t, err)
  156. tests := []struct {
  157. name string
  158. opts ListOrganizationsOptions
  159. wantOrgNames []string
  160. }{
  161. {
  162. name: "only public memberships for a user",
  163. opts: ListOrganizationsOptions{
  164. MemberID: alice.ID,
  165. IncludePrivateMembers: false,
  166. },
  167. wantOrgNames: []string{org2.Name},
  168. },
  169. {
  170. name: "all memberships for a user",
  171. opts: ListOrganizationsOptions{
  172. MemberID: alice.ID,
  173. IncludePrivateMembers: true,
  174. },
  175. wantOrgNames: []string{org1.Name, org2.Name},
  176. },
  177. {
  178. name: "only public ownership for a user",
  179. opts: ListOrganizationsOptions{
  180. OwnerID: alice.ID,
  181. IncludePrivateMembers: false,
  182. },
  183. wantOrgNames: []string{org2.Name},
  184. },
  185. {
  186. name: "all ownership for a user",
  187. opts: ListOrganizationsOptions{
  188. OwnerID: alice.ID,
  189. IncludePrivateMembers: true,
  190. },
  191. wantOrgNames: []string{org1.Name, org2.Name},
  192. },
  193. {
  194. name: "no membership for a non-existent user",
  195. opts: ListOrganizationsOptions{
  196. MemberID: 404,
  197. IncludePrivateMembers: true,
  198. },
  199. wantOrgNames: []string{},
  200. },
  201. }
  202. for _, test := range tests {
  203. t.Run(test.name, func(t *testing.T) {
  204. got, err := db.List(ctx, test.opts)
  205. require.NoError(t, err)
  206. gotOrgNames := make([]string, len(got))
  207. for i := range got {
  208. gotOrgNames[i] = got[i].Name
  209. }
  210. assert.Equal(t, test.wantOrgNames, gotOrgNames)
  211. })
  212. }
  213. t.Run("pagination", func(t *testing.T) {
  214. got, err := db.List(ctx, ListOrganizationsOptions{Page: 1, PageSize: 1})
  215. require.NoError(t, err)
  216. require.Len(t, got, 1)
  217. assert.Equal(t, org1.ID, got[0].ID)
  218. got, err = db.List(ctx, ListOrganizationsOptions{Page: 2, PageSize: 1})
  219. require.NoError(t, err)
  220. require.Len(t, got, 1)
  221. assert.Equal(t, org2.ID, got[0].ID)
  222. got, err = db.List(ctx, ListOrganizationsOptions{Page: 1, PageSize: 4})
  223. require.NoError(t, err)
  224. require.Len(t, got, 3)
  225. assert.Equal(t, org1.ID, got[0].ID)
  226. assert.Equal(t, org2.ID, got[1].ID)
  227. assert.Equal(t, org3.ID, got[2].ID)
  228. })
  229. }
  230. func orgsSearchByName(t *testing.T, ctx context.Context, db *organizations) {
  231. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsSearchByName-tempPictureAvatarUploadPath")
  232. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  233. tempRepositoryRoot := filepath.Join(os.TempDir(), "orgsSearchByName-tempRepositoryRoot")
  234. conf.SetMockRepository(t, conf.RepositoryOpts{Root: tempRepositoryRoot})
  235. org1, err := db.Create(ctx, "org1", 1, CreateOrganizationOptions{FullName: "Acme Corp"})
  236. require.NoError(t, err)
  237. org2, err := db.Create(ctx, "org2", 1, CreateOrganizationOptions{FullName: "Acme Corp 2"})
  238. require.NoError(t, err)
  239. t.Run("search for username org1", func(t *testing.T) {
  240. orgs, count, err := db.SearchByName(ctx, "G1", 1, 1, "")
  241. require.NoError(t, err)
  242. require.Len(t, orgs, int(count))
  243. assert.Equal(t, int64(1), count)
  244. assert.Equal(t, org1.ID, orgs[0].ID)
  245. })
  246. t.Run("search for username org2", func(t *testing.T) {
  247. orgs, count, err := db.SearchByName(ctx, "G2", 1, 1, "")
  248. require.NoError(t, err)
  249. require.Len(t, orgs, int(count))
  250. assert.Equal(t, int64(1), count)
  251. assert.Equal(t, org2.ID, orgs[0].ID)
  252. })
  253. t.Run("search for full name acme", func(t *testing.T) {
  254. orgs, count, err := db.SearchByName(ctx, "ACME", 1, 10, "")
  255. require.NoError(t, err)
  256. require.Len(t, orgs, int(count))
  257. assert.Equal(t, int64(2), count)
  258. })
  259. t.Run("search for full name acme ORDER BY id DESC LIMIT 1", func(t *testing.T) {
  260. orgs, count, err := db.SearchByName(ctx, "ACME", 1, 1, "id DESC")
  261. require.NoError(t, err)
  262. require.Len(t, orgs, 1)
  263. assert.Equal(t, int64(2), count)
  264. assert.Equal(t, org2.ID, orgs[0].ID)
  265. })
  266. }
  267. func orgsCountByUser(t *testing.T, ctx context.Context, db *organizations) {
  268. usersStore := NewUsersStore(db.DB)
  269. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  270. require.NoError(t, err)
  271. bob, err := usersStore.Create(ctx, "bob", "bob@example.com", CreateUserOptions{})
  272. require.NoError(t, err)
  273. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsCountByUser-tempPictureAvatarUploadPath")
  274. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  275. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  276. require.NoError(t, err)
  277. err = db.AddMember(ctx, org1.ID, bob.ID)
  278. require.NoError(t, err)
  279. got, err := db.CountByUser(ctx, alice.ID)
  280. require.NoError(t, err)
  281. assert.Equal(t, int64(1), got)
  282. got, err = db.CountByUser(ctx, 404)
  283. require.NoError(t, err)
  284. assert.Equal(t, int64(0), got)
  285. }
  286. func orgsCount(t *testing.T, ctx context.Context, db *organizations) {
  287. // Has no organization initially
  288. got := db.Count(ctx)
  289. assert.Equal(t, int64(0), got)
  290. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "usersCount-tempPictureAvatarUploadPath")
  291. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  292. _, err := db.Create(ctx, "org1", 1, CreateOrganizationOptions{})
  293. require.NoError(t, err)
  294. // Create a user shouldn't count
  295. _, err = NewUsersStore(db.DB).Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  296. require.NoError(t, err)
  297. got = db.Count(ctx)
  298. assert.Equal(t, int64(1), got)
  299. }
  300. func orgsDeleteByID(t *testing.T, ctx context.Context, db *organizations) {
  301. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsDeleteByID-tempPictureAvatarUploadPath")
  302. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  303. t.Run("organization still has repository ownership", func(t *testing.T) {
  304. org1, err := db.Create(ctx, "org1", 404, CreateOrganizationOptions{})
  305. require.NoError(t, err)
  306. _, err = NewRepositoriesStore(db.DB).Create(ctx, org1.ID, CreateRepoOptions{Name: "repo1"})
  307. require.NoError(t, err)
  308. err = db.DeleteByID(ctx, org1.ID)
  309. wantErr := ErrOrganizationOwnRepos{errutil.Args{"orgID": org1.ID}}
  310. assert.Equal(t, wantErr, err)
  311. })
  312. alice, err := NewUsersStore(db.DB).Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  313. require.NoError(t, err)
  314. org2, err := db.Create(ctx, "org2", alice.ID, CreateOrganizationOptions{})
  315. require.NoError(t, err)
  316. // Mock team membership
  317. // TODO: Use Organizations.CreateTeam to replace SQL hack when the method is available.
  318. team1 := &Team{
  319. OrgID: org2.ID,
  320. LowerName: "team1",
  321. Name: "team1",
  322. NumMembers: 1,
  323. }
  324. err = db.DB.Create(team1).Error
  325. require.NoError(t, err)
  326. // TODO: Use Organizations.AddTeamMember to replace SQL hack when the method is available.
  327. err = db.DB.Create(
  328. &TeamUser{
  329. OrgID: org2.ID,
  330. TeamID: team1.ID,
  331. UID: alice.ID,
  332. },
  333. ).Error
  334. require.NoError(t, err)
  335. // Pull the trigger
  336. err = db.DeleteByID(ctx, org2.ID)
  337. require.NoError(t, err)
  338. // Verify after-the-fact data
  339. for _, table := range []any{
  340. &Team{OrgID: org2.ID},
  341. &TeamUser{OrgID: org2.ID},
  342. } {
  343. var count int64
  344. err = db.DB.Model(table).Where(table).Count(&count).Error
  345. require.NoError(t, err, "table for %T", table)
  346. assert.Equal(t, int64(0), count, "table for %T", table)
  347. }
  348. _, err = db.GetByName(ctx, org2.Name)
  349. wantErr := ErrOrganizationNotExist{errutil.Args{"name": org2.Name}}
  350. assert.Equal(t, wantErr, err)
  351. }
  352. func orgsAddMember(t *testing.T, ctx context.Context, db *organizations) {
  353. usersStore := NewUsersStore(db.DB)
  354. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  355. require.NoError(t, err)
  356. bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
  357. require.NoError(t, err)
  358. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsAddMember-tempPictureAvatarUploadPath")
  359. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  360. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  361. require.NoError(t, err)
  362. // Not yet a member
  363. gotOrgs, err := db.List(ctx, ListOrganizationsOptions{MemberID: bob.ID, IncludePrivateMembers: true})
  364. require.NoError(t, err)
  365. assert.Len(t, gotOrgs, 0)
  366. // Add member
  367. err = db.AddMember(ctx, org1.ID, bob.ID)
  368. require.NoError(t, err)
  369. // Now a member
  370. gotOrgs, err = db.List(ctx, ListOrganizationsOptions{MemberID: bob.ID, IncludePrivateMembers: true})
  371. require.NoError(t, err)
  372. assert.Len(t, gotOrgs, 1)
  373. assert.Equal(t, org1.ID, gotOrgs[0].ID)
  374. // Add member again shouldn't fail
  375. err = db.AddMember(ctx, org1.ID, bob.ID)
  376. require.NoError(t, err)
  377. gotOrg, err := db.GetByName(ctx, org1.Name)
  378. require.NoError(t, err)
  379. assert.Equal(t, 2, gotOrg.NumMembers)
  380. }
  381. func orgsRemoveMember(t *testing.T, ctx context.Context, db *organizations) {
  382. usersStore := NewUsersStore(db.DB)
  383. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  384. require.NoError(t, err)
  385. bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
  386. require.NoError(t, err)
  387. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsRemoveMember-tempPictureAvatarUploadPath")
  388. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  389. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  390. require.NoError(t, err)
  391. t.Run("remove non-existent member", func(t *testing.T) {
  392. err = db.RemoveMember(ctx, org1.ID, bob.ID)
  393. require.NoError(t, err)
  394. })
  395. t.Run("remove last owner", func(t *testing.T) {
  396. err = db.RemoveMember(ctx, org1.ID, alice.ID)
  397. wantErr := ErrLastOrgOwner{errutil.Args{"orgID": org1.ID, "userID": alice.ID}}
  398. assert.Equal(t, wantErr, err)
  399. })
  400. err = db.AddMember(ctx, org1.ID, bob.ID)
  401. require.NoError(t, err)
  402. // Mock repository, watches and collaborations
  403. reposStore := NewRepositoriesStore(db.DB)
  404. repo1, err := reposStore.Create(ctx, org1.ID, CreateRepoOptions{Name: "repo1", Private: true})
  405. require.NoError(t, err)
  406. // TODO: Use Repositories.AddCollaborator to replace SQL hack when the method is available.
  407. err = db.DB.Create(
  408. &Collaboration{
  409. UserID: bob.ID,
  410. RepoID: repo1.ID,
  411. Mode: AccessModeRead,
  412. },
  413. ).Error
  414. require.NoError(t, err)
  415. // Mock team membership
  416. // TODO: Use Organizations.CreateTeam to replace SQL hack when the method is available.
  417. team1 := &Team{
  418. OrgID: org1.ID,
  419. LowerName: "team1",
  420. Name: "team1",
  421. NumMembers: 1,
  422. }
  423. err = db.DB.Create(team1).Error
  424. require.NoError(t, err)
  425. // TODO: Use Organizations.AddTeamMember to replace SQL hack when the method is available.
  426. err = db.DB.Create(
  427. &TeamUser{
  428. OrgID: org1.ID,
  429. TeamID: team1.ID,
  430. UID: bob.ID,
  431. },
  432. ).Error
  433. require.NoError(t, err)
  434. // TODO: Use Organizations.AddTeamRepository to replace SQL hack when the method is available.
  435. err = db.DB.Create(
  436. &TeamRepo{
  437. OrgID: org1.ID,
  438. TeamID: team1.ID,
  439. RepoID: repo1.ID,
  440. },
  441. ).Error
  442. require.NoError(t, err)
  443. // Mock accesses
  444. permsStore := NewPermsStore(db.DB)
  445. err = permsStore.SetRepoPerms(ctx, repo1.ID, map[int64]AccessMode{bob.ID: AccessModeRead})
  446. require.NoError(t, err)
  447. err = reposStore.Watch(
  448. ctx,
  449. WatchRepositoryOptions{
  450. UserID: bob.ID,
  451. RepoID: repo1.ID,
  452. RepoOwnerID: repo1.OwnerID,
  453. RepoIsPrivate: repo1.IsPrivate,
  454. },
  455. )
  456. require.NoError(t, err)
  457. // Pull the trigger
  458. err = db.RemoveMember(ctx, org1.ID, bob.ID)
  459. require.NoError(t, err)
  460. // Verify after-the-fact data
  461. gotRepo, err := reposStore.GetByID(ctx, repo1.ID)
  462. require.NoError(t, err)
  463. assert.Equal(t, 1, gotRepo.NumWatches)
  464. gotAccessMode := permsStore.AccessMode(ctx, repo1.ID, bob.ID, AccessModeOptions{Private: repo1.IsPrivate})
  465. assert.Equal(t, AccessModeNone, gotAccessMode)
  466. // TODO: Use Repositories.ListCollaborators to replace SQL hack when the method is available.
  467. var count int64
  468. err = db.DB.Model(&Collaboration{}).Where(&Collaboration{RepoID: repo1.ID}).Count(&count).Error
  469. require.NoError(t, err)
  470. assert.Equal(t, int64(0), count)
  471. gotTeam, err := db.GetTeamByName(ctx, org1.ID, team1.Name)
  472. require.NoError(t, err)
  473. assert.Equal(t, 0, gotTeam.NumMembers)
  474. gotOrg, err := db.GetByName(ctx, org1.Name)
  475. require.NoError(t, err)
  476. assert.Equal(t, 1, gotOrg.NumMembers)
  477. }
  478. func orgsHasMember(t *testing.T, ctx context.Context, db *organizations) {
  479. got, _ := db.HasMember(ctx, 1, 1)
  480. assert.False(t, got)
  481. err := db.AddMember(ctx, 1, 1)
  482. require.NoError(t, err)
  483. got, _ = db.HasMember(ctx, 1, 1)
  484. assert.True(t, got)
  485. }
  486. func orgsListMembers(t *testing.T, ctx context.Context, db *organizations) {
  487. usersStore := NewUsersStore(db.DB)
  488. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  489. require.NoError(t, err)
  490. bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
  491. require.NoError(t, err)
  492. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsListMembers-tempPictureAvatarUploadPath")
  493. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  494. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  495. require.NoError(t, err)
  496. err = db.AddMember(ctx, org1.ID, bob.ID)
  497. require.NoError(t, err)
  498. got, err := db.ListMembers(ctx, org1.ID, ListOrgMembersOptions{Limit: 1})
  499. require.NoError(t, err)
  500. require.Len(t, got, 1)
  501. assert.Equal(t, alice.ID, got[0].ID)
  502. got, err = db.ListMembers(ctx, org1.ID, ListOrgMembersOptions{})
  503. require.NoError(t, err)
  504. require.Len(t, got, 2)
  505. assert.Equal(t, alice.ID, got[0].ID)
  506. assert.Equal(t, bob.ID, got[1].ID)
  507. }
  508. func orgsIsOwnedBy(t *testing.T, ctx context.Context, db *organizations) {
  509. usersStore := NewUsersStore(db.DB)
  510. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  511. require.NoError(t, err)
  512. bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
  513. require.NoError(t, err)
  514. cindy, err := usersStore.Create(ctx, "cindy", "cindy@exmaple.com", CreateUserOptions{})
  515. require.NoError(t, err)
  516. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsIsOwnedBy-tempPictureAvatarUploadPath")
  517. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  518. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  519. require.NoError(t, err)
  520. err = db.AddMember(ctx, org1.ID, bob.ID)
  521. require.NoError(t, err)
  522. got := db.IsOwnedBy(ctx, org1.ID, alice.ID)
  523. assert.True(t, got)
  524. got = db.IsOwnedBy(ctx, org1.ID, bob.ID)
  525. assert.False(t, got)
  526. got = db.IsOwnedBy(ctx, org1.ID, cindy.ID)
  527. assert.False(t, got)
  528. }
  529. func orgsSetMemberVisibility(t *testing.T, ctx context.Context, db *organizations) {
  530. usersStore := NewUsersStore(db.DB)
  531. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  532. require.NoError(t, err)
  533. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsSetMemberVisibility-tempPictureAvatarUploadPath")
  534. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  535. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  536. require.NoError(t, err)
  537. got, err := db.List(ctx, ListOrganizationsOptions{MemberID: alice.ID})
  538. require.NoError(t, err)
  539. assert.Len(t, got, 0)
  540. err = db.SetMemberVisibility(ctx, org1.ID, alice.ID, true)
  541. require.NoError(t, err)
  542. got, err = db.List(ctx, ListOrganizationsOptions{MemberID: alice.ID})
  543. require.NoError(t, err)
  544. assert.Len(t, got, 1)
  545. }
  546. func orgsGetTeamByName(t *testing.T, ctx context.Context, db *organizations) {
  547. usersStore := NewUsersStore(db.DB)
  548. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  549. require.NoError(t, err)
  550. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsGetTeamByName-tempPictureAvatarUploadPath")
  551. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  552. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  553. require.NoError(t, err)
  554. t.Run("non-existent team", func(t *testing.T) {
  555. _, err := db.GetTeamByName(ctx, org1.ID, "non-existent")
  556. wantErr := ErrTeamNotExist{errutil.Args{"orgID": org1.ID, "name": "non-existent"}}
  557. assert.Equal(t, wantErr, err)
  558. })
  559. t.Run("team of another organization", func(t *testing.T) {
  560. org2, err := db.Create(ctx, "org2", alice.ID, CreateOrganizationOptions{})
  561. require.NoError(t, err)
  562. // TODO: Use Organizations.CreateTeam to replace SQL hack when the method is available.
  563. team1 := &Team{
  564. OrgID: org2.ID,
  565. LowerName: "team1",
  566. Name: "team1",
  567. NumMembers: 1,
  568. }
  569. err = db.DB.Create(team1).Error
  570. require.NoError(t, err)
  571. _, err = db.GetTeamByName(ctx, org1.ID, team1.Name)
  572. wantErr := ErrTeamNotExist{errutil.Args{"orgID": org1.ID, "name": team1.Name}}
  573. assert.Equal(t, wantErr, err)
  574. })
  575. _, err = db.GetTeamByName(ctx, org1.ID, TeamNameOwners)
  576. require.NoError(t, err)
  577. }
  578. func orgsAccessibleRepositoriesByUser(t *testing.T, ctx context.Context, db *organizations) {
  579. // todo
  580. }
  581. func orgsMirrorRepositoriesByUser(t *testing.T, ctx context.Context, db *organizations) {
  582. // todo
  583. }