organizations_test.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704
  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, db *organizations) {
  287. ctx := context.Background()
  288. // Has no organization initially
  289. got := db.Count(ctx)
  290. assert.Equal(t, int64(0), got)
  291. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "usersCount-tempPictureAvatarUploadPath")
  292. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  293. _, err := db.Create(ctx, "org1", 1, CreateOrganizationOptions{})
  294. require.NoError(t, err)
  295. // Create a user shouldn't count
  296. _, err = NewUsersStore(db.DB).Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  297. require.NoError(t, err)
  298. got = db.Count(ctx)
  299. assert.Equal(t, int64(1), got)
  300. }
  301. func orgsDeleteByID(t *testing.T, db *organizations) {
  302. ctx := context.Background()
  303. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsDeleteByID-tempPictureAvatarUploadPath")
  304. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  305. t.Run("organization still has repository ownership", func(t *testing.T) {
  306. org1, err := db.Create(ctx, "org1", 404, CreateOrganizationOptions{})
  307. require.NoError(t, err)
  308. _, err = NewRepositoriesStore(db.DB).Create(ctx, org1.ID, CreateRepoOptions{Name: "repo1"})
  309. require.NoError(t, err)
  310. err = db.DeleteByID(ctx, org1.ID)
  311. wantErr := ErrOrganizationOwnRepos{errutil.Args{"orgID": org1.ID}}
  312. assert.Equal(t, wantErr, err)
  313. })
  314. alice, err := NewUsersStore(db.DB).Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  315. require.NoError(t, err)
  316. org2, err := db.Create(ctx, "org2", alice.ID, CreateOrganizationOptions{})
  317. require.NoError(t, err)
  318. // Mock team membership
  319. // TODO: Use Organizations.CreateTeam to replace SQL hack when the method is available.
  320. team1 := &Team{
  321. OrgID: org2.ID,
  322. LowerName: "team1",
  323. Name: "team1",
  324. NumMembers: 1,
  325. }
  326. err = db.DB.Create(team1).Error
  327. require.NoError(t, err)
  328. // TODO: Use Organizations.AddTeamMember to replace SQL hack when the method is available.
  329. err = db.DB.Create(
  330. &TeamUser{
  331. OrgID: org2.ID,
  332. TeamID: team1.ID,
  333. UID: alice.ID,
  334. },
  335. ).Error
  336. require.NoError(t, err)
  337. // Pull the trigger
  338. err = db.DeleteByID(ctx, org2.ID)
  339. require.NoError(t, err)
  340. // Verify after-the-fact data
  341. for _, table := range []any{
  342. &Team{OrgID: org2.ID},
  343. &TeamUser{OrgID: org2.ID},
  344. } {
  345. var count int64
  346. err = db.DB.Model(table).Where(table).Count(&count).Error
  347. require.NoError(t, err, "table for %T", table)
  348. assert.Equal(t, int64(0), count, "table for %T", table)
  349. }
  350. _, err = db.GetByName(ctx, org2.Name)
  351. wantErr := ErrOrganizationNotExist{errutil.Args{"name": org2.Name}}
  352. assert.Equal(t, wantErr, err)
  353. }
  354. func orgsAddMember(t *testing.T, db *organizations) {
  355. ctx := context.Background()
  356. usersStore := NewUsersStore(db.DB)
  357. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  358. require.NoError(t, err)
  359. bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
  360. require.NoError(t, err)
  361. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsAddMember-tempPictureAvatarUploadPath")
  362. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  363. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  364. require.NoError(t, err)
  365. // Not yet a member
  366. gotOrgs, err := db.List(ctx, ListOrganizationsOptions{MemberID: bob.ID, IncludePrivateMembers: true})
  367. require.NoError(t, err)
  368. assert.Len(t, gotOrgs, 0)
  369. // Add member
  370. err = db.AddMember(ctx, org1.ID, bob.ID)
  371. require.NoError(t, err)
  372. // Now a member
  373. gotOrgs, err = db.List(ctx, ListOrganizationsOptions{MemberID: bob.ID, IncludePrivateMembers: true})
  374. require.NoError(t, err)
  375. assert.Len(t, gotOrgs, 1)
  376. assert.Equal(t, org1.ID, gotOrgs[0].ID)
  377. // Add member again shouldn't fail
  378. err = db.AddMember(ctx, org1.ID, bob.ID)
  379. require.NoError(t, err)
  380. gotOrg, err := db.GetByName(ctx, org1.Name)
  381. require.NoError(t, err)
  382. assert.Equal(t, 2, gotOrg.NumMembers)
  383. }
  384. func orgsRemoveMember(t *testing.T, db *organizations) {
  385. ctx := context.Background()
  386. usersStore := NewUsersStore(db.DB)
  387. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  388. require.NoError(t, err)
  389. bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
  390. require.NoError(t, err)
  391. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsRemoveMember-tempPictureAvatarUploadPath")
  392. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  393. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  394. require.NoError(t, err)
  395. t.Run("remove non-existent member", func(t *testing.T) {
  396. err = db.RemoveMember(ctx, org1.ID, bob.ID)
  397. require.NoError(t, err)
  398. })
  399. t.Run("remove last owner", func(t *testing.T) {
  400. err = db.RemoveMember(ctx, org1.ID, alice.ID)
  401. wantErr := ErrLastOrgOwner{errutil.Args{"orgID": org1.ID, "userID": alice.ID}}
  402. assert.Equal(t, wantErr, err)
  403. })
  404. err = db.AddMember(ctx, org1.ID, bob.ID)
  405. require.NoError(t, err)
  406. // Mock repository, watches and collaborations
  407. reposStore := NewRepositoriesStore(db.DB)
  408. repo1, err := reposStore.Create(ctx, org1.ID, CreateRepoOptions{Name: "repo1", Private: true})
  409. require.NoError(t, err)
  410. // TODO: Use Repositories.AddCollaborator to replace SQL hack when the method is available.
  411. err = db.DB.Create(
  412. &Collaboration{
  413. UserID: bob.ID,
  414. RepoID: repo1.ID,
  415. Mode: AccessModeRead,
  416. },
  417. ).Error
  418. require.NoError(t, err)
  419. // Mock team membership
  420. // TODO: Use Organizations.CreateTeam to replace SQL hack when the method is available.
  421. team1 := &Team{
  422. OrgID: org1.ID,
  423. LowerName: "team1",
  424. Name: "team1",
  425. NumMembers: 1,
  426. }
  427. err = db.DB.Create(team1).Error
  428. require.NoError(t, err)
  429. // TODO: Use Organizations.AddTeamMember to replace SQL hack when the method is available.
  430. err = db.DB.Create(
  431. &TeamUser{
  432. OrgID: org1.ID,
  433. TeamID: team1.ID,
  434. UID: bob.ID,
  435. },
  436. ).Error
  437. require.NoError(t, err)
  438. // TODO: Use Organizations.AddTeamRepository to replace SQL hack when the method is available.
  439. err = db.DB.Create(
  440. &TeamRepo{
  441. OrgID: org1.ID,
  442. TeamID: team1.ID,
  443. RepoID: repo1.ID,
  444. },
  445. ).Error
  446. require.NoError(t, err)
  447. // Mock accesses
  448. permsStore := NewPermsStore(db.DB)
  449. err = permsStore.SetRepoPerms(ctx, repo1.ID, map[int64]AccessMode{bob.ID: AccessModeRead})
  450. require.NoError(t, err)
  451. err = reposStore.Watch(
  452. ctx,
  453. WatchRepositoryOptions{
  454. UserID: bob.ID,
  455. RepoID: repo1.ID,
  456. RepoOwnerID: repo1.OwnerID,
  457. RepoIsPrivate: repo1.IsPrivate,
  458. },
  459. )
  460. require.NoError(t, err)
  461. // Pull the trigger
  462. err = db.RemoveMember(ctx, org1.ID, bob.ID)
  463. require.NoError(t, err)
  464. // Verify after-the-fact data
  465. gotRepo, err := reposStore.GetByID(ctx, repo1.ID)
  466. require.NoError(t, err)
  467. assert.Equal(t, 1, gotRepo.NumWatches)
  468. gotAccessMode := permsStore.AccessMode(ctx, repo1.ID, bob.ID, AccessModeOptions{Private: repo1.IsPrivate})
  469. assert.Equal(t, AccessModeNone, gotAccessMode)
  470. // TODO: Use Repositories.ListCollaborators to replace SQL hack when the method is available.
  471. var count int64
  472. err = db.DB.Model(&Collaboration{}).Where(&Collaboration{RepoID: repo1.ID}).Count(&count).Error
  473. require.NoError(t, err)
  474. assert.Equal(t, int64(0), count)
  475. gotTeam, err := db.GetTeamByName(ctx, org1.ID, team1.Name)
  476. require.NoError(t, err)
  477. assert.Equal(t, 0, gotTeam.NumMembers)
  478. gotOrg, err := db.GetByName(ctx, org1.Name)
  479. require.NoError(t, err)
  480. assert.Equal(t, 1, gotOrg.NumMembers)
  481. }
  482. func orgsHasMember(t *testing.T, db *organizations) {
  483. ctx := context.Background()
  484. got, _ := db.HasMember(ctx, 1, 1)
  485. assert.False(t, got)
  486. err := db.AddMember(ctx, 1, 1)
  487. require.NoError(t, err)
  488. got, _ = db.HasMember(ctx, 1, 1)
  489. assert.True(t, got)
  490. }
  491. func orgsListMembers(t *testing.T, db *organizations) {
  492. ctx := context.Background()
  493. usersStore := NewUsersStore(db.DB)
  494. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  495. require.NoError(t, err)
  496. bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
  497. require.NoError(t, err)
  498. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsListMembers-tempPictureAvatarUploadPath")
  499. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  500. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  501. require.NoError(t, err)
  502. err = db.AddMember(ctx, org1.ID, bob.ID)
  503. require.NoError(t, err)
  504. got, err := db.ListMembers(ctx, org1.ID, ListOrgMembersOptions{Limit: 1})
  505. require.NoError(t, err)
  506. require.Len(t, got, 1)
  507. assert.Equal(t, alice.ID, got[0].ID)
  508. got, err = db.ListMembers(ctx, org1.ID, ListOrgMembersOptions{})
  509. require.NoError(t, err)
  510. require.Len(t, got, 2)
  511. assert.Equal(t, alice.ID, got[0].ID)
  512. assert.Equal(t, bob.ID, got[1].ID)
  513. }
  514. func orgsIsOwnedBy(t *testing.T, db *organizations) {
  515. ctx := context.Background()
  516. usersStore := NewUsersStore(db.DB)
  517. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  518. require.NoError(t, err)
  519. bob, err := usersStore.Create(ctx, "bob", "bob@exmaple.com", CreateUserOptions{})
  520. require.NoError(t, err)
  521. cindy, err := usersStore.Create(ctx, "cindy", "cindy@exmaple.com", CreateUserOptions{})
  522. require.NoError(t, err)
  523. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsIsOwnedBy-tempPictureAvatarUploadPath")
  524. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  525. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  526. require.NoError(t, err)
  527. err = db.AddMember(ctx, org1.ID, bob.ID)
  528. require.NoError(t, err)
  529. got := db.IsOwnedBy(ctx, org1.ID, alice.ID)
  530. assert.True(t, got)
  531. got = db.IsOwnedBy(ctx, org1.ID, bob.ID)
  532. assert.False(t, got)
  533. got = db.IsOwnedBy(ctx, org1.ID, cindy.ID)
  534. assert.False(t, got)
  535. }
  536. func orgsSetMemberVisibility(t *testing.T, db *organizations) {
  537. ctx := context.Background()
  538. usersStore := NewUsersStore(db.DB)
  539. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  540. require.NoError(t, err)
  541. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsSetMemberVisibility-tempPictureAvatarUploadPath")
  542. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  543. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  544. require.NoError(t, err)
  545. got, err := db.List(ctx, ListOrganizationsOptions{MemberID: alice.ID})
  546. require.NoError(t, err)
  547. assert.Len(t, got, 0)
  548. err = db.SetMemberVisibility(ctx, org1.ID, alice.ID, true)
  549. require.NoError(t, err)
  550. got, err = db.List(ctx, ListOrganizationsOptions{MemberID: alice.ID})
  551. require.NoError(t, err)
  552. assert.Len(t, got, 1)
  553. }
  554. func orgsGetTeamByName(t *testing.T, db *organizations) {
  555. ctx := context.Background()
  556. usersStore := NewUsersStore(db.DB)
  557. alice, err := usersStore.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  558. require.NoError(t, err)
  559. tempPictureAvatarUploadPath := filepath.Join(os.TempDir(), "orgsGetTeamByName-tempPictureAvatarUploadPath")
  560. conf.SetMockPicture(t, conf.PictureOpts{AvatarUploadPath: tempPictureAvatarUploadPath})
  561. org1, err := db.Create(ctx, "org1", alice.ID, CreateOrganizationOptions{})
  562. require.NoError(t, err)
  563. t.Run("non-existent team", func(t *testing.T) {
  564. _, err := db.GetTeamByName(ctx, org1.ID, "non-existent")
  565. wantErr := ErrTeamNotExist{errutil.Args{"orgID": org1.ID, "name": "non-existent"}}
  566. assert.Equal(t, wantErr, err)
  567. })
  568. t.Run("team of another organization", func(t *testing.T) {
  569. org2, err := db.Create(ctx, "org2", alice.ID, CreateOrganizationOptions{})
  570. require.NoError(t, err)
  571. // TODO: Use Organizations.CreateTeam to replace SQL hack when the method is available.
  572. team1 := &Team{
  573. OrgID: org2.ID,
  574. LowerName: "team1",
  575. Name: "team1",
  576. NumMembers: 1,
  577. }
  578. err = db.DB.Create(team1).Error
  579. require.NoError(t, err)
  580. _, err = db.GetTeamByName(ctx, org1.ID, team1.Name)
  581. wantErr := ErrTeamNotExist{errutil.Args{"orgID": org1.ID, "name": team1.Name}}
  582. assert.Equal(t, wantErr, err)
  583. })
  584. _, err = db.GetTeamByName(ctx, org1.ID, TeamNameOwners)
  585. require.NoError(t, err)
  586. }
  587. func orgsAccessibleRepositoriesByUser(t *testing.T, db *organizations) {
  588. // todo
  589. }
  590. func orgsMirrorRepositoriesByUser(t *testing.T, db *organizations) {
  591. // todo
  592. }