gitea源码

12345678910111213141516171819202122232425262728293031323334353637383940
  1. // Copyright 2020 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package convert
  4. import (
  5. "testing"
  6. "code.gitea.io/gitea/models/unittest"
  7. user_model "code.gitea.io/gitea/models/user"
  8. api "code.gitea.io/gitea/modules/structs"
  9. "github.com/stretchr/testify/assert"
  10. )
  11. func TestUser_ToUser(t *testing.T) {
  12. assert.NoError(t, unittest.PrepareTestDatabase())
  13. user1 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1, IsAdmin: true})
  14. apiUser := toUser(t.Context(), user1, true, true)
  15. assert.True(t, apiUser.IsAdmin)
  16. assert.Contains(t, apiUser.AvatarURL, "://")
  17. user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2, IsAdmin: false})
  18. apiUser = toUser(t.Context(), user2, true, true)
  19. assert.False(t, apiUser.IsAdmin)
  20. apiUser = toUser(t.Context(), user1, false, false)
  21. assert.False(t, apiUser.IsAdmin)
  22. assert.Equal(t, api.VisibleTypePublic.String(), apiUser.Visibility)
  23. user31 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 31, IsAdmin: false, Visibility: api.VisibleTypePrivate})
  24. apiUser = toUser(t.Context(), user31, true, true)
  25. assert.False(t, apiUser.IsAdmin)
  26. assert.Equal(t, api.VisibleTypePrivate.String(), apiUser.Visibility)
  27. }