gitea源码

openid_test.go 1.6KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. // Copyright 2017 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package user_test
  4. import (
  5. "testing"
  6. "code.gitea.io/gitea/models/unittest"
  7. user_model "code.gitea.io/gitea/models/user"
  8. "github.com/stretchr/testify/assert"
  9. "github.com/stretchr/testify/require"
  10. )
  11. func TestGetUserOpenIDs(t *testing.T) {
  12. assert.NoError(t, unittest.PrepareTestDatabase())
  13. oids, err := user_model.GetUserOpenIDs(t.Context(), int64(1))
  14. if assert.NoError(t, err) && assert.Len(t, oids, 2) {
  15. assert.Equal(t, "https://user1.domain1.tld/", oids[0].URI)
  16. assert.False(t, oids[0].Show)
  17. assert.Equal(t, "http://user1.domain2.tld/", oids[1].URI)
  18. assert.True(t, oids[1].Show)
  19. }
  20. oids, err = user_model.GetUserOpenIDs(t.Context(), int64(2))
  21. if assert.NoError(t, err) && assert.Len(t, oids, 1) {
  22. assert.Equal(t, "https://domain1.tld/user2/", oids[0].URI)
  23. assert.True(t, oids[0].Show)
  24. }
  25. }
  26. func TestToggleUserOpenIDVisibility(t *testing.T) {
  27. assert.NoError(t, unittest.PrepareTestDatabase())
  28. oids, err := user_model.GetUserOpenIDs(t.Context(), int64(2))
  29. require.NoError(t, err)
  30. require.Len(t, oids, 1)
  31. assert.True(t, oids[0].Show)
  32. err = user_model.ToggleUserOpenIDVisibility(t.Context(), oids[0].ID)
  33. require.NoError(t, err)
  34. oids, err = user_model.GetUserOpenIDs(t.Context(), int64(2))
  35. require.NoError(t, err)
  36. require.Len(t, oids, 1)
  37. assert.False(t, oids[0].Show)
  38. err = user_model.ToggleUserOpenIDVisibility(t.Context(), oids[0].ID)
  39. require.NoError(t, err)
  40. oids, err = user_model.GetUserOpenIDs(t.Context(), int64(2))
  41. require.NoError(t, err)
  42. if assert.Len(t, oids, 1) {
  43. assert.True(t, oids[0].Show)
  44. }
  45. }