users_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711
  1. // Copyright 2020 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. "fmt"
  8. "os"
  9. "path/filepath"
  10. "strings"
  11. "testing"
  12. "time"
  13. "github.com/stretchr/testify/assert"
  14. "github.com/stretchr/testify/require"
  15. "gorm.io/gorm"
  16. "gogs.io/gogs/internal/auth"
  17. "gogs.io/gogs/internal/conf"
  18. "gogs.io/gogs/internal/dbtest"
  19. "gogs.io/gogs/internal/dbutil"
  20. "gogs.io/gogs/internal/errutil"
  21. "gogs.io/gogs/internal/osutil"
  22. "gogs.io/gogs/internal/repoutil"
  23. "gogs.io/gogs/internal/userutil"
  24. "gogs.io/gogs/public"
  25. )
  26. func TestUser_BeforeCreate(t *testing.T) {
  27. now := time.Now()
  28. db := &gorm.DB{
  29. Config: &gorm.Config{
  30. SkipDefaultTransaction: true,
  31. NowFunc: func() time.Time {
  32. return now
  33. },
  34. },
  35. }
  36. t.Run("CreatedUnix has been set", func(t *testing.T) {
  37. user := &User{
  38. CreatedUnix: 1,
  39. }
  40. _ = user.BeforeCreate(db)
  41. assert.Equal(t, int64(1), user.CreatedUnix)
  42. assert.Equal(t, int64(0), user.UpdatedUnix)
  43. })
  44. t.Run("CreatedUnix has not been set", func(t *testing.T) {
  45. user := &User{}
  46. _ = user.BeforeCreate(db)
  47. assert.Equal(t, db.NowFunc().Unix(), user.CreatedUnix)
  48. assert.Equal(t, db.NowFunc().Unix(), user.UpdatedUnix)
  49. })
  50. }
  51. func TestUser_AfterFind(t *testing.T) {
  52. now := time.Now()
  53. db := &gorm.DB{
  54. Config: &gorm.Config{
  55. SkipDefaultTransaction: true,
  56. NowFunc: func() time.Time {
  57. return now
  58. },
  59. },
  60. }
  61. user := &User{
  62. CreatedUnix: now.Unix(),
  63. UpdatedUnix: now.Unix(),
  64. }
  65. _ = user.AfterFind(db)
  66. assert.Equal(t, user.CreatedUnix, user.Created.Unix())
  67. assert.Equal(t, user.UpdatedUnix, user.Updated.Unix())
  68. }
  69. func TestUsers(t *testing.T) {
  70. if testing.Short() {
  71. t.Skip()
  72. }
  73. t.Parallel()
  74. tables := []interface{}{new(User), new(EmailAddress), new(Repository), new(Follow), new(PullRequest)}
  75. db := &users{
  76. DB: dbtest.NewDB(t, "users", tables...),
  77. }
  78. for _, tc := range []struct {
  79. name string
  80. test func(t *testing.T, db *users)
  81. }{
  82. {"Authenticate", usersAuthenticate},
  83. {"ChangeUsername", usersChangeUsername},
  84. {"Count", usersCount},
  85. {"Create", usersCreate},
  86. {"DeleteCustomAvatar", usersDeleteCustomAvatar},
  87. {"GetByEmail", usersGetByEmail},
  88. {"GetByID", usersGetByID},
  89. {"GetByUsername", usersGetByUsername},
  90. {"HasForkedRepository", usersHasForkedRepository},
  91. {"IsUsernameUsed", usersIsUsernameUsed},
  92. {"List", usersList},
  93. {"ListFollowers", usersListFollowers},
  94. {"ListFollowings", usersListFollowings},
  95. {"UseCustomAvatar", usersUseCustomAvatar},
  96. } {
  97. t.Run(tc.name, func(t *testing.T) {
  98. t.Cleanup(func() {
  99. err := clearTables(t, db.DB, tables...)
  100. require.NoError(t, err)
  101. })
  102. tc.test(t, db)
  103. })
  104. if t.Failed() {
  105. break
  106. }
  107. }
  108. }
  109. func usersAuthenticate(t *testing.T, db *users) {
  110. ctx := context.Background()
  111. password := "pa$$word"
  112. alice, err := db.Create(ctx, "alice", "alice@example.com",
  113. CreateUserOptions{
  114. Password: password,
  115. },
  116. )
  117. require.NoError(t, err)
  118. t.Run("user not found", func(t *testing.T) {
  119. _, err := db.Authenticate(ctx, "bob", password, -1)
  120. wantErr := auth.ErrBadCredentials{Args: map[string]interface{}{"login": "bob"}}
  121. assert.Equal(t, wantErr, err)
  122. })
  123. t.Run("invalid password", func(t *testing.T) {
  124. _, err := db.Authenticate(ctx, alice.Name, "bad_password", -1)
  125. wantErr := auth.ErrBadCredentials{Args: map[string]interface{}{"login": alice.Name, "userID": alice.ID}}
  126. assert.Equal(t, wantErr, err)
  127. })
  128. t.Run("via email and password", func(t *testing.T) {
  129. user, err := db.Authenticate(ctx, alice.Email, password, -1)
  130. require.NoError(t, err)
  131. assert.Equal(t, alice.Name, user.Name)
  132. })
  133. t.Run("via username and password", func(t *testing.T) {
  134. user, err := db.Authenticate(ctx, alice.Name, password, -1)
  135. require.NoError(t, err)
  136. assert.Equal(t, alice.Name, user.Name)
  137. })
  138. t.Run("login source mismatch", func(t *testing.T) {
  139. _, err := db.Authenticate(ctx, alice.Email, password, 1)
  140. gotErr := fmt.Sprintf("%v", err)
  141. wantErr := ErrLoginSourceMismatch{args: map[string]interface{}{"actual": 0, "expect": 1}}.Error()
  142. assert.Equal(t, wantErr, gotErr)
  143. })
  144. t.Run("via login source", func(t *testing.T) {
  145. mockLoginSources := NewMockLoginSourcesStore()
  146. mockLoginSources.GetByIDFunc.SetDefaultHook(func(ctx context.Context, id int64) (*LoginSource, error) {
  147. mockProvider := NewMockProvider()
  148. mockProvider.AuthenticateFunc.SetDefaultReturn(&auth.ExternalAccount{}, nil)
  149. s := &LoginSource{
  150. IsActived: true,
  151. Provider: mockProvider,
  152. }
  153. return s, nil
  154. })
  155. setMockLoginSourcesStore(t, mockLoginSources)
  156. bob, err := db.Create(ctx, "bob", "bob@example.com",
  157. CreateUserOptions{
  158. Password: password,
  159. LoginSource: 1,
  160. },
  161. )
  162. require.NoError(t, err)
  163. user, err := db.Authenticate(ctx, bob.Email, password, 1)
  164. require.NoError(t, err)
  165. assert.Equal(t, bob.Name, user.Name)
  166. })
  167. t.Run("new user via login source", func(t *testing.T) {
  168. mockLoginSources := NewMockLoginSourcesStore()
  169. mockLoginSources.GetByIDFunc.SetDefaultHook(func(ctx context.Context, id int64) (*LoginSource, error) {
  170. mockProvider := NewMockProvider()
  171. mockProvider.AuthenticateFunc.SetDefaultReturn(
  172. &auth.ExternalAccount{
  173. Name: "cindy",
  174. Email: "cindy@example.com",
  175. },
  176. nil,
  177. )
  178. s := &LoginSource{
  179. IsActived: true,
  180. Provider: mockProvider,
  181. }
  182. return s, nil
  183. })
  184. setMockLoginSourcesStore(t, mockLoginSources)
  185. user, err := db.Authenticate(ctx, "cindy", password, 1)
  186. require.NoError(t, err)
  187. assert.Equal(t, "cindy", user.Name)
  188. user, err = db.GetByUsername(ctx, "cindy")
  189. require.NoError(t, err)
  190. assert.Equal(t, "cindy@example.com", user.Email)
  191. })
  192. }
  193. func usersChangeUsername(t *testing.T, db *users) {
  194. ctx := context.Background()
  195. alice, err := db.Create(
  196. ctx,
  197. "alice",
  198. "alice@example.com",
  199. CreateUserOptions{
  200. Activated: true,
  201. },
  202. )
  203. require.NoError(t, err)
  204. t.Run("name not allowed", func(t *testing.T) {
  205. err := db.ChangeUsername(ctx, alice.ID, "-")
  206. wantErr := ErrNameNotAllowed{
  207. args: errutil.Args{
  208. "reason": "reserved",
  209. "name": "-",
  210. },
  211. }
  212. assert.Equal(t, wantErr, err)
  213. })
  214. t.Run("name already exists", func(t *testing.T) {
  215. err := db.ChangeUsername(ctx, alice.ID, alice.Name)
  216. wantErr := ErrUserAlreadyExist{
  217. args: errutil.Args{
  218. "name": alice.Name,
  219. },
  220. }
  221. assert.Equal(t, wantErr, err)
  222. })
  223. tempRepositoryRoot := filepath.Join(os.TempDir(), "usersChangeUsername-tempRepositoryRoot")
  224. conf.SetMockRepository(
  225. t,
  226. conf.RepositoryOpts{
  227. Root: tempRepositoryRoot,
  228. },
  229. )
  230. err = os.RemoveAll(tempRepositoryRoot)
  231. require.NoError(t, err)
  232. defer func() { _ = os.RemoveAll(tempRepositoryRoot) }()
  233. tempServerAppDataPath := filepath.Join(os.TempDir(), "usersChangeUsername-tempServerAppDataPath")
  234. conf.SetMockServer(
  235. t,
  236. conf.ServerOpts{
  237. AppDataPath: tempServerAppDataPath,
  238. },
  239. )
  240. err = os.RemoveAll(tempServerAppDataPath)
  241. require.NoError(t, err)
  242. defer func() { _ = os.RemoveAll(tempServerAppDataPath) }()
  243. repo, err := NewReposStore(db.DB).Create(
  244. ctx,
  245. alice.ID,
  246. CreateRepoOptions{
  247. Name: "test-repo-1",
  248. },
  249. )
  250. require.NoError(t, err)
  251. // TODO: Use PullRequests.Create to replace SQL hack when the method is available.
  252. err = db.Exec(`INSERT INTO pull_request (head_user_name) VALUES (?)`, alice.Name).Error
  253. require.NoError(t, err)
  254. err = db.Model(&User{}).Where("id = ?", alice.ID).Update("updated_unix", 0).Error
  255. require.NoError(t, err)
  256. err = os.MkdirAll(repoutil.UserPath(alice.Name), os.ModePerm)
  257. require.NoError(t, err)
  258. err = os.MkdirAll(repoutil.RepositoryLocalPath(repo.ID), os.ModePerm)
  259. require.NoError(t, err)
  260. err = os.MkdirAll(repoutil.RepositoryLocalWikiPath(repo.ID), os.ModePerm)
  261. require.NoError(t, err)
  262. // Make sure mock data is set up correctly
  263. // TODO: Use PullRequests.GetByID to replace SQL hack when the method is available.
  264. var headUserName string
  265. err = db.Model(&PullRequest{}).Select("head_user_name").Row().Scan(&headUserName)
  266. require.NoError(t, err)
  267. assert.Equal(t, headUserName, alice.Name)
  268. var updatedUnix int64
  269. err = db.Model(&User{}).Select("updated_unix").Row().Scan(&updatedUnix)
  270. require.NoError(t, err)
  271. assert.Equal(t, int64(0), updatedUnix)
  272. assert.True(t, osutil.IsExist(repoutil.UserPath(alice.Name)))
  273. assert.True(t, osutil.IsExist(repoutil.RepositoryLocalPath(repo.ID)))
  274. assert.True(t, osutil.IsExist(repoutil.RepositoryLocalWikiPath(repo.ID)))
  275. const newUsername = "alice-new"
  276. err = db.ChangeUsername(ctx, alice.ID, newUsername)
  277. require.NoError(t, err)
  278. // TODO: Use PullRequests.GetByID to replace SQL hack when the method is available.
  279. err = db.Model(&PullRequest{}).Select("head_user_name").Row().Scan(&headUserName)
  280. require.NoError(t, err)
  281. assert.Equal(t, headUserName, newUsername)
  282. assert.True(t, osutil.IsExist(repoutil.UserPath(newUsername)))
  283. assert.False(t, osutil.IsExist(repoutil.UserPath(alice.Name)))
  284. assert.False(t, osutil.IsExist(repoutil.RepositoryLocalPath(repo.ID)))
  285. assert.False(t, osutil.IsExist(repoutil.RepositoryLocalWikiPath(repo.ID)))
  286. alice, err = db.GetByID(ctx, alice.ID)
  287. require.NoError(t, err)
  288. assert.Equal(t, newUsername, alice.Name)
  289. assert.Equal(t, db.NowFunc().Unix(), alice.UpdatedUnix)
  290. }
  291. func usersCount(t *testing.T, db *users) {
  292. ctx := context.Background()
  293. // Has no user initially
  294. got := db.Count(ctx)
  295. assert.Equal(t, int64(0), got)
  296. _, err := 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. // Create an organization shouldn't count
  301. // TODO: Use Orgs.Create to replace SQL hack when the method is available.
  302. org1, err := db.Create(ctx, "org1", "org1@example.com", CreateUserOptions{})
  303. require.NoError(t, err)
  304. err = db.Exec(
  305. dbutil.Quote("UPDATE %s SET type = ? WHERE id = ?", "user"),
  306. UserTypeOrganization, org1.ID,
  307. ).Error
  308. require.NoError(t, err)
  309. got = db.Count(ctx)
  310. assert.Equal(t, int64(1), got)
  311. }
  312. func usersCreate(t *testing.T, db *users) {
  313. ctx := context.Background()
  314. alice, err := db.Create(
  315. ctx,
  316. "alice",
  317. "alice@example.com",
  318. CreateUserOptions{
  319. Activated: true,
  320. },
  321. )
  322. require.NoError(t, err)
  323. t.Run("name not allowed", func(t *testing.T) {
  324. _, err := db.Create(ctx, "-", "", CreateUserOptions{})
  325. wantErr := ErrNameNotAllowed{
  326. args: errutil.Args{
  327. "reason": "reserved",
  328. "name": "-",
  329. },
  330. }
  331. assert.Equal(t, wantErr, err)
  332. })
  333. t.Run("name already exists", func(t *testing.T) {
  334. _, err := db.Create(ctx, alice.Name, "", CreateUserOptions{})
  335. wantErr := ErrUserAlreadyExist{
  336. args: errutil.Args{
  337. "name": alice.Name,
  338. },
  339. }
  340. assert.Equal(t, wantErr, err)
  341. })
  342. t.Run("email already exists", func(t *testing.T) {
  343. _, err := db.Create(ctx, "bob", alice.Email, CreateUserOptions{})
  344. wantErr := ErrEmailAlreadyUsed{
  345. args: errutil.Args{
  346. "email": alice.Email,
  347. },
  348. }
  349. assert.Equal(t, wantErr, err)
  350. })
  351. user, err := db.GetByUsername(ctx, alice.Name)
  352. require.NoError(t, err)
  353. assert.Equal(t, db.NowFunc().Format(time.RFC3339), user.Created.UTC().Format(time.RFC3339))
  354. assert.Equal(t, db.NowFunc().Format(time.RFC3339), user.Updated.UTC().Format(time.RFC3339))
  355. }
  356. func usersDeleteCustomAvatar(t *testing.T, db *users) {
  357. ctx := context.Background()
  358. alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  359. require.NoError(t, err)
  360. avatar, err := public.Files.ReadFile("img/avatar_default.png")
  361. require.NoError(t, err)
  362. avatarPath := userutil.CustomAvatarPath(alice.ID)
  363. _ = os.Remove(avatarPath)
  364. defer func() { _ = os.Remove(avatarPath) }()
  365. err = db.UseCustomAvatar(ctx, alice.ID, avatar)
  366. require.NoError(t, err)
  367. // Make sure avatar is saved and the user flag is updated.
  368. got := osutil.IsFile(avatarPath)
  369. assert.True(t, got)
  370. alice, err = db.GetByID(ctx, alice.ID)
  371. require.NoError(t, err)
  372. assert.True(t, alice.UseCustomAvatar)
  373. // Delete avatar should remove the file and revert the user flag.
  374. err = db.DeleteCustomAvatar(ctx, alice.ID)
  375. require.NoError(t, err)
  376. got = osutil.IsFile(avatarPath)
  377. assert.False(t, got)
  378. alice, err = db.GetByID(ctx, alice.ID)
  379. require.NoError(t, err)
  380. assert.False(t, alice.UseCustomAvatar)
  381. }
  382. func usersGetByEmail(t *testing.T, db *users) {
  383. ctx := context.Background()
  384. t.Run("empty email", func(t *testing.T) {
  385. _, err := db.GetByEmail(ctx, "")
  386. wantErr := ErrUserNotExist{args: errutil.Args{"email": ""}}
  387. assert.Equal(t, wantErr, err)
  388. })
  389. t.Run("ignore organization", func(t *testing.T) {
  390. // TODO: Use Orgs.Create to replace SQL hack when the method is available.
  391. org, err := db.Create(ctx, "gogs", "gogs@exmaple.com", CreateUserOptions{})
  392. require.NoError(t, err)
  393. err = db.Model(&User{}).Where("id", org.ID).UpdateColumn("type", UserTypeOrganization).Error
  394. require.NoError(t, err)
  395. _, err = db.GetByEmail(ctx, org.Email)
  396. wantErr := ErrUserNotExist{args: errutil.Args{"email": org.Email}}
  397. assert.Equal(t, wantErr, err)
  398. })
  399. t.Run("by primary email", func(t *testing.T) {
  400. alice, err := db.Create(ctx, "alice", "alice@exmaple.com", CreateUserOptions{})
  401. require.NoError(t, err)
  402. _, err = db.GetByEmail(ctx, alice.Email)
  403. wantErr := ErrUserNotExist{args: errutil.Args{"email": alice.Email}}
  404. assert.Equal(t, wantErr, err)
  405. // Mark user as activated
  406. // TODO: Use UserEmails.Verify to replace SQL hack when the method is available.
  407. err = db.Model(&User{}).Where("id", alice.ID).UpdateColumn("is_active", true).Error
  408. require.NoError(t, err)
  409. user, err := db.GetByEmail(ctx, alice.Email)
  410. require.NoError(t, err)
  411. assert.Equal(t, alice.Name, user.Name)
  412. })
  413. t.Run("by secondary email", func(t *testing.T) {
  414. bob, err := db.Create(ctx, "bob", "bob@example.com", CreateUserOptions{})
  415. require.NoError(t, err)
  416. // TODO: Use UserEmails.Create to replace SQL hack when the method is available.
  417. email2 := "bob2@exmaple.com"
  418. err = db.Exec(`INSERT INTO email_address (uid, email) VALUES (?, ?)`, bob.ID, email2).Error
  419. require.NoError(t, err)
  420. _, err = db.GetByEmail(ctx, email2)
  421. wantErr := ErrUserNotExist{args: errutil.Args{"email": email2}}
  422. assert.Equal(t, wantErr, err)
  423. // TODO: Use UserEmails.Verify to replace SQL hack when the method is available.
  424. err = db.Exec(`UPDATE email_address SET is_activated = ? WHERE email = ?`, true, email2).Error
  425. require.NoError(t, err)
  426. user, err := db.GetByEmail(ctx, email2)
  427. require.NoError(t, err)
  428. assert.Equal(t, bob.Name, user.Name)
  429. })
  430. }
  431. func usersGetByID(t *testing.T, db *users) {
  432. ctx := context.Background()
  433. alice, err := db.Create(ctx, "alice", "alice@exmaple.com", CreateUserOptions{})
  434. require.NoError(t, err)
  435. user, err := db.GetByID(ctx, alice.ID)
  436. require.NoError(t, err)
  437. assert.Equal(t, alice.Name, user.Name)
  438. _, err = db.GetByID(ctx, 404)
  439. wantErr := ErrUserNotExist{args: errutil.Args{"userID": int64(404)}}
  440. assert.Equal(t, wantErr, err)
  441. }
  442. func usersGetByUsername(t *testing.T, db *users) {
  443. ctx := context.Background()
  444. alice, err := db.Create(ctx, "alice", "alice@exmaple.com", CreateUserOptions{})
  445. require.NoError(t, err)
  446. user, err := db.GetByUsername(ctx, alice.Name)
  447. require.NoError(t, err)
  448. assert.Equal(t, alice.Name, user.Name)
  449. _, err = db.GetByUsername(ctx, "bad_username")
  450. wantErr := ErrUserNotExist{args: errutil.Args{"name": "bad_username"}}
  451. assert.Equal(t, wantErr, err)
  452. }
  453. func usersHasForkedRepository(t *testing.T, db *users) {
  454. ctx := context.Background()
  455. has := db.HasForkedRepository(ctx, 1, 1)
  456. assert.False(t, has)
  457. _, err := NewReposStore(db.DB).Create(
  458. ctx,
  459. 1,
  460. CreateRepoOptions{
  461. Name: "repo1",
  462. ForkID: 1,
  463. },
  464. )
  465. require.NoError(t, err)
  466. has = db.HasForkedRepository(ctx, 1, 1)
  467. assert.True(t, has)
  468. }
  469. func usersIsUsernameUsed(t *testing.T, db *users) {
  470. ctx := context.Background()
  471. alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  472. require.NoError(t, err)
  473. got := db.IsUsernameUsed(ctx, alice.Name)
  474. assert.True(t, got)
  475. got = db.IsUsernameUsed(ctx, "bob")
  476. assert.False(t, got)
  477. }
  478. func usersList(t *testing.T, db *users) {
  479. ctx := context.Background()
  480. alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  481. require.NoError(t, err)
  482. bob, err := db.Create(ctx, "bob", "bob@example.com", CreateUserOptions{})
  483. require.NoError(t, err)
  484. // Create an organization shouldn't count
  485. // TODO: Use Orgs.Create to replace SQL hack when the method is available.
  486. org1, err := db.Create(ctx, "org1", "org1@example.com", CreateUserOptions{})
  487. require.NoError(t, err)
  488. err = db.Exec(
  489. dbutil.Quote("UPDATE %s SET type = ? WHERE id = ?", "user"),
  490. UserTypeOrganization, org1.ID,
  491. ).Error
  492. require.NoError(t, err)
  493. got, err := db.List(ctx, 1, 1)
  494. require.NoError(t, err)
  495. require.Len(t, got, 1)
  496. assert.Equal(t, alice.ID, got[0].ID)
  497. got, err = db.List(ctx, 2, 1)
  498. require.NoError(t, err)
  499. require.Len(t, got, 1)
  500. assert.Equal(t, bob.ID, got[0].ID)
  501. got, err = db.List(ctx, 1, 3)
  502. require.NoError(t, err)
  503. require.Len(t, got, 2)
  504. assert.Equal(t, alice.ID, got[0].ID)
  505. assert.Equal(t, bob.ID, got[1].ID)
  506. }
  507. func usersListFollowers(t *testing.T, db *users) {
  508. ctx := context.Background()
  509. john, err := db.Create(ctx, "john", "john@example.com", CreateUserOptions{})
  510. require.NoError(t, err)
  511. got, err := db.ListFollowers(ctx, john.ID, 1, 1)
  512. require.NoError(t, err)
  513. assert.Empty(t, got)
  514. alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  515. require.NoError(t, err)
  516. bob, err := db.Create(ctx, "bob", "bob@example.com", CreateUserOptions{})
  517. require.NoError(t, err)
  518. followsStore := NewFollowsStore(db.DB)
  519. err = followsStore.Follow(ctx, alice.ID, john.ID)
  520. require.NoError(t, err)
  521. err = followsStore.Follow(ctx, bob.ID, john.ID)
  522. require.NoError(t, err)
  523. // First page only has bob
  524. got, err = db.ListFollowers(ctx, john.ID, 1, 1)
  525. require.NoError(t, err)
  526. require.Len(t, got, 1)
  527. assert.Equal(t, bob.ID, got[0].ID)
  528. // Second page only has alice
  529. got, err = db.ListFollowers(ctx, john.ID, 2, 1)
  530. require.NoError(t, err)
  531. require.Len(t, got, 1)
  532. assert.Equal(t, alice.ID, got[0].ID)
  533. }
  534. func usersListFollowings(t *testing.T, db *users) {
  535. ctx := context.Background()
  536. john, err := db.Create(ctx, "john", "john@example.com", CreateUserOptions{})
  537. require.NoError(t, err)
  538. got, err := db.ListFollowers(ctx, john.ID, 1, 1)
  539. require.NoError(t, err)
  540. assert.Empty(t, got)
  541. alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  542. require.NoError(t, err)
  543. bob, err := db.Create(ctx, "bob", "bob@example.com", CreateUserOptions{})
  544. require.NoError(t, err)
  545. followsStore := NewFollowsStore(db.DB)
  546. err = followsStore.Follow(ctx, john.ID, alice.ID)
  547. require.NoError(t, err)
  548. err = followsStore.Follow(ctx, john.ID, bob.ID)
  549. require.NoError(t, err)
  550. // First page only has bob
  551. got, err = db.ListFollowings(ctx, john.ID, 1, 1)
  552. require.NoError(t, err)
  553. require.Len(t, got, 1)
  554. assert.Equal(t, bob.ID, got[0].ID)
  555. // Second page only has alice
  556. got, err = db.ListFollowings(ctx, john.ID, 2, 1)
  557. require.NoError(t, err)
  558. require.Len(t, got, 1)
  559. assert.Equal(t, alice.ID, got[0].ID)
  560. }
  561. func usersUseCustomAvatar(t *testing.T, db *users) {
  562. ctx := context.Background()
  563. alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{})
  564. require.NoError(t, err)
  565. avatar, err := public.Files.ReadFile("img/avatar_default.png")
  566. require.NoError(t, err)
  567. avatarPath := userutil.CustomAvatarPath(alice.ID)
  568. _ = os.Remove(avatarPath)
  569. defer func() { _ = os.Remove(avatarPath) }()
  570. err = db.UseCustomAvatar(ctx, alice.ID, avatar)
  571. require.NoError(t, err)
  572. // Make sure avatar is saved and the user flag is updated.
  573. got := osutil.IsFile(avatarPath)
  574. assert.True(t, got)
  575. alice, err = db.GetByID(ctx, alice.ID)
  576. require.NoError(t, err)
  577. assert.True(t, alice.UseCustomAvatar)
  578. }
  579. func TestIsUsernameAllowed(t *testing.T) {
  580. for name := range reservedUsernames {
  581. t.Run(name, func(t *testing.T) {
  582. assert.True(t, IsErrNameNotAllowed(isUsernameAllowed(name)))
  583. })
  584. }
  585. for _, pattern := range reservedUsernamePatterns {
  586. t.Run(pattern, func(t *testing.T) {
  587. username := strings.ReplaceAll(pattern, "*", "alice")
  588. assert.True(t, IsErrNameNotAllowed(isUsernameAllowed(username)))
  589. })
  590. }
  591. }