Explorar o código

db: use `context` and go-mockgen for login sources (#7041)

Joe Chen %!s(int64=2) %!d(string=hai) anos
pai
achega
9776bdc9b8

+ 2 - 2
go.mod

@@ -5,6 +5,7 @@ go 1.16
 require (
 	github.com/Masterminds/semver/v3 v3.1.1
 	github.com/bgentry/speakeasy v0.1.0 // indirect
+	github.com/derision-test/go-mockgen v1.2.0
 	github.com/editorconfig/editorconfig-core-go/v2 v2.4.4
 	github.com/fatih/color v1.9.0 // indirect
 	github.com/go-ldap/ldap/v3 v3.4.3
@@ -52,8 +53,7 @@ require (
 	github.com/unknwon/paginater v0.0.0-20170405233947-45e5d631308e
 	github.com/urfave/cli v1.22.9
 	golang.org/x/crypto v0.0.0-20220331220935-ae2d96664a29
-	golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2
-	golang.org/x/sync v0.0.0-20210220032951-036812b2e83c // indirect
+	golang.org/x/net v0.0.0-20220325170049-de3da57026de
 	golang.org/x/text v0.3.7
 	gopkg.in/DATA-DOG/go-sqlmock.v2 v2.0.0-20180914054222-c19298f520d0
 	gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc // indirect

+ 50 - 5
go.sum

@@ -44,12 +44,14 @@ github.com/Masterminds/semver/v3 v3.1.1/go.mod h1:VPu/7SZ7ePZ3QOrcuXROw5FAcLl4a0
 github.com/Shopify/sarama v1.19.0/go.mod h1:FVkBWblsNy7DGZRfXLU0O9RCGt5g3g3yEuWXgklEdEo=
 github.com/Shopify/toxiproxy v2.1.4+incompatible/go.mod h1:OXgGpZ6Cli1/URJOF1DMxUHB2q5Ap20/P/eIdh4G0pI=
 github.com/alecthomas/chroma/v2 v2.0.1/go.mod h1:vf4zrexSH54oEjJ7EdB65tGNHmH3pGZmVkgTP5RHvAs=
+github.com/alecthomas/kingpin v2.2.6+incompatible/go.mod h1:59OFYbFVLKQKq+mqrL6Rw5bR0c3ACQaawgXx0QYndlE=
 github.com/alecthomas/repr v0.0.0-20220113201626-b1b626ac65ae/go.mod h1:2kn6fqh/zIyPLmm3ugklbEi5hg5wS435eygvNfaDQL8=
 github.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=
 github.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=
 github.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=
 github.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=
 github.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d/go.mod h1:rBZYJk541a8SKzHPHnH3zbiI+7dagKZ0cgpgrD7Fyho=
+github.com/alecthomas/units v0.0.0-20211218093645-b94a6e3cc137/go.mod h1:OMCwj8VM1Kc9e19TLln2VL61YJF0x1XFtfdL4JdbSyE=
 github.com/apache/thrift v0.12.0/go.mod h1:cp2SuWMxlEZw2r+iP2GNCdIi4C1qmUzdZFSVb+bacwQ=
 github.com/aymerick/douceur v0.2.0 h1:Mv+mAeH1Q+n9Fr+oyamOlAkUNPWPlA8PPGR0QAaYuPk=
 github.com/aymerick/douceur v0.2.0/go.mod h1:wlT5vV2O3h55X9m7iVYN0TBM0NH/MmbLnd30/FjWUq4=
@@ -84,12 +86,22 @@ github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d h1:U+s90UTSY
 github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=
 github.com/creack/pty v1.1.7/go.mod h1:lj5s0c3V2DBrqTV7llrYr5NG6My20zk30Fl46Y7DoTY=
 github.com/cupcake/rdb v0.0.0-20161107195141-43ba34106c76/go.mod h1:vYwsqCOLxGiisLwp9rITslkFNpZD5rz43tf41QFkTWY=
+github.com/dave/astrid v0.0.0-20170323122508-8c2895878b14/go.mod h1:Sth2QfxfATb/nW4EsrSi2KyJmbcniZ8TgTaji17D6ms=
+github.com/dave/brenda v1.1.0/go.mod h1:4wCUr6gSlu5/1Tk7akE5X7UorwiQ8Rij0SKH3/BGMOM=
+github.com/dave/courtney v0.3.0/go.mod h1:BAv3hA06AYfNUjfjQr+5gc6vxeBVOupLqrColj+QSD8=
+github.com/dave/gopackages v0.0.0-20170318123100-46e7023ec56e/go.mod h1:i00+b/gKdIDIxuLDFob7ustLAVqhsZRk2qVZrArELGQ=
+github.com/dave/jennifer v1.5.0/go.mod h1:4MnyiFIlZS3l5tSDn8VnzE6ffAhYBMB2SZntBsZGUok=
+github.com/dave/kerr v0.0.0-20170318121727-bc25dd6abe8e/go.mod h1:qZqlPyPvfsDJt+3wHJ1EvSXDuVjFTK0j2p/ca+gtsb8=
+github.com/dave/patsy v0.0.0-20210517141501-957256f50cba/go.mod h1:qfR88CgEGLoiqDaE+xxDCi5QA5v4vUoW0UCX2Nd5Tlc=
+github.com/dave/rebecca v0.9.1/go.mod h1:N6XYdMD/OKw3lkF3ywh8Z6wPGuwNFDNtWYEMFWEmXBA=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/denisenkom/go-mssqldb v0.0.0-20190707035753-2be1aa521ff4/go.mod h1:zAg7JM8CkOJ43xKXIj7eRO9kmWm/TW578qo+oDO6tuM=
 github.com/denisenkom/go-mssqldb v0.12.0 h1:VtrkII767ttSPNRfFekePK3sctr+joXgO58stqQbtUA=
 github.com/denisenkom/go-mssqldb v0.12.0/go.mod h1:iiK0YP1ZeepvmBQk/QpLEhhTNJgfzrpArPY/aFvc9yU=
+github.com/derision-test/go-mockgen v1.2.0 h1:SCF7p4FuO6IZId3CMjSHH9K0SMDVXNz7jc3AOeJaBd8=
+github.com/derision-test/go-mockgen v1.2.0/go.mod h1:/TXUePlhtHmDDCaDAi/a4g6xOHqMDz3Wf0r2NPGskB4=
 github.com/dlclark/regexp2 v1.4.0/go.mod h1:2pZnwuY/m+8K6iRw6wQdMtk+rH5tNGR1i55kozfMjCc=
 github.com/dnaeon/go-vcr v1.2.0/go.mod h1:R4UdLID7HZT3taECzJs4YgbbH6PIGXB6W/sc5OLb6RQ=
 github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=
@@ -108,6 +120,7 @@ github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5Kwzbycv
 github.com/fatih/color v1.9.0 h1:8xPHl4/q1VyqGIPif1F+1V3Y3lSmrq01EabUW3CoW5s=
 github.com/fatih/color v1.9.0/go.mod h1:eQcE1qtQxscV5RaZvpXrrb8Drkc3/DdQ+uUYCNjL+zU=
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
+github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=
 github.com/go-asn1-ber/asn1-ber v1.5.4 h1:vXT6d/FNDiELJnLb6hGNa309LMsrCoYFvpwHDF0+Y1A=
 github.com/go-asn1-ber/asn1-ber v1.5.4/go.mod h1:hEBeB/ic+5LoWskz+yKT7vGhhPYkProFKoKdwZRWMe0=
 github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU=
@@ -143,6 +156,7 @@ github.com/go-sql-driver/mysql v1.4.1/go.mod h1:zAC/RDZ24gD3HViQzih4MyKcchzm+sOG
 github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
 github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
 github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
+github.com/go-task/slim-sprig v0.0.0-20210107165309-348f09dbbbc0/go.mod h1:fyg7847qk6SyHyPtNmDHnmrv/HOrqktSC+C9fM+CJOE=
 github.com/go-xorm/sqlfiddle v0.0.0-20180821085327-62ce714f951a h1:9wScpmSP5A3Bk8V3XHWUcJmYTh+ZnlHVyc+A4oZYS3Y=
 github.com/go-xorm/sqlfiddle v0.0.0-20180821085327-62ce714f951a/go.mod h1:56xuuqnHyryaerycW3BfssRdxQstACi0Epw/yC5E2xM=
 github.com/gofrs/uuid v4.0.0+incompatible h1:1SD/1F5pU8p29ybwgQSwpQk+mwdRrXCYuPhW6m+TnJw=
@@ -225,6 +239,7 @@ github.com/google/pprof v0.0.0-20200212024743-f11f1df84d12/go.mod h1:ZgVRPoUq/hf
 github.com/google/pprof v0.0.0-20200229191704-1ebb73c60ed3/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=
 github.com/google/pprof v0.0.0-20200430221834-fc25d7d30c6d/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=
 github.com/google/pprof v0.0.0-20200708004538-1a94d8640e99/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=
+github.com/google/pprof v0.0.0-20210407192527-94a9f03dee38/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=
 github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=
 github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
 github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
@@ -242,6 +257,7 @@ github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ
 github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
 github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
 github.com/ianlancetaylor/demangle v0.0.0-20181102032728-5e5cf60278f6/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc=
+github.com/ianlancetaylor/demangle v0.0.0-20200824232613-28f6c0f3b639/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc=
 github.com/issue9/assert/v2 v2.0.0 h1:vN7fr70g5ND6zM39tPZk/E4WCyjGMqApmFbujSTmEo0=
 github.com/issue9/assert/v2 v2.0.0/go.mod h1:rKr1eVGzXUhAo2af1thiKAhIA8uiSK9Wyn7mcZ4BzAg=
 github.com/issue9/identicon v1.2.1 h1:9RUq3DcmDJvfXAYZWJDaq/Bi45oS/Fr79W0CazbXNaY=
@@ -365,6 +381,7 @@ github.com/mcuadros/go-version v0.0.0-20190830083331-035f6764e8d2 h1:YocNLcTBdEd
 github.com/mcuadros/go-version v0.0.0-20190830083331-035f6764e8d2/go.mod h1:76rfSfYPWj01Z85hUf/ituArm797mNKcvINh1OlsZKo=
 github.com/microcosm-cc/bluemonday v1.0.18 h1:6HcxvXDAi3ARt3slx6nTesbvorIc3QeTzBNRvWktHBo=
 github.com/microcosm-cc/bluemonday v1.0.18/go.mod h1:Z0r70sCuXHig8YpBzCc5eGHAap2K7e/u082ZUpDRRqM=
+github.com/mitchellh/go-wordwrap v1.0.1/go.mod h1:R62XHJLzvMFRBbcrT7m7WgmE1eOyTSsCt+hzestvNj0=
 github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
 github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
 github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
@@ -381,13 +398,22 @@ github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 h1:zYyBkD/k9seD2A7fsi6
 github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8=
 github.com/niklasfasching/go-org v1.6.5 h1:5YAIqNTdl6lAOb7lD2AyQ1RuFGPVrAKvUexphk8PGbo=
 github.com/niklasfasching/go-org v1.6.5/go.mod h1:ybv0eGDnxylFUfFE+ySaQc734j/L3+/ChKZ/h63a2wM=
+github.com/nxadm/tail v1.4.4/go.mod h1:kenIhsEOeOJmVchQTgglprH7qJGnHDVpk1VPCcaMI8A=
+github.com/nxadm/tail v1.4.8/go.mod h1:+ncqLTQzXmGhMZNUePPaPqPvBxHAIsmXswZKocGu+AU=
 github.com/olekukonko/tablewriter v0.0.5 h1:P2Ga83D34wi1o9J6Wh1mRuqd4mF/x/lgBS7N7AbDhec=
 github.com/olekukonko/tablewriter v0.0.5/go.mod h1:hPp6KlRPjbx+hW8ykQs1w3UBbZlj6HuIJcUGPhkA7kY=
 github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
 github.com/onsi/ginkgo v1.7.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
 github.com/onsi/ginkgo v1.8.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/ginkgo v1.12.1/go.mod h1:zj2OWP4+oCPe1qIXoGWkgMRwljMUYCdkwsT2108oapk=
+github.com/onsi/ginkgo v1.16.4/go.mod h1:dX+/inL/fNMqNlz0e9LfyB9TswhZpCVdJM/Z6Vvnwo0=
+github.com/onsi/ginkgo/v2 v2.1.3/go.mod h1:vw5CSIxN1JObi/U8gcbwft7ZxR2dgaR70JSE3/PpL4c=
 github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
 github.com/onsi/gomega v1.5.0/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
+github.com/onsi/gomega v1.7.1/go.mod h1:XdKZgCCFLUoM/7CFJVPcG8C1xQ1AJ0vpAezJrB7JYyY=
+github.com/onsi/gomega v1.10.1/go.mod h1:iN09h71vgCQne3DLsj+A5owkum+a2tYe+TOCB1ybHNo=
+github.com/onsi/gomega v1.17.0/go.mod h1:HnhC7FXeEQY45zxNK3PPoIUhzk/80Xly9PcubAlGdZY=
+github.com/onsi/gomega v1.19.0/go.mod h1:LY+I3pBVzYsTBU1AnDwOSxaYi9WoWiqgwooUqq9yPro=
 github.com/openzipkin/zipkin-go v0.1.6/go.mod h1:QgAqvLzwWbR/WpD4A3cGpPtJrZXNIiJc5AZX7/PBEpw=
 github.com/pelletier/go-toml v1.4.0/go.mod h1:PN7xzY2wHTK0K9p34ErDQMlFxa51Fk0OUruD3k1mMwo=
 github.com/pierrec/lz4 v2.0.5+incompatible/go.mod h1:pdkljMzZIN41W+lC3N2tnIh5sFi+IEE17M5jbnwPHcY=
@@ -474,6 +500,7 @@ github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UV
 github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
 github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.7.2 h1:4jaiDzPyXQvSd7D0EjG45355tLlV3VOECpq10pLC+8s=
 github.com/stretchr/testify v1.7.2/go.mod h1:R6va5+xMeoiuVRoj+gSkQ7d3FALtqAAGI1FQKckRals=
 github.com/syndtr/goleveldb v1.0.0/go.mod h1:ZVVdQEZoIme9iO1Ch2Jdy24qqXrMMOU6lpPAyBWyWuQ=
@@ -492,6 +519,7 @@ github.com/yuin/goldmark v1.1.25/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9de
 github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.32/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
+github.com/yuin/goldmark v1.4.1/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
 github.com/zenazn/goji v0.9.0/go.mod h1:7S9M489iMyHBNxwZnk9/EHS098H4/F6TATF2mIxtB1Q=
 github.com/ziutek/mymysql v1.5.4 h1:GB0qdRGsTwQSBVYuVShFBKaXSnSnYYC2d9knnE1LHFs=
 github.com/ziutek/mymysql v1.5.4/go.mod h1:LMSpPZ6DbqWFxNCHW77HeMg9I646SAhApZ/wKdgO/C0=
@@ -559,8 +587,9 @@ golang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzB
 golang.org/x/mod v0.1.1-0.20191107180719-034126e5016b/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=
 golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
 golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
-golang.org/x/mod v0.5.1 h1:OJxoQ/rynoF0dcCdI7cLPktw/hR2cueqYfjm43oqK38=
 golang.org/x/mod v0.5.1/go.mod h1:5OXOZSfqPIIbmVBIIKWRFfZjPR0E5r58TLhUjH0a2Ro=
+golang.org/x/mod v0.6.0-dev.0.20220106191415-9b9b3d81d5e3 h1:kQgndtyPBW/JIYERgdxfwMYh3AVStj88WQTlNDi2a+o=
+golang.org/x/mod v0.6.0-dev.0.20220106191415-9b9b3d81d5e3/go.mod h1:3p9vT2HGsQu2K1YbXdKPJLVgG5VJdoTa1poYQBtP1AY=
 golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
@@ -588,6 +617,7 @@ golang.org/x/net v0.0.0-20200324143707-d3edc9973b7e/go.mod h1:qpuaurCH72eLCgpAm/
 golang.org/x/net v0.0.0-20200501053045-e0ff5e5a1de5/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=
 golang.org/x/net v0.0.0-20200506145744-7e3656a0809f/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=
 golang.org/x/net v0.0.0-20200513185701-a91f0712d120/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=
+golang.org/x/net v0.0.0-20200520004742-59133d7f0dd7/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=
 golang.org/x/net v0.0.0-20200520182314-0ba52f642ac2/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=
 golang.org/x/net v0.0.0-20200625001655-4c5254603344/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
 golang.org/x/net v0.0.0-20200707034311-ab3426394381/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
@@ -595,11 +625,15 @@ golang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81R
 golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
 golang.org/x/net v0.0.0-20201224014010-6772e930b67b/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
 golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
+golang.org/x/net v0.0.0-20210428140749-89ef3d95e781/go.mod h1:OJAsFXCWl8Ukc7SiCT/9KSuxbyM7479/AVlXFRxuMCk=
 golang.org/x/net v0.0.0-20210525063256-abc453219eb5/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
 golang.org/x/net v0.0.0-20210610132358-84b48f89b13b/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
 golang.org/x/net v0.0.0-20210614182718-04defd469f4e/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
-golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2 h1:CIJ76btIcR3eFI5EgSo6k1qKw9KJexJuRLI9G7Hp5wE=
+golang.org/x/net v0.0.0-20211015210444-4f30a5c0130f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
 golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
+golang.org/x/net v0.0.0-20220225172249-27dd8689420f/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=
+golang.org/x/net v0.0.0-20220325170049-de3da57026de h1:pZB1TWnKi+o4bENlbzAgLrEbY4RMYmUIRobMcSmfeYc=
+golang.org/x/net v0.0.0-20220325170049-de3da57026de/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=
 golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
 golang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
 golang.org/x/oauth2 v0.0.0-20190604053449-0f29369cfe45/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
@@ -636,9 +670,12 @@ golang.org/x/sys v0.0.0-20190624142023-c5567b49c5d0/go.mod h1:h1NjWce9XRLGQEsW7w
 golang.org/x/sys v0.0.0-20190726091711-fc99dfbffb4e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20190804053845-51ab0e2deafa/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20190813064441-fde4db37ae7a/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20190904154756-749cb33beabd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191001151750-bb3f8db39f24/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20191005200804-aed5e4c7ecf9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191008105621-543471e840be/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20191120155948-bd437916bb0e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191204072324-ce4227a45e2e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191228213918-04cbcbbfeed8/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20200106162015-b016eb3dc98e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
@@ -660,16 +697,21 @@ golang.org/x/sys v0.0.0-20200625212154-ddb9806d33ae/go.mod h1:h1NjWce9XRLGQEsW7w
 golang.org/x/sys v0.0.0-20200803210538-64077c9b5642/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210112080510-489259a85091/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210603081109-ebe580a85c40/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20211007075335-d3039528d8ac/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20220114195835-da31bd327af9 h1:XfKQ4OlFl8okEOr5UvAqFRVj8pY/4yfcXrddB8qAbU0=
+golang.org/x/sys v0.0.0-20211019181941-9d821ace8654/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220114195835-da31bd327af9/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220328115105-d36c6a25d886 h1:eJv7u3ksNXoLbGSKuv2s/SIO4tJVxc/A+MTpzxDgz/Q=
+golang.org/x/sys v0.0.0-20220328115105-d36c6a25d886/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXRh80K7BxOlk5/8JxuGnuhpl+muw=
-golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1 h1:v+OssWQX+hTHEmOBgwxdZxK4zHq3yOs8F9J7mk0PY8E=
 golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/term v0.0.0-20210927222741-03fcf44c2211 h1:JGgROgKl9N8DuW20oFS5gxc+lE67/N3FcwmBPMe7ArY=
+golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
 golang.org/x/text v0.0.0-20170915032832-14c0d48ead0c/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.1-0.20180807135948-17ff2d5776d2/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
@@ -730,8 +772,11 @@ golang.org/x/tools v0.0.0-20200618134242-20370b0cb4b2/go.mod h1:EkVYQZoAsY45+roY
 golang.org/x/tools v0.0.0-20200729194436-6467de6f59a7/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=
 golang.org/x/tools v0.0.0-20200804011535-6c149bb5ef0d/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=
 golang.org/x/tools v0.0.0-20200825202427-b303f430e36d/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=
-golang.org/x/tools v0.0.0-20201124115921-2c860bdd6e78 h1:M8tBwCtWD/cZV9DZpFYRUgaymAYAr+aIUTWzDaM3uPs=
 golang.org/x/tools v0.0.0-20201124115921-2c860bdd6e78/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
+golang.org/x/tools v0.0.0-20201224043029-2b0845dc783e/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
+golang.org/x/tools v0.1.8/go.mod h1:nABZi5QlRsZVlzPpHl034qft6wpY4eDcsTt5AaioBiU=
+golang.org/x/tools v0.1.10 h1:QjFRCZxdOhBJ/UNgnBZLbNV13DlbnK0quyivTnXJM20=
+golang.org/x/tools v0.1.10/go.mod h1:Uh6Zz+xoGYZom868N8YTex3t7RhtHDBrE8Gzo9bV56E=
 golang.org/x/xerrors v0.0.0-20190410155217-1f06c39b4373/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20190513163551-3ee3066db522/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=

+ 6 - 7
internal/db/login_source_files_test.go

@@ -9,11 +9,12 @@ import (
 	"time"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 
 	"gogs.io/gogs/internal/errutil"
 )
 
-func Test_loginSourceFiles_GetByID(t *testing.T) {
+func TestLoginSourceFiles_GetByID(t *testing.T) {
 	store := &loginSourceFiles{
 		sources: []*LoginSource{
 			{ID: 101},
@@ -28,14 +29,12 @@ func Test_loginSourceFiles_GetByID(t *testing.T) {
 
 	t.Run("source exists", func(t *testing.T) {
 		source, err := store.GetByID(101)
-		if err != nil {
-			t.Fatal(err)
-		}
+		require.NoError(t, err)
 		assert.Equal(t, int64(101), source.ID)
 	})
 }
 
-func Test_loginSourceFiles_Len(t *testing.T) {
+func TestLoginSourceFiles_Len(t *testing.T) {
 	store := &loginSourceFiles{
 		sources: []*LoginSource{
 			{ID: 101},
@@ -45,7 +44,7 @@ func Test_loginSourceFiles_Len(t *testing.T) {
 	assert.Equal(t, 1, store.Len())
 }
 
-func Test_loginSourceFiles_List(t *testing.T) {
+func TestLoginSourceFiles_List(t *testing.T) {
 	store := &loginSourceFiles{
 		sources: []*LoginSource{
 			{ID: 101, IsActived: true},
@@ -65,7 +64,7 @@ func Test_loginSourceFiles_List(t *testing.T) {
 	})
 }
 
-func Test_loginSourceFiles_Update(t *testing.T) {
+func TestLoginSourceFiles_Update(t *testing.T) {
 	store := &loginSourceFiles{
 		sources: []*LoginSource{
 			{ID: 101, IsActived: true, IsDefault: true},

+ 40 - 35
internal/db/login_sources.go

@@ -5,6 +5,7 @@
 package db
 
 import (
+	"context"
 	"fmt"
 	"strconv"
 	"time"
@@ -25,24 +26,24 @@ import (
 //
 // NOTE: All methods are sorted in alphabetical order.
 type LoginSourcesStore interface {
-	// Create creates a new login source and persist to database.
-	// It returns ErrLoginSourceAlreadyExist when a login source with same name already exists.
-	Create(opts CreateLoginSourceOpts) (*LoginSource, error)
+	// Create creates a new login source and persist to database. It returns
+	// ErrLoginSourceAlreadyExist when a login source with same name already exists.
+	Create(ctx context.Context, opts CreateLoginSourceOpts) (*LoginSource, error)
 	// Count returns the total number of login sources.
-	Count() int64
-	// DeleteByID deletes a login source by given ID.
-	// It returns ErrLoginSourceInUse if at least one user is associated with the login source.
-	DeleteByID(id int64) error
-	// GetByID returns the login source with given ID.
-	// It returns ErrLoginSourceNotExist when not found.
-	GetByID(id int64) (*LoginSource, error)
+	Count(ctx context.Context) int64
+	// DeleteByID deletes a login source by given ID. It returns ErrLoginSourceInUse
+	// if at least one user is associated with the login source.
+	DeleteByID(ctx context.Context, id int64) error
+	// GetByID returns the login source with given ID. It returns
+	// ErrLoginSourceNotExist when not found.
+	GetByID(ctx context.Context, id int64) (*LoginSource, error)
 	// List returns a list of login sources filtered by options.
-	List(opts ListLoginSourceOpts) ([]*LoginSource, error)
+	List(ctx context.Context, opts ListLoginSourceOpts) ([]*LoginSource, error)
 	// ResetNonDefault clears default flag for all the other login sources.
-	ResetNonDefault(source *LoginSource) error
-	// Save persists all values of given login source to database or local file.
-	// The Updated field is set to current time automatically.
-	Save(t *LoginSource) error
+	ResetNonDefault(ctx context.Context, source *LoginSource) error
+	// Save persists all values of given login source to database or local file. The
+	// Updated field is set to current time automatically.
+	Save(ctx context.Context, t *LoginSource) error
 }
 
 var LoginSources LoginSourcesStore
@@ -65,7 +66,7 @@ type LoginSource struct {
 	File loginSourceFileStore `xorm:"-" gorm:"-" json:"-"`
 }
 
-// NOTE: This is a GORM save hook.
+// BeforeSave implements the GORM save hook.
 func (s *LoginSource) BeforeSave(_ *gorm.DB) (err error) {
 	if s.Provider == nil {
 		return nil
@@ -74,7 +75,7 @@ func (s *LoginSource) BeforeSave(_ *gorm.DB) (err error) {
 	return err
 }
 
-// NOTE: This is a GORM create hook.
+// BeforeCreate implements the GORM create hook.
 func (s *LoginSource) BeforeCreate(tx *gorm.DB) error {
 	if s.CreatedUnix == 0 {
 		s.CreatedUnix = tx.NowFunc().Unix()
@@ -83,13 +84,13 @@ func (s *LoginSource) BeforeCreate(tx *gorm.DB) error {
 	return nil
 }
 
-// NOTE: This is a GORM update hook.
+// BeforeUpdate implements the GORM update hook.
 func (s *LoginSource) BeforeUpdate(tx *gorm.DB) error {
 	s.UpdatedUnix = tx.NowFunc().Unix()
 	return nil
 }
 
-// NOTE: This is a GORM query hook.
+// AfterFind implements the GORM query hook.
 func (s *LoginSource) AfterFind(_ *gorm.DB) error {
 	s.Created = time.Unix(s.CreatedUnix, 0).Local()
 	s.Updated = time.Unix(s.UpdatedUnix, 0).Local()
@@ -209,8 +210,8 @@ func (err ErrLoginSourceAlreadyExist) Error() string {
 	return fmt.Sprintf("login source already exists: %v", err.args)
 }
 
-func (db *loginSources) Create(opts CreateLoginSourceOpts) (*LoginSource, error) {
-	err := db.Where("name = ?", opts.Name).First(new(LoginSource)).Error
+func (db *loginSources) Create(ctx context.Context, opts CreateLoginSourceOpts) (*LoginSource, error) {
+	err := db.WithContext(ctx).Where("name = ?", opts.Name).First(new(LoginSource)).Error
 	if err == nil {
 		return nil, ErrLoginSourceAlreadyExist{args: errutil.Args{"name": opts.Name}}
 	} else if err != gorm.ErrRecordNotFound {
@@ -227,12 +228,12 @@ func (db *loginSources) Create(opts CreateLoginSourceOpts) (*LoginSource, error)
 	if err != nil {
 		return nil, err
 	}
-	return source, db.DB.Create(source).Error
+	return source, db.WithContext(ctx).Create(source).Error
 }
 
-func (db *loginSources) Count() int64 {
+func (db *loginSources) Count(ctx context.Context) int64 {
 	var count int64
-	db.Model(new(LoginSource)).Count(&count)
+	db.WithContext(ctx).Model(new(LoginSource)).Count(&count)
 	return count + int64(db.files.Len())
 }
 
@@ -249,21 +250,21 @@ func (err ErrLoginSourceInUse) Error() string {
 	return fmt.Sprintf("login source is still used by some users: %v", err.args)
 }
 
-func (db *loginSources) DeleteByID(id int64) error {
+func (db *loginSources) DeleteByID(ctx context.Context, id int64) error {
 	var count int64
-	err := db.Model(new(User)).Where("login_source = ?", id).Count(&count).Error
+	err := db.WithContext(ctx).Model(new(User)).Where("login_source = ?", id).Count(&count).Error
 	if err != nil {
 		return err
 	} else if count > 0 {
 		return ErrLoginSourceInUse{args: errutil.Args{"id": id}}
 	}
 
-	return db.Where("id = ?", id).Delete(new(LoginSource)).Error
+	return db.WithContext(ctx).Where("id = ?", id).Delete(new(LoginSource)).Error
 }
 
-func (db *loginSources) GetByID(id int64) (*LoginSource, error) {
+func (db *loginSources) GetByID(ctx context.Context, id int64) (*LoginSource, error) {
 	source := new(LoginSource)
-	err := db.Where("id = ?", id).First(source).Error
+	err := db.WithContext(ctx).Where("id = ?", id).First(source).Error
 	if err != nil {
 		if err == gorm.ErrRecordNotFound {
 			return db.files.GetByID(id)
@@ -278,9 +279,9 @@ type ListLoginSourceOpts struct {
 	OnlyActivated bool
 }
 
-func (db *loginSources) List(opts ListLoginSourceOpts) ([]*LoginSource, error) {
+func (db *loginSources) List(ctx context.Context, opts ListLoginSourceOpts) ([]*LoginSource, error) {
 	var sources []*LoginSource
-	query := db.Order("id ASC")
+	query := db.WithContext(ctx).Order("id ASC")
 	if opts.OnlyActivated {
 		query = query.Where("is_actived = ?", true)
 	}
@@ -292,8 +293,12 @@ func (db *loginSources) List(opts ListLoginSourceOpts) ([]*LoginSource, error) {
 	return append(sources, db.files.List(opts)...), nil
 }
 
-func (db *loginSources) ResetNonDefault(dflt *LoginSource) error {
-	err := db.Model(new(LoginSource)).Where("id != ?", dflt.ID).Updates(map[string]interface{}{"is_default": false}).Error
+func (db *loginSources) ResetNonDefault(ctx context.Context, dflt *LoginSource) error {
+	err := db.WithContext(ctx).
+		Model(new(LoginSource)).
+		Where("id != ?", dflt.ID).
+		Updates(map[string]interface{}{"is_default": false}).
+		Error
 	if err != nil {
 		return err
 	}
@@ -311,9 +316,9 @@ func (db *loginSources) ResetNonDefault(dflt *LoginSource) error {
 	return nil
 }
 
-func (db *loginSources) Save(source *LoginSource) error {
+func (db *loginSources) Save(ctx context.Context, source *LoginSource) error {
 	if source.File == nil {
-		return db.DB.Save(source).Error
+		return db.WithContext(ctx).Save(source).Error
 	}
 
 	source.File.SetGeneral("name", source.Name)

+ 211 - 243
internal/db/login_sources_test.go

@@ -5,10 +5,13 @@
 package db
 
 import (
+	"context"
 	"testing"
 	"time"
 
+	mockrequire "github.com/derision-test/go-mockgen/testutil/require"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 	"gorm.io/gorm"
 
 	"gogs.io/gogs/internal/auth"
@@ -31,9 +34,7 @@ func TestLoginSource_BeforeSave(t *testing.T) {
 	t.Run("Config has not been set", func(t *testing.T) {
 		s := &LoginSource{}
 		err := s.BeforeSave(db)
-		if err != nil {
-			t.Fatal(err)
-		}
+		require.NoError(t, err)
 		assert.Empty(t, s.Config)
 	})
 
@@ -44,9 +45,7 @@ func TestLoginSource_BeforeSave(t *testing.T) {
 			}),
 		}
 		err := s.BeforeSave(db)
-		if err != nil {
-			t.Fatal(err)
-		}
+		require.NoError(t, err)
 		assert.Equal(t, `{"ServiceName":"pam_service"}`, s.Config)
 	})
 }
@@ -93,20 +92,18 @@ func Test_loginSources(t *testing.T) {
 		name string
 		test func(*testing.T, *loginSources)
 	}{
-		{"Create", test_loginSources_Create},
-		{"Count", test_loginSources_Count},
-		{"DeleteByID", test_loginSources_DeleteByID},
-		{"GetByID", test_loginSources_GetByID},
-		{"List", test_loginSources_List},
-		{"ResetNonDefault", test_loginSources_ResetNonDefault},
-		{"Save", test_loginSources_Save},
+		{"Create", loginSourcesCreate},
+		{"Count", loginSourcesCount},
+		{"DeleteByID", loginSourcesDeleteByID},
+		{"GetByID", loginSourcesGetByID},
+		{"List", loginSourcesList},
+		{"ResetNonDefault", loginSourcesResetNonDefault},
+		{"Save", loginSourcesSave},
 	} {
 		t.Run(tc.name, func(t *testing.T) {
 			t.Cleanup(func() {
 				err := clearTables(t, db.DB, tables...)
-				if err != nil {
-					t.Fatal(err)
-				}
+				require.NoError(t, err)
 			})
 			tc.test(t, db)
 		})
@@ -116,62 +113,41 @@ func Test_loginSources(t *testing.T) {
 	}
 }
 
-func test_loginSources_Create(t *testing.T, db *loginSources) {
+func loginSourcesCreate(t *testing.T, db *loginSources) {
+	ctx := context.Background()
+
 	// Create first login source with name "GitHub"
-	source, err := db.Create(CreateLoginSourceOpts{
-		Type:      auth.GitHub,
-		Name:      "GitHub",
-		Activated: true,
-		Default:   false,
-		Config: &github.Config{
-			APIEndpoint: "https://api.github.com",
+	source, err := db.Create(ctx,
+		CreateLoginSourceOpts{
+			Type:      auth.GitHub,
+			Name:      "GitHub",
+			Activated: true,
+			Default:   false,
+			Config: &github.Config{
+				APIEndpoint: "https://api.github.com",
+			},
 		},
-	})
-	if err != nil {
-		t.Fatal(err)
-	}
+	)
+	require.NoError(t, err)
 
 	// Get it back and check the Created field
-	source, err = db.GetByID(source.ID)
-	if err != nil {
-		t.Fatal(err)
-	}
+	source, err = db.GetByID(ctx, source.ID)
+	require.NoError(t, err)
 	assert.Equal(t, db.NowFunc().Format(time.RFC3339), source.Created.UTC().Format(time.RFC3339))
 	assert.Equal(t, db.NowFunc().Format(time.RFC3339), source.Updated.UTC().Format(time.RFC3339))
 
 	// Try create second login source with same name should fail
-	_, err = db.Create(CreateLoginSourceOpts{Name: source.Name})
+	_, err = db.Create(ctx, CreateLoginSourceOpts{Name: source.Name})
 	expErr := ErrLoginSourceAlreadyExist{args: errutil.Args{"name": source.Name}}
 	assert.Equal(t, expErr, err)
 }
 
-func test_loginSources_Count(t *testing.T, db *loginSources) {
-	// Create two login sources, one in database and one as source file.
-	_, err := db.Create(CreateLoginSourceOpts{
-		Type:      auth.GitHub,
-		Name:      "GitHub",
-		Activated: true,
-		Default:   false,
-		Config: &github.Config{
-			APIEndpoint: "https://api.github.com",
-		},
-	})
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	setMockLoginSourceFilesStore(t, db, &mockLoginSourceFilesStore{
-		MockLen: func() int {
-			return 2
-		},
-	})
-
-	assert.Equal(t, int64(3), db.Count())
-}
+func loginSourcesCount(t *testing.T, db *loginSources) {
+	ctx := context.Background()
 
-func test_loginSources_DeleteByID(t *testing.T, db *loginSources) {
-	t.Run("delete but in used", func(t *testing.T) {
-		source, err := db.Create(CreateLoginSourceOpts{
+	// Create two login sources, one in database and one as source file.
+	_, err := db.Create(ctx,
+		CreateLoginSourceOpts{
 			Type:      auth.GitHub,
 			Name:      "GitHub",
 			Activated: true,
@@ -179,279 +155,271 @@ func test_loginSources_DeleteByID(t *testing.T, db *loginSources) {
 			Config: &github.Config{
 				APIEndpoint: "https://api.github.com",
 			},
-		})
-		if err != nil {
-			t.Fatal(err)
-		}
+		},
+	)
+	require.NoError(t, err)
+
+	mock := NewMockLoginSourceFilesStore()
+	mock.LenFunc.SetDefaultReturn(2)
+	setMockLoginSourceFilesStore(t, db, mock)
+
+	assert.Equal(t, int64(3), db.Count(ctx))
+}
+
+func loginSourcesDeleteByID(t *testing.T, db *loginSources) {
+	ctx := context.Background()
+
+	t.Run("delete but in used", func(t *testing.T) {
+		source, err := db.Create(ctx,
+			CreateLoginSourceOpts{
+				Type:      auth.GitHub,
+				Name:      "GitHub",
+				Activated: true,
+				Default:   false,
+				Config: &github.Config{
+					APIEndpoint: "https://api.github.com",
+				},
+			},
+		)
+		require.NoError(t, err)
 
 		// Create a user that uses this login source
 		_, err = (&users{DB: db.DB}).Create("alice", "", CreateUserOpts{
 			LoginSource: source.ID,
 		})
-		if err != nil {
-			t.Fatal(err)
-		}
+		require.NoError(t, err)
 
 		// Delete the login source will result in error
-		err = db.DeleteByID(source.ID)
+		err = db.DeleteByID(ctx, source.ID)
 		expErr := ErrLoginSourceInUse{args: errutil.Args{"id": source.ID}}
 		assert.Equal(t, expErr, err)
 	})
 
-	setMockLoginSourceFilesStore(t, db, &mockLoginSourceFilesStore{
-		MockGetByID: func(id int64) (*LoginSource, error) {
-			return nil, ErrLoginSourceNotExist{args: errutil.Args{"id": id}}
-		},
+	mock := NewMockLoginSourceFilesStore()
+	mock.GetByIDFunc.SetDefaultHook(func(id int64) (*LoginSource, error) {
+		return nil, ErrLoginSourceNotExist{args: errutil.Args{"id": id}}
 	})
+	setMockLoginSourceFilesStore(t, db, mock)
 
 	// Create a login source with name "GitHub2"
-	source, err := db.Create(CreateLoginSourceOpts{
-		Type:      auth.GitHub,
-		Name:      "GitHub2",
-		Activated: true,
-		Default:   false,
-		Config: &github.Config{
-			APIEndpoint: "https://api.github.com",
+	source, err := db.Create(ctx,
+		CreateLoginSourceOpts{
+			Type:      auth.GitHub,
+			Name:      "GitHub2",
+			Activated: true,
+			Default:   false,
+			Config: &github.Config{
+				APIEndpoint: "https://api.github.com",
+			},
 		},
-	})
-	if err != nil {
-		t.Fatal(err)
-	}
+	)
+	require.NoError(t, err)
 
 	// Delete a non-existent ID is noop
-	err = db.DeleteByID(9999)
-	if err != nil {
-		t.Fatal(err)
-	}
+	err = db.DeleteByID(ctx, 9999)
+	require.NoError(t, err)
 
 	// We should be able to get it back
-	_, err = db.GetByID(source.ID)
-	if err != nil {
-		t.Fatal(err)
-	}
+	_, err = db.GetByID(ctx, source.ID)
+	require.NoError(t, err)
 
 	// Now delete this login source with ID
-	err = db.DeleteByID(source.ID)
-	if err != nil {
-		t.Fatal(err)
-	}
+	err = db.DeleteByID(ctx, source.ID)
+	require.NoError(t, err)
 
 	// We should get token not found error
-	_, err = db.GetByID(source.ID)
+	_, err = db.GetByID(ctx, source.ID)
 	expErr := ErrLoginSourceNotExist{args: errutil.Args{"id": source.ID}}
 	assert.Equal(t, expErr, err)
 }
 
-func test_loginSources_GetByID(t *testing.T, db *loginSources) {
-	setMockLoginSourceFilesStore(t, db, &mockLoginSourceFilesStore{
-		MockGetByID: func(id int64) (*LoginSource, error) {
-			if id != 101 {
-				return nil, ErrLoginSourceNotExist{args: errutil.Args{"id": id}}
-			}
-			return &LoginSource{ID: id}, nil
-		},
+func loginSourcesGetByID(t *testing.T, db *loginSources) {
+	ctx := context.Background()
+
+	mock := NewMockLoginSourceFilesStore()
+	mock.GetByIDFunc.SetDefaultHook(func(id int64) (*LoginSource, error) {
+		if id != 101 {
+			return nil, ErrLoginSourceNotExist{args: errutil.Args{"id": id}}
+		}
+		return &LoginSource{ID: id}, nil
 	})
+	setMockLoginSourceFilesStore(t, db, mock)
 
 	expConfig := &github.Config{
 		APIEndpoint: "https://api.github.com",
 	}
 
 	// Create a login source with name "GitHub"
-	source, err := db.Create(CreateLoginSourceOpts{
-		Type:      auth.GitHub,
-		Name:      "GitHub",
-		Activated: true,
-		Default:   false,
-		Config:    expConfig,
-	})
-	if err != nil {
-		t.Fatal(err)
-	}
+	source, err := db.Create(ctx,
+		CreateLoginSourceOpts{
+			Type:      auth.GitHub,
+			Name:      "GitHub",
+			Activated: true,
+			Default:   false,
+			Config:    expConfig,
+		},
+	)
+	require.NoError(t, err)
 
 	// Get the one in the database and test the read/write hooks
-	source, err = db.GetByID(source.ID)
-	if err != nil {
-		t.Fatal(err)
-	}
+	source, err = db.GetByID(ctx, source.ID)
+	require.NoError(t, err)
 	assert.Equal(t, expConfig, source.Provider.Config())
 
 	// Get the one in source file store
-	_, err = db.GetByID(101)
-	if err != nil {
-		t.Fatal(err)
-	}
+	_, err = db.GetByID(ctx, 101)
+	require.NoError(t, err)
 }
 
-func test_loginSources_List(t *testing.T, db *loginSources) {
-	setMockLoginSourceFilesStore(t, db, &mockLoginSourceFilesStore{
-		MockList: func(opts ListLoginSourceOpts) []*LoginSource {
-			if opts.OnlyActivated {
-				return []*LoginSource{
-					{ID: 1},
-				}
-			}
+func loginSourcesList(t *testing.T, db *loginSources) {
+	ctx := context.Background()
+
+	mock := NewMockLoginSourceFilesStore()
+	mock.ListFunc.SetDefaultHook(func(opts ListLoginSourceOpts) []*LoginSource {
+		if opts.OnlyActivated {
 			return []*LoginSource{
 				{ID: 1},
-				{ID: 2},
 			}
-		},
+		}
+		return []*LoginSource{
+			{ID: 1},
+			{ID: 2},
+		}
 	})
+	setMockLoginSourceFilesStore(t, db, mock)
 
 	// Create two login sources in database, one activated and the other one not
-	_, err := db.Create(CreateLoginSourceOpts{
-		Type: auth.PAM,
-		Name: "PAM",
-		Config: &pam.Config{
-			ServiceName: "PAM",
+	_, err := db.Create(ctx,
+		CreateLoginSourceOpts{
+			Type: auth.PAM,
+			Name: "PAM",
+			Config: &pam.Config{
+				ServiceName: "PAM",
+			},
 		},
-	})
-	if err != nil {
-		t.Fatal(err)
-	}
-	_, err = db.Create(CreateLoginSourceOpts{
-		Type:      auth.GitHub,
-		Name:      "GitHub",
-		Activated: true,
-		Config: &github.Config{
-			APIEndpoint: "https://api.github.com",
+	)
+	require.NoError(t, err)
+	_, err = db.Create(ctx,
+		CreateLoginSourceOpts{
+			Type:      auth.GitHub,
+			Name:      "GitHub",
+			Activated: true,
+			Config: &github.Config{
+				APIEndpoint: "https://api.github.com",
+			},
 		},
-	})
-	if err != nil {
-		t.Fatal(err)
-	}
+	)
+	require.NoError(t, err)
 
 	// List all login sources
-	sources, err := db.List(ListLoginSourceOpts{})
-	if err != nil {
-		t.Fatal(err)
-	}
+	sources, err := db.List(ctx, ListLoginSourceOpts{})
+	require.NoError(t, err)
 	assert.Equal(t, 4, len(sources), "number of sources")
 
 	// Only list activated login sources
-	sources, err = db.List(ListLoginSourceOpts{OnlyActivated: true})
-	if err != nil {
-		t.Fatal(err)
-	}
+	sources, err = db.List(ctx, ListLoginSourceOpts{OnlyActivated: true})
+	require.NoError(t, err)
 	assert.Equal(t, 2, len(sources), "number of sources")
 }
 
-func test_loginSources_ResetNonDefault(t *testing.T, db *loginSources) {
-	setMockLoginSourceFilesStore(t, db, &mockLoginSourceFilesStore{
-		MockList: func(opts ListLoginSourceOpts) []*LoginSource {
-			return []*LoginSource{
-				{
-					File: &mockLoginSourceFileStore{
-						MockSetGeneral: func(name, value string) {
-							assert.Equal(t, "is_default", name)
-							assert.Equal(t, "false", value)
-						},
-						MockSave: func() error {
-							return nil
-						},
-					},
-				},
-			}
-		},
-		MockUpdate: func(source *LoginSource) {},
+func loginSourcesResetNonDefault(t *testing.T, db *loginSources) {
+	ctx := context.Background()
+
+	mock := NewMockLoginSourceFilesStore()
+	mock.ListFunc.SetDefaultHook(func(opts ListLoginSourceOpts) []*LoginSource {
+		mockFile := NewMockLoginSourceFileStore()
+		mockFile.SetGeneralFunc.SetDefaultHook(func(name, value string) {
+			assert.Equal(t, "is_default", name)
+			assert.Equal(t, "false", value)
+		})
+		return []*LoginSource{
+			{
+				File: mockFile,
+			},
+		}
 	})
+	setMockLoginSourceFilesStore(t, db, mock)
 
 	// Create two login sources both have default on
-	source1, err := db.Create(CreateLoginSourceOpts{
-		Type:    auth.PAM,
-		Name:    "PAM",
-		Default: true,
-		Config: &pam.Config{
-			ServiceName: "PAM",
+	source1, err := db.Create(ctx,
+		CreateLoginSourceOpts{
+			Type:    auth.PAM,
+			Name:    "PAM",
+			Default: true,
+			Config: &pam.Config{
+				ServiceName: "PAM",
+			},
 		},
-	})
-	if err != nil {
-		t.Fatal(err)
-	}
-	source2, err := db.Create(CreateLoginSourceOpts{
-		Type:      auth.GitHub,
-		Name:      "GitHub",
-		Activated: true,
-		Default:   true,
-		Config: &github.Config{
-			APIEndpoint: "https://api.github.com",
+	)
+	require.NoError(t, err)
+	source2, err := db.Create(ctx,
+		CreateLoginSourceOpts{
+			Type:      auth.GitHub,
+			Name:      "GitHub",
+			Activated: true,
+			Default:   true,
+			Config: &github.Config{
+				APIEndpoint: "https://api.github.com",
+			},
 		},
-	})
-	if err != nil {
-		t.Fatal(err)
-	}
+	)
+	require.NoError(t, err)
 
 	// Set source 1 as default
-	err = db.ResetNonDefault(source1)
-	if err != nil {
-		t.Fatal(err)
-	}
+	err = db.ResetNonDefault(ctx, source1)
+	require.NoError(t, err)
 
 	// Verify the default state
-	source1, err = db.GetByID(source1.ID)
-	if err != nil {
-		t.Fatal(err)
-	}
+	source1, err = db.GetByID(ctx, source1.ID)
+	require.NoError(t, err)
 	assert.True(t, source1.IsDefault)
 
-	source2, err = db.GetByID(source2.ID)
-	if err != nil {
-		t.Fatal(err)
-	}
+	source2, err = db.GetByID(ctx, source2.ID)
+	require.NoError(t, err)
 	assert.False(t, source2.IsDefault)
 }
 
-func test_loginSources_Save(t *testing.T, db *loginSources) {
+func loginSourcesSave(t *testing.T, db *loginSources) {
+	ctx := context.Background()
+
 	t.Run("save to database", func(t *testing.T) {
 		// Create a login source with name "GitHub"
-		source, err := db.Create(CreateLoginSourceOpts{
-			Type:      auth.GitHub,
-			Name:      "GitHub",
-			Activated: true,
-			Default:   false,
-			Config: &github.Config{
-				APIEndpoint: "https://api.github.com",
+		source, err := db.Create(ctx,
+			CreateLoginSourceOpts{
+				Type:      auth.GitHub,
+				Name:      "GitHub",
+				Activated: true,
+				Default:   false,
+				Config: &github.Config{
+					APIEndpoint: "https://api.github.com",
+				},
 			},
-		})
-		if err != nil {
-			t.Fatal(err)
-		}
+		)
+		require.NoError(t, err)
 
 		source.IsActived = false
 		source.Provider = github.NewProvider(&github.Config{
 			APIEndpoint: "https://api2.github.com",
 		})
-		err = db.Save(source)
-		if err != nil {
-			t.Fatal(err)
-		}
+		err = db.Save(ctx, source)
+		require.NoError(t, err)
 
-		source, err = db.GetByID(source.ID)
-		if err != nil {
-			t.Fatal(err)
-		}
+		source, err = db.GetByID(ctx, source.ID)
+		require.NoError(t, err)
 		assert.False(t, source.IsActived)
 		assert.Equal(t, "https://api2.github.com", source.GitHub().APIEndpoint)
 	})
 
 	t.Run("save to file", func(t *testing.T) {
-		calledSave := false
+		mockFile := NewMockLoginSourceFileStore()
 		source := &LoginSource{
 			Provider: github.NewProvider(&github.Config{
 				APIEndpoint: "https://api.github.com",
 			}),
-			File: &mockLoginSourceFileStore{
-				MockSetGeneral: func(name, value string) {},
-				MockSetConfig:  func(cfg interface{}) error { return nil },
-				MockSave: func() error {
-					calledSave = true
-					return nil
-				},
-			},
-		}
-		err := db.Save(source)
-		if err != nil {
-			t.Fatal(err)
+			File: mockFile,
 		}
-		assert.True(t, calledSave)
+		err := db.Save(ctx, source)
+		require.NoError(t, err)
+		mockrequire.Called(t, mockFile.SaveFunc)
 	})
 }

+ 1 - 46
internal/db/mock_gen.go

@@ -8,7 +8,7 @@ import (
 	"testing"
 )
 
-//go:generate go-mockgen -f gogs.io/gogs/internal/db -i AccessTokensStore -i LFSStore -i PermsStore -o mocks.go
+//go:generate go-mockgen -f gogs.io/gogs/internal/db -i AccessTokensStore -i LFSStore -i LoginSourcesStore -i LoginSourceFilesStore -i loginSourceFileStore -i PermsStore -o mocks.go
 
 func SetMockAccessTokensStore(t *testing.T, mock AccessTokensStore) {
 	before := AccessTokens
@@ -26,31 +26,6 @@ func SetMockLFSStore(t *testing.T, mock LFSStore) {
 	})
 }
 
-var _ loginSourceFilesStore = (*mockLoginSourceFilesStore)(nil)
-
-type mockLoginSourceFilesStore struct {
-	MockGetByID func(id int64) (*LoginSource, error)
-	MockLen     func() int
-	MockList    func(opts ListLoginSourceOpts) []*LoginSource
-	MockUpdate  func(source *LoginSource)
-}
-
-func (m *mockLoginSourceFilesStore) GetByID(id int64) (*LoginSource, error) {
-	return m.MockGetByID(id)
-}
-
-func (m *mockLoginSourceFilesStore) Len() int {
-	return m.MockLen()
-}
-
-func (m *mockLoginSourceFilesStore) List(opts ListLoginSourceOpts) []*LoginSource {
-	return m.MockList(opts)
-}
-
-func (m *mockLoginSourceFilesStore) Update(source *LoginSource) {
-	m.MockUpdate(source)
-}
-
 func setMockLoginSourceFilesStore(t *testing.T, db *loginSources, mock loginSourceFilesStore) {
 	before := db.files
 	db.files = mock
@@ -59,26 +34,6 @@ func setMockLoginSourceFilesStore(t *testing.T, db *loginSources, mock loginSour
 	})
 }
 
-var _ loginSourceFileStore = (*mockLoginSourceFileStore)(nil)
-
-type mockLoginSourceFileStore struct {
-	MockSetGeneral func(name, value string)
-	MockSetConfig  func(cfg interface{}) error
-	MockSave       func() error
-}
-
-func (m *mockLoginSourceFileStore) SetGeneral(name, value string) {
-	m.MockSetGeneral(name, value)
-}
-
-func (m *mockLoginSourceFileStore) SetConfig(cfg interface{}) error {
-	return m.MockSetConfig(cfg)
-}
-
-func (m *mockLoginSourceFileStore) Save() error {
-	return m.MockSave()
-}
-
 func SetMockPermsStore(t *testing.T, mock PermsStore) {
 	before := Perms
 	Perms = mock

+ 1961 - 171
internal/db/mocks.go

@@ -1077,109 +1077,177 @@ func (c LFSStoreGetObjectsByOIDsFuncCall) Results() []interface{} {
 	return []interface{}{c.Result0, c.Result1}
 }
 
-// MockPermsStore is a mock implementation of the PermsStore interface (from
-// the package gogs.io/gogs/internal/db) used for unit testing.
-type MockPermsStore struct {
-	// AccessModeFunc is an instance of a mock function object controlling
-	// the behavior of the method AccessMode.
-	AccessModeFunc *PermsStoreAccessModeFunc
-	// AuthorizeFunc is an instance of a mock function object controlling
-	// the behavior of the method Authorize.
-	AuthorizeFunc *PermsStoreAuthorizeFunc
-	// SetRepoPermsFunc is an instance of a mock function object controlling
-	// the behavior of the method SetRepoPerms.
-	SetRepoPermsFunc *PermsStoreSetRepoPermsFunc
+// MockLoginSourcesStore is a mock implementation of the LoginSourcesStore
+// interface (from the package gogs.io/gogs/internal/db) used for unit
+// testing.
+type MockLoginSourcesStore struct {
+	// CountFunc is an instance of a mock function object controlling the
+	// behavior of the method Count.
+	CountFunc *LoginSourcesStoreCountFunc
+	// CreateFunc is an instance of a mock function object controlling the
+	// behavior of the method Create.
+	CreateFunc *LoginSourcesStoreCreateFunc
+	// DeleteByIDFunc is an instance of a mock function object controlling
+	// the behavior of the method DeleteByID.
+	DeleteByIDFunc *LoginSourcesStoreDeleteByIDFunc
+	// GetByIDFunc is an instance of a mock function object controlling the
+	// behavior of the method GetByID.
+	GetByIDFunc *LoginSourcesStoreGetByIDFunc
+	// ListFunc is an instance of a mock function object controlling the
+	// behavior of the method List.
+	ListFunc *LoginSourcesStoreListFunc
+	// ResetNonDefaultFunc is an instance of a mock function object
+	// controlling the behavior of the method ResetNonDefault.
+	ResetNonDefaultFunc *LoginSourcesStoreResetNonDefaultFunc
+	// SaveFunc is an instance of a mock function object controlling the
+	// behavior of the method Save.
+	SaveFunc *LoginSourcesStoreSaveFunc
 }
 
-// NewMockPermsStore creates a new mock of the PermsStore interface. All
-// methods return zero values for all results, unless overwritten.
-func NewMockPermsStore() *MockPermsStore {
-	return &MockPermsStore{
-		AccessModeFunc: &PermsStoreAccessModeFunc{
-			defaultHook: func(context.Context, int64, int64, AccessModeOptions) (r0 AccessMode) {
+// NewMockLoginSourcesStore creates a new mock of the LoginSourcesStore
+// interface. All methods return zero values for all results, unless
+// overwritten.
+func NewMockLoginSourcesStore() *MockLoginSourcesStore {
+	return &MockLoginSourcesStore{
+		CountFunc: &LoginSourcesStoreCountFunc{
+			defaultHook: func(context.Context) (r0 int64) {
 				return
 			},
 		},
-		AuthorizeFunc: &PermsStoreAuthorizeFunc{
-			defaultHook: func(context.Context, int64, int64, AccessMode, AccessModeOptions) (r0 bool) {
+		CreateFunc: &LoginSourcesStoreCreateFunc{
+			defaultHook: func(context.Context, CreateLoginSourceOpts) (r0 *LoginSource, r1 error) {
 				return
 			},
 		},
-		SetRepoPermsFunc: &PermsStoreSetRepoPermsFunc{
-			defaultHook: func(context.Context, int64, map[int64]AccessMode) (r0 error) {
+		DeleteByIDFunc: &LoginSourcesStoreDeleteByIDFunc{
+			defaultHook: func(context.Context, int64) (r0 error) {
+				return
+			},
+		},
+		GetByIDFunc: &LoginSourcesStoreGetByIDFunc{
+			defaultHook: func(context.Context, int64) (r0 *LoginSource, r1 error) {
+				return
+			},
+		},
+		ListFunc: &LoginSourcesStoreListFunc{
+			defaultHook: func(context.Context, ListLoginSourceOpts) (r0 []*LoginSource, r1 error) {
+				return
+			},
+		},
+		ResetNonDefaultFunc: &LoginSourcesStoreResetNonDefaultFunc{
+			defaultHook: func(context.Context, *LoginSource) (r0 error) {
+				return
+			},
+		},
+		SaveFunc: &LoginSourcesStoreSaveFunc{
+			defaultHook: func(context.Context, *LoginSource) (r0 error) {
 				return
 			},
 		},
 	}
 }
 
-// NewStrictMockPermsStore creates a new mock of the PermsStore interface.
-// All methods panic on invocation, unless overwritten.
-func NewStrictMockPermsStore() *MockPermsStore {
-	return &MockPermsStore{
-		AccessModeFunc: &PermsStoreAccessModeFunc{
-			defaultHook: func(context.Context, int64, int64, AccessModeOptions) AccessMode {
-				panic("unexpected invocation of MockPermsStore.AccessMode")
+// NewStrictMockLoginSourcesStore creates a new mock of the
+// LoginSourcesStore interface. All methods panic on invocation, unless
+// overwritten.
+func NewStrictMockLoginSourcesStore() *MockLoginSourcesStore {
+	return &MockLoginSourcesStore{
+		CountFunc: &LoginSourcesStoreCountFunc{
+			defaultHook: func(context.Context) int64 {
+				panic("unexpected invocation of MockLoginSourcesStore.Count")
 			},
 		},
-		AuthorizeFunc: &PermsStoreAuthorizeFunc{
-			defaultHook: func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool {
-				panic("unexpected invocation of MockPermsStore.Authorize")
+		CreateFunc: &LoginSourcesStoreCreateFunc{
+			defaultHook: func(context.Context, CreateLoginSourceOpts) (*LoginSource, error) {
+				panic("unexpected invocation of MockLoginSourcesStore.Create")
 			},
 		},
-		SetRepoPermsFunc: &PermsStoreSetRepoPermsFunc{
-			defaultHook: func(context.Context, int64, map[int64]AccessMode) error {
-				panic("unexpected invocation of MockPermsStore.SetRepoPerms")
+		DeleteByIDFunc: &LoginSourcesStoreDeleteByIDFunc{
+			defaultHook: func(context.Context, int64) error {
+				panic("unexpected invocation of MockLoginSourcesStore.DeleteByID")
+			},
+		},
+		GetByIDFunc: &LoginSourcesStoreGetByIDFunc{
+			defaultHook: func(context.Context, int64) (*LoginSource, error) {
+				panic("unexpected invocation of MockLoginSourcesStore.GetByID")
+			},
+		},
+		ListFunc: &LoginSourcesStoreListFunc{
+			defaultHook: func(context.Context, ListLoginSourceOpts) ([]*LoginSource, error) {
+				panic("unexpected invocation of MockLoginSourcesStore.List")
+			},
+		},
+		ResetNonDefaultFunc: &LoginSourcesStoreResetNonDefaultFunc{
+			defaultHook: func(context.Context, *LoginSource) error {
+				panic("unexpected invocation of MockLoginSourcesStore.ResetNonDefault")
+			},
+		},
+		SaveFunc: &LoginSourcesStoreSaveFunc{
+			defaultHook: func(context.Context, *LoginSource) error {
+				panic("unexpected invocation of MockLoginSourcesStore.Save")
 			},
 		},
 	}
 }
 
-// NewMockPermsStoreFrom creates a new mock of the MockPermsStore interface.
-// All methods delegate to the given implementation, unless overwritten.
-func NewMockPermsStoreFrom(i PermsStore) *MockPermsStore {
-	return &MockPermsStore{
-		AccessModeFunc: &PermsStoreAccessModeFunc{
-			defaultHook: i.AccessMode,
+// NewMockLoginSourcesStoreFrom creates a new mock of the
+// MockLoginSourcesStore interface. All methods delegate to the given
+// implementation, unless overwritten.
+func NewMockLoginSourcesStoreFrom(i LoginSourcesStore) *MockLoginSourcesStore {
+	return &MockLoginSourcesStore{
+		CountFunc: &LoginSourcesStoreCountFunc{
+			defaultHook: i.Count,
 		},
-		AuthorizeFunc: &PermsStoreAuthorizeFunc{
-			defaultHook: i.Authorize,
+		CreateFunc: &LoginSourcesStoreCreateFunc{
+			defaultHook: i.Create,
 		},
-		SetRepoPermsFunc: &PermsStoreSetRepoPermsFunc{
-			defaultHook: i.SetRepoPerms,
+		DeleteByIDFunc: &LoginSourcesStoreDeleteByIDFunc{
+			defaultHook: i.DeleteByID,
+		},
+		GetByIDFunc: &LoginSourcesStoreGetByIDFunc{
+			defaultHook: i.GetByID,
+		},
+		ListFunc: &LoginSourcesStoreListFunc{
+			defaultHook: i.List,
+		},
+		ResetNonDefaultFunc: &LoginSourcesStoreResetNonDefaultFunc{
+			defaultHook: i.ResetNonDefault,
+		},
+		SaveFunc: &LoginSourcesStoreSaveFunc{
+			defaultHook: i.Save,
 		},
 	}
 }
 
-// PermsStoreAccessModeFunc describes the behavior when the AccessMode
-// method of the parent MockPermsStore instance is invoked.
-type PermsStoreAccessModeFunc struct {
-	defaultHook func(context.Context, int64, int64, AccessModeOptions) AccessMode
-	hooks       []func(context.Context, int64, int64, AccessModeOptions) AccessMode
-	history     []PermsStoreAccessModeFuncCall
+// LoginSourcesStoreCountFunc describes the behavior when the Count method
+// of the parent MockLoginSourcesStore instance is invoked.
+type LoginSourcesStoreCountFunc struct {
+	defaultHook func(context.Context) int64
+	hooks       []func(context.Context) int64
+	history     []LoginSourcesStoreCountFuncCall
 	mutex       sync.Mutex
 }
 
-// AccessMode delegates to the next hook function in the queue and stores
-// the parameter and result values of this invocation.
-func (m *MockPermsStore) AccessMode(v0 context.Context, v1 int64, v2 int64, v3 AccessModeOptions) AccessMode {
-	r0 := m.AccessModeFunc.nextHook()(v0, v1, v2, v3)
-	m.AccessModeFunc.appendCall(PermsStoreAccessModeFuncCall{v0, v1, v2, v3, r0})
+// Count delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourcesStore) Count(v0 context.Context) int64 {
+	r0 := m.CountFunc.nextHook()(v0)
+	m.CountFunc.appendCall(LoginSourcesStoreCountFuncCall{v0, r0})
 	return r0
 }
 
-// SetDefaultHook sets function that is called when the AccessMode method of
-// the parent MockPermsStore instance is invoked and the hook queue is
+// SetDefaultHook sets function that is called when the Count method of the
+// parent MockLoginSourcesStore instance is invoked and the hook queue is
 // empty.
-func (f *PermsStoreAccessModeFunc) SetDefaultHook(hook func(context.Context, int64, int64, AccessModeOptions) AccessMode) {
+func (f *LoginSourcesStoreCountFunc) SetDefaultHook(hook func(context.Context) int64) {
 	f.defaultHook = hook
 }
 
 // PushHook adds a function to the end of hook queue. Each invocation of the
-// AccessMode method of the parent MockPermsStore instance invokes the hook
-// at the front of the queue and discards it. After the queue is empty, the
-// default hook function is invoked for any future action.
-func (f *PermsStoreAccessModeFunc) PushHook(hook func(context.Context, int64, int64, AccessModeOptions) AccessMode) {
+// Count method of the parent MockLoginSourcesStore instance invokes the
+// hook at the front of the queue and discards it. After the queue is empty,
+// the default hook function is invoked for any future action.
+func (f *LoginSourcesStoreCountFunc) PushHook(hook func(context.Context) int64) {
 	f.mutex.Lock()
 	f.hooks = append(f.hooks, hook)
 	f.mutex.Unlock()
@@ -1187,20 +1255,20 @@ func (f *PermsStoreAccessModeFunc) PushHook(hook func(context.Context, int64, in
 
 // SetDefaultReturn calls SetDefaultHook with a function that returns the
 // given values.
-func (f *PermsStoreAccessModeFunc) SetDefaultReturn(r0 AccessMode) {
-	f.SetDefaultHook(func(context.Context, int64, int64, AccessModeOptions) AccessMode {
+func (f *LoginSourcesStoreCountFunc) SetDefaultReturn(r0 int64) {
+	f.SetDefaultHook(func(context.Context) int64 {
 		return r0
 	})
 }
 
 // PushReturn calls PushHook with a function that returns the given values.
-func (f *PermsStoreAccessModeFunc) PushReturn(r0 AccessMode) {
-	f.PushHook(func(context.Context, int64, int64, AccessModeOptions) AccessMode {
+func (f *LoginSourcesStoreCountFunc) PushReturn(r0 int64) {
+	f.PushHook(func(context.Context) int64 {
 		return r0
 	})
 }
 
-func (f *PermsStoreAccessModeFunc) nextHook() func(context.Context, int64, int64, AccessModeOptions) AccessMode {
+func (f *LoginSourcesStoreCountFunc) nextHook() func(context.Context) int64 {
 	f.mutex.Lock()
 	defer f.mutex.Unlock()
 
@@ -1213,84 +1281,75 @@ func (f *PermsStoreAccessModeFunc) nextHook() func(context.Context, int64, int64
 	return hook
 }
 
-func (f *PermsStoreAccessModeFunc) appendCall(r0 PermsStoreAccessModeFuncCall) {
+func (f *LoginSourcesStoreCountFunc) appendCall(r0 LoginSourcesStoreCountFuncCall) {
 	f.mutex.Lock()
 	f.history = append(f.history, r0)
 	f.mutex.Unlock()
 }
 
-// History returns a sequence of PermsStoreAccessModeFuncCall objects
+// History returns a sequence of LoginSourcesStoreCountFuncCall objects
 // describing the invocations of this function.
-func (f *PermsStoreAccessModeFunc) History() []PermsStoreAccessModeFuncCall {
+func (f *LoginSourcesStoreCountFunc) History() []LoginSourcesStoreCountFuncCall {
 	f.mutex.Lock()
-	history := make([]PermsStoreAccessModeFuncCall, len(f.history))
+	history := make([]LoginSourcesStoreCountFuncCall, len(f.history))
 	copy(history, f.history)
 	f.mutex.Unlock()
 
 	return history
 }
 
-// PermsStoreAccessModeFuncCall is an object that describes an invocation of
-// method AccessMode on an instance of MockPermsStore.
-type PermsStoreAccessModeFuncCall struct {
+// LoginSourcesStoreCountFuncCall is an object that describes an invocation
+// of method Count on an instance of MockLoginSourcesStore.
+type LoginSourcesStoreCountFuncCall struct {
 	// Arg0 is the value of the 1st argument passed to this method
 	// invocation.
 	Arg0 context.Context
-	// Arg1 is the value of the 2nd argument passed to this method
-	// invocation.
-	Arg1 int64
-	// Arg2 is the value of the 3rd argument passed to this method
-	// invocation.
-	Arg2 int64
-	// Arg3 is the value of the 4th argument passed to this method
-	// invocation.
-	Arg3 AccessModeOptions
 	// Result0 is the value of the 1st result returned from this method
 	// invocation.
-	Result0 AccessMode
+	Result0 int64
 }
 
 // Args returns an interface slice containing the arguments of this
 // invocation.
-func (c PermsStoreAccessModeFuncCall) Args() []interface{} {
-	return []interface{}{c.Arg0, c.Arg1, c.Arg2, c.Arg3}
+func (c LoginSourcesStoreCountFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0}
 }
 
 // Results returns an interface slice containing the results of this
 // invocation.
-func (c PermsStoreAccessModeFuncCall) Results() []interface{} {
+func (c LoginSourcesStoreCountFuncCall) Results() []interface{} {
 	return []interface{}{c.Result0}
 }
 
-// PermsStoreAuthorizeFunc describes the behavior when the Authorize method
-// of the parent MockPermsStore instance is invoked.
-type PermsStoreAuthorizeFunc struct {
-	defaultHook func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool
-	hooks       []func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool
-	history     []PermsStoreAuthorizeFuncCall
+// LoginSourcesStoreCreateFunc describes the behavior when the Create method
+// of the parent MockLoginSourcesStore instance is invoked.
+type LoginSourcesStoreCreateFunc struct {
+	defaultHook func(context.Context, CreateLoginSourceOpts) (*LoginSource, error)
+	hooks       []func(context.Context, CreateLoginSourceOpts) (*LoginSource, error)
+	history     []LoginSourcesStoreCreateFuncCall
 	mutex       sync.Mutex
 }
 
-// Authorize delegates to the next hook function in the queue and stores the
+// Create delegates to the next hook function in the queue and stores the
 // parameter and result values of this invocation.
-func (m *MockPermsStore) Authorize(v0 context.Context, v1 int64, v2 int64, v3 AccessMode, v4 AccessModeOptions) bool {
-	r0 := m.AuthorizeFunc.nextHook()(v0, v1, v2, v3, v4)
-	m.AuthorizeFunc.appendCall(PermsStoreAuthorizeFuncCall{v0, v1, v2, v3, v4, r0})
-	return r0
+func (m *MockLoginSourcesStore) Create(v0 context.Context, v1 CreateLoginSourceOpts) (*LoginSource, error) {
+	r0, r1 := m.CreateFunc.nextHook()(v0, v1)
+	m.CreateFunc.appendCall(LoginSourcesStoreCreateFuncCall{v0, v1, r0, r1})
+	return r0, r1
 }
 
-// SetDefaultHook sets function that is called when the Authorize method of
-// the parent MockPermsStore instance is invoked and the hook queue is
+// SetDefaultHook sets function that is called when the Create method of the
+// parent MockLoginSourcesStore instance is invoked and the hook queue is
 // empty.
-func (f *PermsStoreAuthorizeFunc) SetDefaultHook(hook func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool) {
+func (f *LoginSourcesStoreCreateFunc) SetDefaultHook(hook func(context.Context, CreateLoginSourceOpts) (*LoginSource, error)) {
 	f.defaultHook = hook
 }
 
 // PushHook adds a function to the end of hook queue. Each invocation of the
-// Authorize method of the parent MockPermsStore instance invokes the hook
-// at the front of the queue and discards it. After the queue is empty, the
-// default hook function is invoked for any future action.
-func (f *PermsStoreAuthorizeFunc) PushHook(hook func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool) {
+// Create method of the parent MockLoginSourcesStore instance invokes the
+// hook at the front of the queue and discards it. After the queue is empty,
+// the default hook function is invoked for any future action.
+func (f *LoginSourcesStoreCreateFunc) PushHook(hook func(context.Context, CreateLoginSourceOpts) (*LoginSource, error)) {
 	f.mutex.Lock()
 	f.hooks = append(f.hooks, hook)
 	f.mutex.Unlock()
@@ -1298,20 +1357,20 @@ func (f *PermsStoreAuthorizeFunc) PushHook(hook func(context.Context, int64, int
 
 // SetDefaultReturn calls SetDefaultHook with a function that returns the
 // given values.
-func (f *PermsStoreAuthorizeFunc) SetDefaultReturn(r0 bool) {
-	f.SetDefaultHook(func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool {
-		return r0
+func (f *LoginSourcesStoreCreateFunc) SetDefaultReturn(r0 *LoginSource, r1 error) {
+	f.SetDefaultHook(func(context.Context, CreateLoginSourceOpts) (*LoginSource, error) {
+		return r0, r1
 	})
 }
 
 // PushReturn calls PushHook with a function that returns the given values.
-func (f *PermsStoreAuthorizeFunc) PushReturn(r0 bool) {
-	f.PushHook(func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool {
-		return r0
+func (f *LoginSourcesStoreCreateFunc) PushReturn(r0 *LoginSource, r1 error) {
+	f.PushHook(func(context.Context, CreateLoginSourceOpts) (*LoginSource, error) {
+		return r0, r1
 	})
 }
 
-func (f *PermsStoreAuthorizeFunc) nextHook() func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool {
+func (f *LoginSourcesStoreCreateFunc) nextHook() func(context.Context, CreateLoginSourceOpts) (*LoginSource, error) {
 	f.mutex.Lock()
 	defer f.mutex.Unlock()
 
@@ -1324,87 +1383,82 @@ func (f *PermsStoreAuthorizeFunc) nextHook() func(context.Context, int64, int64,
 	return hook
 }
 
-func (f *PermsStoreAuthorizeFunc) appendCall(r0 PermsStoreAuthorizeFuncCall) {
+func (f *LoginSourcesStoreCreateFunc) appendCall(r0 LoginSourcesStoreCreateFuncCall) {
 	f.mutex.Lock()
 	f.history = append(f.history, r0)
 	f.mutex.Unlock()
 }
 
-// History returns a sequence of PermsStoreAuthorizeFuncCall objects
+// History returns a sequence of LoginSourcesStoreCreateFuncCall objects
 // describing the invocations of this function.
-func (f *PermsStoreAuthorizeFunc) History() []PermsStoreAuthorizeFuncCall {
+func (f *LoginSourcesStoreCreateFunc) History() []LoginSourcesStoreCreateFuncCall {
 	f.mutex.Lock()
-	history := make([]PermsStoreAuthorizeFuncCall, len(f.history))
+	history := make([]LoginSourcesStoreCreateFuncCall, len(f.history))
 	copy(history, f.history)
 	f.mutex.Unlock()
 
 	return history
 }
 
-// PermsStoreAuthorizeFuncCall is an object that describes an invocation of
-// method Authorize on an instance of MockPermsStore.
-type PermsStoreAuthorizeFuncCall struct {
+// LoginSourcesStoreCreateFuncCall is an object that describes an invocation
+// of method Create on an instance of MockLoginSourcesStore.
+type LoginSourcesStoreCreateFuncCall struct {
 	// Arg0 is the value of the 1st argument passed to this method
 	// invocation.
 	Arg0 context.Context
 	// Arg1 is the value of the 2nd argument passed to this method
 	// invocation.
-	Arg1 int64
-	// Arg2 is the value of the 3rd argument passed to this method
-	// invocation.
-	Arg2 int64
-	// Arg3 is the value of the 4th argument passed to this method
-	// invocation.
-	Arg3 AccessMode
-	// Arg4 is the value of the 5th argument passed to this method
-	// invocation.
-	Arg4 AccessModeOptions
+	Arg1 CreateLoginSourceOpts
 	// Result0 is the value of the 1st result returned from this method
 	// invocation.
-	Result0 bool
+	Result0 *LoginSource
+	// Result1 is the value of the 2nd result returned from this method
+	// invocation.
+	Result1 error
 }
 
 // Args returns an interface slice containing the arguments of this
 // invocation.
-func (c PermsStoreAuthorizeFuncCall) Args() []interface{} {
-	return []interface{}{c.Arg0, c.Arg1, c.Arg2, c.Arg3, c.Arg4}
+func (c LoginSourcesStoreCreateFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1}
 }
 
 // Results returns an interface slice containing the results of this
 // invocation.
-func (c PermsStoreAuthorizeFuncCall) Results() []interface{} {
-	return []interface{}{c.Result0}
+func (c LoginSourcesStoreCreateFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0, c.Result1}
 }
 
-// PermsStoreSetRepoPermsFunc describes the behavior when the SetRepoPerms
-// method of the parent MockPermsStore instance is invoked.
-type PermsStoreSetRepoPermsFunc struct {
-	defaultHook func(context.Context, int64, map[int64]AccessMode) error
-	hooks       []func(context.Context, int64, map[int64]AccessMode) error
-	history     []PermsStoreSetRepoPermsFuncCall
+// LoginSourcesStoreDeleteByIDFunc describes the behavior when the
+// DeleteByID method of the parent MockLoginSourcesStore instance is
+// invoked.
+type LoginSourcesStoreDeleteByIDFunc struct {
+	defaultHook func(context.Context, int64) error
+	hooks       []func(context.Context, int64) error
+	history     []LoginSourcesStoreDeleteByIDFuncCall
 	mutex       sync.Mutex
 }
 
-// SetRepoPerms delegates to the next hook function in the queue and stores
+// DeleteByID delegates to the next hook function in the queue and stores
 // the parameter and result values of this invocation.
-func (m *MockPermsStore) SetRepoPerms(v0 context.Context, v1 int64, v2 map[int64]AccessMode) error {
-	r0 := m.SetRepoPermsFunc.nextHook()(v0, v1, v2)
-	m.SetRepoPermsFunc.appendCall(PermsStoreSetRepoPermsFuncCall{v0, v1, v2, r0})
+func (m *MockLoginSourcesStore) DeleteByID(v0 context.Context, v1 int64) error {
+	r0 := m.DeleteByIDFunc.nextHook()(v0, v1)
+	m.DeleteByIDFunc.appendCall(LoginSourcesStoreDeleteByIDFuncCall{v0, v1, r0})
 	return r0
 }
 
-// SetDefaultHook sets function that is called when the SetRepoPerms method
-// of the parent MockPermsStore instance is invoked and the hook queue is
-// empty.
-func (f *PermsStoreSetRepoPermsFunc) SetDefaultHook(hook func(context.Context, int64, map[int64]AccessMode) error) {
+// SetDefaultHook sets function that is called when the DeleteByID method of
+// the parent MockLoginSourcesStore instance is invoked and the hook queue
+// is empty.
+func (f *LoginSourcesStoreDeleteByIDFunc) SetDefaultHook(hook func(context.Context, int64) error) {
 	f.defaultHook = hook
 }
 
 // PushHook adds a function to the end of hook queue. Each invocation of the
-// SetRepoPerms method of the parent MockPermsStore instance invokes the
-// hook at the front of the queue and discards it. After the queue is empty,
-// the default hook function is invoked for any future action.
-func (f *PermsStoreSetRepoPermsFunc) PushHook(hook func(context.Context, int64, map[int64]AccessMode) error) {
+// DeleteByID method of the parent MockLoginSourcesStore instance invokes
+// the hook at the front of the queue and discards it. After the queue is
+// empty, the default hook function is invoked for any future action.
+func (f *LoginSourcesStoreDeleteByIDFunc) PushHook(hook func(context.Context, int64) error) {
 	f.mutex.Lock()
 	f.hooks = append(f.hooks, hook)
 	f.mutex.Unlock()
@@ -1412,20 +1466,20 @@ func (f *PermsStoreSetRepoPermsFunc) PushHook(hook func(context.Context, int64,
 
 // SetDefaultReturn calls SetDefaultHook with a function that returns the
 // given values.
-func (f *PermsStoreSetRepoPermsFunc) SetDefaultReturn(r0 error) {
-	f.SetDefaultHook(func(context.Context, int64, map[int64]AccessMode) error {
+func (f *LoginSourcesStoreDeleteByIDFunc) SetDefaultReturn(r0 error) {
+	f.SetDefaultHook(func(context.Context, int64) error {
 		return r0
 	})
 }
 
 // PushReturn calls PushHook with a function that returns the given values.
-func (f *PermsStoreSetRepoPermsFunc) PushReturn(r0 error) {
-	f.PushHook(func(context.Context, int64, map[int64]AccessMode) error {
+func (f *LoginSourcesStoreDeleteByIDFunc) PushReturn(r0 error) {
+	f.PushHook(func(context.Context, int64) error {
 		return r0
 	})
 }
 
-func (f *PermsStoreSetRepoPermsFunc) nextHook() func(context.Context, int64, map[int64]AccessMode) error {
+func (f *LoginSourcesStoreDeleteByIDFunc) nextHook() func(context.Context, int64) error {
 	f.mutex.Lock()
 	defer f.mutex.Unlock()
 
@@ -1438,35 +1492,32 @@ func (f *PermsStoreSetRepoPermsFunc) nextHook() func(context.Context, int64, map
 	return hook
 }
 
-func (f *PermsStoreSetRepoPermsFunc) appendCall(r0 PermsStoreSetRepoPermsFuncCall) {
+func (f *LoginSourcesStoreDeleteByIDFunc) appendCall(r0 LoginSourcesStoreDeleteByIDFuncCall) {
 	f.mutex.Lock()
 	f.history = append(f.history, r0)
 	f.mutex.Unlock()
 }
 
-// History returns a sequence of PermsStoreSetRepoPermsFuncCall objects
+// History returns a sequence of LoginSourcesStoreDeleteByIDFuncCall objects
 // describing the invocations of this function.
-func (f *PermsStoreSetRepoPermsFunc) History() []PermsStoreSetRepoPermsFuncCall {
+func (f *LoginSourcesStoreDeleteByIDFunc) History() []LoginSourcesStoreDeleteByIDFuncCall {
 	f.mutex.Lock()
-	history := make([]PermsStoreSetRepoPermsFuncCall, len(f.history))
+	history := make([]LoginSourcesStoreDeleteByIDFuncCall, len(f.history))
 	copy(history, f.history)
 	f.mutex.Unlock()
 
 	return history
 }
 
-// PermsStoreSetRepoPermsFuncCall is an object that describes an invocation
-// of method SetRepoPerms on an instance of MockPermsStore.
-type PermsStoreSetRepoPermsFuncCall struct {
+// LoginSourcesStoreDeleteByIDFuncCall is an object that describes an
+// invocation of method DeleteByID on an instance of MockLoginSourcesStore.
+type LoginSourcesStoreDeleteByIDFuncCall struct {
 	// Arg0 is the value of the 1st argument passed to this method
 	// invocation.
 	Arg0 context.Context
 	// Arg1 is the value of the 2nd argument passed to this method
 	// invocation.
 	Arg1 int64
-	// Arg2 is the value of the 3rd argument passed to this method
-	// invocation.
-	Arg2 map[int64]AccessMode
 	// Result0 is the value of the 1st result returned from this method
 	// invocation.
 	Result0 error
@@ -1474,12 +1525,1751 @@ type PermsStoreSetRepoPermsFuncCall struct {
 
 // Args returns an interface slice containing the arguments of this
 // invocation.
-func (c PermsStoreSetRepoPermsFuncCall) Args() []interface{} {
-	return []interface{}{c.Arg0, c.Arg1, c.Arg2}
+func (c LoginSourcesStoreDeleteByIDFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1}
 }
 
 // Results returns an interface slice containing the results of this
 // invocation.
-func (c PermsStoreSetRepoPermsFuncCall) Results() []interface{} {
+func (c LoginSourcesStoreDeleteByIDFuncCall) Results() []interface{} {
 	return []interface{}{c.Result0}
 }
+
+// LoginSourcesStoreGetByIDFunc describes the behavior when the GetByID
+// method of the parent MockLoginSourcesStore instance is invoked.
+type LoginSourcesStoreGetByIDFunc struct {
+	defaultHook func(context.Context, int64) (*LoginSource, error)
+	hooks       []func(context.Context, int64) (*LoginSource, error)
+	history     []LoginSourcesStoreGetByIDFuncCall
+	mutex       sync.Mutex
+}
+
+// GetByID delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourcesStore) GetByID(v0 context.Context, v1 int64) (*LoginSource, error) {
+	r0, r1 := m.GetByIDFunc.nextHook()(v0, v1)
+	m.GetByIDFunc.appendCall(LoginSourcesStoreGetByIDFuncCall{v0, v1, r0, r1})
+	return r0, r1
+}
+
+// SetDefaultHook sets function that is called when the GetByID method of
+// the parent MockLoginSourcesStore instance is invoked and the hook queue
+// is empty.
+func (f *LoginSourcesStoreGetByIDFunc) SetDefaultHook(hook func(context.Context, int64) (*LoginSource, error)) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// GetByID method of the parent MockLoginSourcesStore instance invokes the
+// hook at the front of the queue and discards it. After the queue is empty,
+// the default hook function is invoked for any future action.
+func (f *LoginSourcesStoreGetByIDFunc) PushHook(hook func(context.Context, int64) (*LoginSource, error)) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourcesStoreGetByIDFunc) SetDefaultReturn(r0 *LoginSource, r1 error) {
+	f.SetDefaultHook(func(context.Context, int64) (*LoginSource, error) {
+		return r0, r1
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourcesStoreGetByIDFunc) PushReturn(r0 *LoginSource, r1 error) {
+	f.PushHook(func(context.Context, int64) (*LoginSource, error) {
+		return r0, r1
+	})
+}
+
+func (f *LoginSourcesStoreGetByIDFunc) nextHook() func(context.Context, int64) (*LoginSource, error) {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourcesStoreGetByIDFunc) appendCall(r0 LoginSourcesStoreGetByIDFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourcesStoreGetByIDFuncCall objects
+// describing the invocations of this function.
+func (f *LoginSourcesStoreGetByIDFunc) History() []LoginSourcesStoreGetByIDFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourcesStoreGetByIDFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourcesStoreGetByIDFuncCall is an object that describes an
+// invocation of method GetByID on an instance of MockLoginSourcesStore.
+type LoginSourcesStoreGetByIDFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 context.Context
+	// Arg1 is the value of the 2nd argument passed to this method
+	// invocation.
+	Arg1 int64
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 *LoginSource
+	// Result1 is the value of the 2nd result returned from this method
+	// invocation.
+	Result1 error
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourcesStoreGetByIDFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourcesStoreGetByIDFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0, c.Result1}
+}
+
+// LoginSourcesStoreListFunc describes the behavior when the List method of
+// the parent MockLoginSourcesStore instance is invoked.
+type LoginSourcesStoreListFunc struct {
+	defaultHook func(context.Context, ListLoginSourceOpts) ([]*LoginSource, error)
+	hooks       []func(context.Context, ListLoginSourceOpts) ([]*LoginSource, error)
+	history     []LoginSourcesStoreListFuncCall
+	mutex       sync.Mutex
+}
+
+// List delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourcesStore) List(v0 context.Context, v1 ListLoginSourceOpts) ([]*LoginSource, error) {
+	r0, r1 := m.ListFunc.nextHook()(v0, v1)
+	m.ListFunc.appendCall(LoginSourcesStoreListFuncCall{v0, v1, r0, r1})
+	return r0, r1
+}
+
+// SetDefaultHook sets function that is called when the List method of the
+// parent MockLoginSourcesStore instance is invoked and the hook queue is
+// empty.
+func (f *LoginSourcesStoreListFunc) SetDefaultHook(hook func(context.Context, ListLoginSourceOpts) ([]*LoginSource, error)) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// List method of the parent MockLoginSourcesStore instance invokes the hook
+// at the front of the queue and discards it. After the queue is empty, the
+// default hook function is invoked for any future action.
+func (f *LoginSourcesStoreListFunc) PushHook(hook func(context.Context, ListLoginSourceOpts) ([]*LoginSource, error)) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourcesStoreListFunc) SetDefaultReturn(r0 []*LoginSource, r1 error) {
+	f.SetDefaultHook(func(context.Context, ListLoginSourceOpts) ([]*LoginSource, error) {
+		return r0, r1
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourcesStoreListFunc) PushReturn(r0 []*LoginSource, r1 error) {
+	f.PushHook(func(context.Context, ListLoginSourceOpts) ([]*LoginSource, error) {
+		return r0, r1
+	})
+}
+
+func (f *LoginSourcesStoreListFunc) nextHook() func(context.Context, ListLoginSourceOpts) ([]*LoginSource, error) {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourcesStoreListFunc) appendCall(r0 LoginSourcesStoreListFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourcesStoreListFuncCall objects
+// describing the invocations of this function.
+func (f *LoginSourcesStoreListFunc) History() []LoginSourcesStoreListFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourcesStoreListFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourcesStoreListFuncCall is an object that describes an invocation
+// of method List on an instance of MockLoginSourcesStore.
+type LoginSourcesStoreListFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 context.Context
+	// Arg1 is the value of the 2nd argument passed to this method
+	// invocation.
+	Arg1 ListLoginSourceOpts
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 []*LoginSource
+	// Result1 is the value of the 2nd result returned from this method
+	// invocation.
+	Result1 error
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourcesStoreListFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourcesStoreListFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0, c.Result1}
+}
+
+// LoginSourcesStoreResetNonDefaultFunc describes the behavior when the
+// ResetNonDefault method of the parent MockLoginSourcesStore instance is
+// invoked.
+type LoginSourcesStoreResetNonDefaultFunc struct {
+	defaultHook func(context.Context, *LoginSource) error
+	hooks       []func(context.Context, *LoginSource) error
+	history     []LoginSourcesStoreResetNonDefaultFuncCall
+	mutex       sync.Mutex
+}
+
+// ResetNonDefault delegates to the next hook function in the queue and
+// stores the parameter and result values of this invocation.
+func (m *MockLoginSourcesStore) ResetNonDefault(v0 context.Context, v1 *LoginSource) error {
+	r0 := m.ResetNonDefaultFunc.nextHook()(v0, v1)
+	m.ResetNonDefaultFunc.appendCall(LoginSourcesStoreResetNonDefaultFuncCall{v0, v1, r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the ResetNonDefault
+// method of the parent MockLoginSourcesStore instance is invoked and the
+// hook queue is empty.
+func (f *LoginSourcesStoreResetNonDefaultFunc) SetDefaultHook(hook func(context.Context, *LoginSource) error) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// ResetNonDefault method of the parent MockLoginSourcesStore instance
+// invokes the hook at the front of the queue and discards it. After the
+// queue is empty, the default hook function is invoked for any future
+// action.
+func (f *LoginSourcesStoreResetNonDefaultFunc) PushHook(hook func(context.Context, *LoginSource) error) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourcesStoreResetNonDefaultFunc) SetDefaultReturn(r0 error) {
+	f.SetDefaultHook(func(context.Context, *LoginSource) error {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourcesStoreResetNonDefaultFunc) PushReturn(r0 error) {
+	f.PushHook(func(context.Context, *LoginSource) error {
+		return r0
+	})
+}
+
+func (f *LoginSourcesStoreResetNonDefaultFunc) nextHook() func(context.Context, *LoginSource) error {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourcesStoreResetNonDefaultFunc) appendCall(r0 LoginSourcesStoreResetNonDefaultFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourcesStoreResetNonDefaultFuncCall
+// objects describing the invocations of this function.
+func (f *LoginSourcesStoreResetNonDefaultFunc) History() []LoginSourcesStoreResetNonDefaultFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourcesStoreResetNonDefaultFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourcesStoreResetNonDefaultFuncCall is an object that describes an
+// invocation of method ResetNonDefault on an instance of
+// MockLoginSourcesStore.
+type LoginSourcesStoreResetNonDefaultFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 context.Context
+	// Arg1 is the value of the 2nd argument passed to this method
+	// invocation.
+	Arg1 *LoginSource
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 error
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourcesStoreResetNonDefaultFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourcesStoreResetNonDefaultFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// LoginSourcesStoreSaveFunc describes the behavior when the Save method of
+// the parent MockLoginSourcesStore instance is invoked.
+type LoginSourcesStoreSaveFunc struct {
+	defaultHook func(context.Context, *LoginSource) error
+	hooks       []func(context.Context, *LoginSource) error
+	history     []LoginSourcesStoreSaveFuncCall
+	mutex       sync.Mutex
+}
+
+// Save delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourcesStore) Save(v0 context.Context, v1 *LoginSource) error {
+	r0 := m.SaveFunc.nextHook()(v0, v1)
+	m.SaveFunc.appendCall(LoginSourcesStoreSaveFuncCall{v0, v1, r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the Save method of the
+// parent MockLoginSourcesStore instance is invoked and the hook queue is
+// empty.
+func (f *LoginSourcesStoreSaveFunc) SetDefaultHook(hook func(context.Context, *LoginSource) error) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// Save method of the parent MockLoginSourcesStore instance invokes the hook
+// at the front of the queue and discards it. After the queue is empty, the
+// default hook function is invoked for any future action.
+func (f *LoginSourcesStoreSaveFunc) PushHook(hook func(context.Context, *LoginSource) error) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourcesStoreSaveFunc) SetDefaultReturn(r0 error) {
+	f.SetDefaultHook(func(context.Context, *LoginSource) error {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourcesStoreSaveFunc) PushReturn(r0 error) {
+	f.PushHook(func(context.Context, *LoginSource) error {
+		return r0
+	})
+}
+
+func (f *LoginSourcesStoreSaveFunc) nextHook() func(context.Context, *LoginSource) error {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourcesStoreSaveFunc) appendCall(r0 LoginSourcesStoreSaveFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourcesStoreSaveFuncCall objects
+// describing the invocations of this function.
+func (f *LoginSourcesStoreSaveFunc) History() []LoginSourcesStoreSaveFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourcesStoreSaveFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourcesStoreSaveFuncCall is an object that describes an invocation
+// of method Save on an instance of MockLoginSourcesStore.
+type LoginSourcesStoreSaveFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 context.Context
+	// Arg1 is the value of the 2nd argument passed to this method
+	// invocation.
+	Arg1 *LoginSource
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 error
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourcesStoreSaveFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourcesStoreSaveFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// MockPermsStore is a mock implementation of the PermsStore interface (from
+// the package gogs.io/gogs/internal/db) used for unit testing.
+type MockPermsStore struct {
+	// AccessModeFunc is an instance of a mock function object controlling
+	// the behavior of the method AccessMode.
+	AccessModeFunc *PermsStoreAccessModeFunc
+	// AuthorizeFunc is an instance of a mock function object controlling
+	// the behavior of the method Authorize.
+	AuthorizeFunc *PermsStoreAuthorizeFunc
+	// SetRepoPermsFunc is an instance of a mock function object controlling
+	// the behavior of the method SetRepoPerms.
+	SetRepoPermsFunc *PermsStoreSetRepoPermsFunc
+}
+
+// NewMockPermsStore creates a new mock of the PermsStore interface. All
+// methods return zero values for all results, unless overwritten.
+func NewMockPermsStore() *MockPermsStore {
+	return &MockPermsStore{
+		AccessModeFunc: &PermsStoreAccessModeFunc{
+			defaultHook: func(context.Context, int64, int64, AccessModeOptions) (r0 AccessMode) {
+				return
+			},
+		},
+		AuthorizeFunc: &PermsStoreAuthorizeFunc{
+			defaultHook: func(context.Context, int64, int64, AccessMode, AccessModeOptions) (r0 bool) {
+				return
+			},
+		},
+		SetRepoPermsFunc: &PermsStoreSetRepoPermsFunc{
+			defaultHook: func(context.Context, int64, map[int64]AccessMode) (r0 error) {
+				return
+			},
+		},
+	}
+}
+
+// NewStrictMockPermsStore creates a new mock of the PermsStore interface.
+// All methods panic on invocation, unless overwritten.
+func NewStrictMockPermsStore() *MockPermsStore {
+	return &MockPermsStore{
+		AccessModeFunc: &PermsStoreAccessModeFunc{
+			defaultHook: func(context.Context, int64, int64, AccessModeOptions) AccessMode {
+				panic("unexpected invocation of MockPermsStore.AccessMode")
+			},
+		},
+		AuthorizeFunc: &PermsStoreAuthorizeFunc{
+			defaultHook: func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool {
+				panic("unexpected invocation of MockPermsStore.Authorize")
+			},
+		},
+		SetRepoPermsFunc: &PermsStoreSetRepoPermsFunc{
+			defaultHook: func(context.Context, int64, map[int64]AccessMode) error {
+				panic("unexpected invocation of MockPermsStore.SetRepoPerms")
+			},
+		},
+	}
+}
+
+// NewMockPermsStoreFrom creates a new mock of the MockPermsStore interface.
+// All methods delegate to the given implementation, unless overwritten.
+func NewMockPermsStoreFrom(i PermsStore) *MockPermsStore {
+	return &MockPermsStore{
+		AccessModeFunc: &PermsStoreAccessModeFunc{
+			defaultHook: i.AccessMode,
+		},
+		AuthorizeFunc: &PermsStoreAuthorizeFunc{
+			defaultHook: i.Authorize,
+		},
+		SetRepoPermsFunc: &PermsStoreSetRepoPermsFunc{
+			defaultHook: i.SetRepoPerms,
+		},
+	}
+}
+
+// PermsStoreAccessModeFunc describes the behavior when the AccessMode
+// method of the parent MockPermsStore instance is invoked.
+type PermsStoreAccessModeFunc struct {
+	defaultHook func(context.Context, int64, int64, AccessModeOptions) AccessMode
+	hooks       []func(context.Context, int64, int64, AccessModeOptions) AccessMode
+	history     []PermsStoreAccessModeFuncCall
+	mutex       sync.Mutex
+}
+
+// AccessMode delegates to the next hook function in the queue and stores
+// the parameter and result values of this invocation.
+func (m *MockPermsStore) AccessMode(v0 context.Context, v1 int64, v2 int64, v3 AccessModeOptions) AccessMode {
+	r0 := m.AccessModeFunc.nextHook()(v0, v1, v2, v3)
+	m.AccessModeFunc.appendCall(PermsStoreAccessModeFuncCall{v0, v1, v2, v3, r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the AccessMode method of
+// the parent MockPermsStore instance is invoked and the hook queue is
+// empty.
+func (f *PermsStoreAccessModeFunc) SetDefaultHook(hook func(context.Context, int64, int64, AccessModeOptions) AccessMode) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// AccessMode method of the parent MockPermsStore instance invokes the hook
+// at the front of the queue and discards it. After the queue is empty, the
+// default hook function is invoked for any future action.
+func (f *PermsStoreAccessModeFunc) PushHook(hook func(context.Context, int64, int64, AccessModeOptions) AccessMode) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *PermsStoreAccessModeFunc) SetDefaultReturn(r0 AccessMode) {
+	f.SetDefaultHook(func(context.Context, int64, int64, AccessModeOptions) AccessMode {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *PermsStoreAccessModeFunc) PushReturn(r0 AccessMode) {
+	f.PushHook(func(context.Context, int64, int64, AccessModeOptions) AccessMode {
+		return r0
+	})
+}
+
+func (f *PermsStoreAccessModeFunc) nextHook() func(context.Context, int64, int64, AccessModeOptions) AccessMode {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *PermsStoreAccessModeFunc) appendCall(r0 PermsStoreAccessModeFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of PermsStoreAccessModeFuncCall objects
+// describing the invocations of this function.
+func (f *PermsStoreAccessModeFunc) History() []PermsStoreAccessModeFuncCall {
+	f.mutex.Lock()
+	history := make([]PermsStoreAccessModeFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// PermsStoreAccessModeFuncCall is an object that describes an invocation of
+// method AccessMode on an instance of MockPermsStore.
+type PermsStoreAccessModeFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 context.Context
+	// Arg1 is the value of the 2nd argument passed to this method
+	// invocation.
+	Arg1 int64
+	// Arg2 is the value of the 3rd argument passed to this method
+	// invocation.
+	Arg2 int64
+	// Arg3 is the value of the 4th argument passed to this method
+	// invocation.
+	Arg3 AccessModeOptions
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 AccessMode
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c PermsStoreAccessModeFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1, c.Arg2, c.Arg3}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c PermsStoreAccessModeFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// PermsStoreAuthorizeFunc describes the behavior when the Authorize method
+// of the parent MockPermsStore instance is invoked.
+type PermsStoreAuthorizeFunc struct {
+	defaultHook func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool
+	hooks       []func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool
+	history     []PermsStoreAuthorizeFuncCall
+	mutex       sync.Mutex
+}
+
+// Authorize delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockPermsStore) Authorize(v0 context.Context, v1 int64, v2 int64, v3 AccessMode, v4 AccessModeOptions) bool {
+	r0 := m.AuthorizeFunc.nextHook()(v0, v1, v2, v3, v4)
+	m.AuthorizeFunc.appendCall(PermsStoreAuthorizeFuncCall{v0, v1, v2, v3, v4, r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the Authorize method of
+// the parent MockPermsStore instance is invoked and the hook queue is
+// empty.
+func (f *PermsStoreAuthorizeFunc) SetDefaultHook(hook func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// Authorize method of the parent MockPermsStore instance invokes the hook
+// at the front of the queue and discards it. After the queue is empty, the
+// default hook function is invoked for any future action.
+func (f *PermsStoreAuthorizeFunc) PushHook(hook func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *PermsStoreAuthorizeFunc) SetDefaultReturn(r0 bool) {
+	f.SetDefaultHook(func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *PermsStoreAuthorizeFunc) PushReturn(r0 bool) {
+	f.PushHook(func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool {
+		return r0
+	})
+}
+
+func (f *PermsStoreAuthorizeFunc) nextHook() func(context.Context, int64, int64, AccessMode, AccessModeOptions) bool {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *PermsStoreAuthorizeFunc) appendCall(r0 PermsStoreAuthorizeFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of PermsStoreAuthorizeFuncCall objects
+// describing the invocations of this function.
+func (f *PermsStoreAuthorizeFunc) History() []PermsStoreAuthorizeFuncCall {
+	f.mutex.Lock()
+	history := make([]PermsStoreAuthorizeFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// PermsStoreAuthorizeFuncCall is an object that describes an invocation of
+// method Authorize on an instance of MockPermsStore.
+type PermsStoreAuthorizeFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 context.Context
+	// Arg1 is the value of the 2nd argument passed to this method
+	// invocation.
+	Arg1 int64
+	// Arg2 is the value of the 3rd argument passed to this method
+	// invocation.
+	Arg2 int64
+	// Arg3 is the value of the 4th argument passed to this method
+	// invocation.
+	Arg3 AccessMode
+	// Arg4 is the value of the 5th argument passed to this method
+	// invocation.
+	Arg4 AccessModeOptions
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 bool
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c PermsStoreAuthorizeFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1, c.Arg2, c.Arg3, c.Arg4}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c PermsStoreAuthorizeFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// PermsStoreSetRepoPermsFunc describes the behavior when the SetRepoPerms
+// method of the parent MockPermsStore instance is invoked.
+type PermsStoreSetRepoPermsFunc struct {
+	defaultHook func(context.Context, int64, map[int64]AccessMode) error
+	hooks       []func(context.Context, int64, map[int64]AccessMode) error
+	history     []PermsStoreSetRepoPermsFuncCall
+	mutex       sync.Mutex
+}
+
+// SetRepoPerms delegates to the next hook function in the queue and stores
+// the parameter and result values of this invocation.
+func (m *MockPermsStore) SetRepoPerms(v0 context.Context, v1 int64, v2 map[int64]AccessMode) error {
+	r0 := m.SetRepoPermsFunc.nextHook()(v0, v1, v2)
+	m.SetRepoPermsFunc.appendCall(PermsStoreSetRepoPermsFuncCall{v0, v1, v2, r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the SetRepoPerms method
+// of the parent MockPermsStore instance is invoked and the hook queue is
+// empty.
+func (f *PermsStoreSetRepoPermsFunc) SetDefaultHook(hook func(context.Context, int64, map[int64]AccessMode) error) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// SetRepoPerms method of the parent MockPermsStore instance invokes the
+// hook at the front of the queue and discards it. After the queue is empty,
+// the default hook function is invoked for any future action.
+func (f *PermsStoreSetRepoPermsFunc) PushHook(hook func(context.Context, int64, map[int64]AccessMode) error) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *PermsStoreSetRepoPermsFunc) SetDefaultReturn(r0 error) {
+	f.SetDefaultHook(func(context.Context, int64, map[int64]AccessMode) error {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *PermsStoreSetRepoPermsFunc) PushReturn(r0 error) {
+	f.PushHook(func(context.Context, int64, map[int64]AccessMode) error {
+		return r0
+	})
+}
+
+func (f *PermsStoreSetRepoPermsFunc) nextHook() func(context.Context, int64, map[int64]AccessMode) error {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *PermsStoreSetRepoPermsFunc) appendCall(r0 PermsStoreSetRepoPermsFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of PermsStoreSetRepoPermsFuncCall objects
+// describing the invocations of this function.
+func (f *PermsStoreSetRepoPermsFunc) History() []PermsStoreSetRepoPermsFuncCall {
+	f.mutex.Lock()
+	history := make([]PermsStoreSetRepoPermsFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// PermsStoreSetRepoPermsFuncCall is an object that describes an invocation
+// of method SetRepoPerms on an instance of MockPermsStore.
+type PermsStoreSetRepoPermsFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 context.Context
+	// Arg1 is the value of the 2nd argument passed to this method
+	// invocation.
+	Arg1 int64
+	// Arg2 is the value of the 3rd argument passed to this method
+	// invocation.
+	Arg2 map[int64]AccessMode
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 error
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c PermsStoreSetRepoPermsFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1, c.Arg2}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c PermsStoreSetRepoPermsFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// MockLoginSourceFileStore is a mock implementation of the
+// loginSourceFileStore interface (from the package
+// gogs.io/gogs/internal/db) used for unit testing.
+type MockLoginSourceFileStore struct {
+	// SaveFunc is an instance of a mock function object controlling the
+	// behavior of the method Save.
+	SaveFunc *LoginSourceFileStoreSaveFunc
+	// SetConfigFunc is an instance of a mock function object controlling
+	// the behavior of the method SetConfig.
+	SetConfigFunc *LoginSourceFileStoreSetConfigFunc
+	// SetGeneralFunc is an instance of a mock function object controlling
+	// the behavior of the method SetGeneral.
+	SetGeneralFunc *LoginSourceFileStoreSetGeneralFunc
+}
+
+// NewMockLoginSourceFileStore creates a new mock of the
+// loginSourceFileStore interface. All methods return zero values for all
+// results, unless overwritten.
+func NewMockLoginSourceFileStore() *MockLoginSourceFileStore {
+	return &MockLoginSourceFileStore{
+		SaveFunc: &LoginSourceFileStoreSaveFunc{
+			defaultHook: func() (r0 error) {
+				return
+			},
+		},
+		SetConfigFunc: &LoginSourceFileStoreSetConfigFunc{
+			defaultHook: func(interface{}) (r0 error) {
+				return
+			},
+		},
+		SetGeneralFunc: &LoginSourceFileStoreSetGeneralFunc{
+			defaultHook: func(string, string) {
+				return
+			},
+		},
+	}
+}
+
+// NewStrictMockLoginSourceFileStore creates a new mock of the
+// loginSourceFileStore interface. All methods panic on invocation, unless
+// overwritten.
+func NewStrictMockLoginSourceFileStore() *MockLoginSourceFileStore {
+	return &MockLoginSourceFileStore{
+		SaveFunc: &LoginSourceFileStoreSaveFunc{
+			defaultHook: func() error {
+				panic("unexpected invocation of MockLoginSourceFileStore.Save")
+			},
+		},
+		SetConfigFunc: &LoginSourceFileStoreSetConfigFunc{
+			defaultHook: func(interface{}) error {
+				panic("unexpected invocation of MockLoginSourceFileStore.SetConfig")
+			},
+		},
+		SetGeneralFunc: &LoginSourceFileStoreSetGeneralFunc{
+			defaultHook: func(string, string) {
+				panic("unexpected invocation of MockLoginSourceFileStore.SetGeneral")
+			},
+		},
+	}
+}
+
+// surrogateMockLoginSourceFileStore is a copy of the loginSourceFileStore
+// interface (from the package gogs.io/gogs/internal/db). It is redefined
+// here as it is unexported in the source package.
+type surrogateMockLoginSourceFileStore interface {
+	Save() error
+	SetConfig(interface{}) error
+	SetGeneral(string, string)
+}
+
+// NewMockLoginSourceFileStoreFrom creates a new mock of the
+// MockLoginSourceFileStore interface. All methods delegate to the given
+// implementation, unless overwritten.
+func NewMockLoginSourceFileStoreFrom(i surrogateMockLoginSourceFileStore) *MockLoginSourceFileStore {
+	return &MockLoginSourceFileStore{
+		SaveFunc: &LoginSourceFileStoreSaveFunc{
+			defaultHook: i.Save,
+		},
+		SetConfigFunc: &LoginSourceFileStoreSetConfigFunc{
+			defaultHook: i.SetConfig,
+		},
+		SetGeneralFunc: &LoginSourceFileStoreSetGeneralFunc{
+			defaultHook: i.SetGeneral,
+		},
+	}
+}
+
+// LoginSourceFileStoreSaveFunc describes the behavior when the Save method
+// of the parent MockLoginSourceFileStore instance is invoked.
+type LoginSourceFileStoreSaveFunc struct {
+	defaultHook func() error
+	hooks       []func() error
+	history     []LoginSourceFileStoreSaveFuncCall
+	mutex       sync.Mutex
+}
+
+// Save delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourceFileStore) Save() error {
+	r0 := m.SaveFunc.nextHook()()
+	m.SaveFunc.appendCall(LoginSourceFileStoreSaveFuncCall{r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the Save method of the
+// parent MockLoginSourceFileStore instance is invoked and the hook queue is
+// empty.
+func (f *LoginSourceFileStoreSaveFunc) SetDefaultHook(hook func() error) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// Save method of the parent MockLoginSourceFileStore instance invokes the
+// hook at the front of the queue and discards it. After the queue is empty,
+// the default hook function is invoked for any future action.
+func (f *LoginSourceFileStoreSaveFunc) PushHook(hook func() error) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourceFileStoreSaveFunc) SetDefaultReturn(r0 error) {
+	f.SetDefaultHook(func() error {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourceFileStoreSaveFunc) PushReturn(r0 error) {
+	f.PushHook(func() error {
+		return r0
+	})
+}
+
+func (f *LoginSourceFileStoreSaveFunc) nextHook() func() error {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourceFileStoreSaveFunc) appendCall(r0 LoginSourceFileStoreSaveFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourceFileStoreSaveFuncCall objects
+// describing the invocations of this function.
+func (f *LoginSourceFileStoreSaveFunc) History() []LoginSourceFileStoreSaveFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourceFileStoreSaveFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourceFileStoreSaveFuncCall is an object that describes an
+// invocation of method Save on an instance of MockLoginSourceFileStore.
+type LoginSourceFileStoreSaveFuncCall struct {
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 error
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourceFileStoreSaveFuncCall) Args() []interface{} {
+	return []interface{}{}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourceFileStoreSaveFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// LoginSourceFileStoreSetConfigFunc describes the behavior when the
+// SetConfig method of the parent MockLoginSourceFileStore instance is
+// invoked.
+type LoginSourceFileStoreSetConfigFunc struct {
+	defaultHook func(interface{}) error
+	hooks       []func(interface{}) error
+	history     []LoginSourceFileStoreSetConfigFuncCall
+	mutex       sync.Mutex
+}
+
+// SetConfig delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourceFileStore) SetConfig(v0 interface{}) error {
+	r0 := m.SetConfigFunc.nextHook()(v0)
+	m.SetConfigFunc.appendCall(LoginSourceFileStoreSetConfigFuncCall{v0, r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the SetConfig method of
+// the parent MockLoginSourceFileStore instance is invoked and the hook
+// queue is empty.
+func (f *LoginSourceFileStoreSetConfigFunc) SetDefaultHook(hook func(interface{}) error) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// SetConfig method of the parent MockLoginSourceFileStore instance invokes
+// the hook at the front of the queue and discards it. After the queue is
+// empty, the default hook function is invoked for any future action.
+func (f *LoginSourceFileStoreSetConfigFunc) PushHook(hook func(interface{}) error) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourceFileStoreSetConfigFunc) SetDefaultReturn(r0 error) {
+	f.SetDefaultHook(func(interface{}) error {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourceFileStoreSetConfigFunc) PushReturn(r0 error) {
+	f.PushHook(func(interface{}) error {
+		return r0
+	})
+}
+
+func (f *LoginSourceFileStoreSetConfigFunc) nextHook() func(interface{}) error {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourceFileStoreSetConfigFunc) appendCall(r0 LoginSourceFileStoreSetConfigFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourceFileStoreSetConfigFuncCall
+// objects describing the invocations of this function.
+func (f *LoginSourceFileStoreSetConfigFunc) History() []LoginSourceFileStoreSetConfigFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourceFileStoreSetConfigFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourceFileStoreSetConfigFuncCall is an object that describes an
+// invocation of method SetConfig on an instance of
+// MockLoginSourceFileStore.
+type LoginSourceFileStoreSetConfigFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 interface{}
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 error
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourceFileStoreSetConfigFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourceFileStoreSetConfigFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// LoginSourceFileStoreSetGeneralFunc describes the behavior when the
+// SetGeneral method of the parent MockLoginSourceFileStore instance is
+// invoked.
+type LoginSourceFileStoreSetGeneralFunc struct {
+	defaultHook func(string, string)
+	hooks       []func(string, string)
+	history     []LoginSourceFileStoreSetGeneralFuncCall
+	mutex       sync.Mutex
+}
+
+// SetGeneral delegates to the next hook function in the queue and stores
+// the parameter and result values of this invocation.
+func (m *MockLoginSourceFileStore) SetGeneral(v0 string, v1 string) {
+	m.SetGeneralFunc.nextHook()(v0, v1)
+	m.SetGeneralFunc.appendCall(LoginSourceFileStoreSetGeneralFuncCall{v0, v1})
+	return
+}
+
+// SetDefaultHook sets function that is called when the SetGeneral method of
+// the parent MockLoginSourceFileStore instance is invoked and the hook
+// queue is empty.
+func (f *LoginSourceFileStoreSetGeneralFunc) SetDefaultHook(hook func(string, string)) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// SetGeneral method of the parent MockLoginSourceFileStore instance invokes
+// the hook at the front of the queue and discards it. After the queue is
+// empty, the default hook function is invoked for any future action.
+func (f *LoginSourceFileStoreSetGeneralFunc) PushHook(hook func(string, string)) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourceFileStoreSetGeneralFunc) SetDefaultReturn() {
+	f.SetDefaultHook(func(string, string) {
+		return
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourceFileStoreSetGeneralFunc) PushReturn() {
+	f.PushHook(func(string, string) {
+		return
+	})
+}
+
+func (f *LoginSourceFileStoreSetGeneralFunc) nextHook() func(string, string) {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourceFileStoreSetGeneralFunc) appendCall(r0 LoginSourceFileStoreSetGeneralFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourceFileStoreSetGeneralFuncCall
+// objects describing the invocations of this function.
+func (f *LoginSourceFileStoreSetGeneralFunc) History() []LoginSourceFileStoreSetGeneralFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourceFileStoreSetGeneralFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourceFileStoreSetGeneralFuncCall is an object that describes an
+// invocation of method SetGeneral on an instance of
+// MockLoginSourceFileStore.
+type LoginSourceFileStoreSetGeneralFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 string
+	// Arg1 is the value of the 2nd argument passed to this method
+	// invocation.
+	Arg1 string
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourceFileStoreSetGeneralFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0, c.Arg1}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourceFileStoreSetGeneralFuncCall) Results() []interface{} {
+	return []interface{}{}
+}
+
+// MockLoginSourceFilesStore is a mock implementation of the
+// loginSourceFilesStore interface (from the package
+// gogs.io/gogs/internal/db) used for unit testing.
+type MockLoginSourceFilesStore struct {
+	// GetByIDFunc is an instance of a mock function object controlling the
+	// behavior of the method GetByID.
+	GetByIDFunc *LoginSourceFilesStoreGetByIDFunc
+	// LenFunc is an instance of a mock function object controlling the
+	// behavior of the method Len.
+	LenFunc *LoginSourceFilesStoreLenFunc
+	// ListFunc is an instance of a mock function object controlling the
+	// behavior of the method List.
+	ListFunc *LoginSourceFilesStoreListFunc
+	// UpdateFunc is an instance of a mock function object controlling the
+	// behavior of the method Update.
+	UpdateFunc *LoginSourceFilesStoreUpdateFunc
+}
+
+// NewMockLoginSourceFilesStore creates a new mock of the
+// loginSourceFilesStore interface. All methods return zero values for all
+// results, unless overwritten.
+func NewMockLoginSourceFilesStore() *MockLoginSourceFilesStore {
+	return &MockLoginSourceFilesStore{
+		GetByIDFunc: &LoginSourceFilesStoreGetByIDFunc{
+			defaultHook: func(int64) (r0 *LoginSource, r1 error) {
+				return
+			},
+		},
+		LenFunc: &LoginSourceFilesStoreLenFunc{
+			defaultHook: func() (r0 int) {
+				return
+			},
+		},
+		ListFunc: &LoginSourceFilesStoreListFunc{
+			defaultHook: func(ListLoginSourceOpts) (r0 []*LoginSource) {
+				return
+			},
+		},
+		UpdateFunc: &LoginSourceFilesStoreUpdateFunc{
+			defaultHook: func(*LoginSource) {
+				return
+			},
+		},
+	}
+}
+
+// NewStrictMockLoginSourceFilesStore creates a new mock of the
+// loginSourceFilesStore interface. All methods panic on invocation, unless
+// overwritten.
+func NewStrictMockLoginSourceFilesStore() *MockLoginSourceFilesStore {
+	return &MockLoginSourceFilesStore{
+		GetByIDFunc: &LoginSourceFilesStoreGetByIDFunc{
+			defaultHook: func(int64) (*LoginSource, error) {
+				panic("unexpected invocation of MockLoginSourceFilesStore.GetByID")
+			},
+		},
+		LenFunc: &LoginSourceFilesStoreLenFunc{
+			defaultHook: func() int {
+				panic("unexpected invocation of MockLoginSourceFilesStore.Len")
+			},
+		},
+		ListFunc: &LoginSourceFilesStoreListFunc{
+			defaultHook: func(ListLoginSourceOpts) []*LoginSource {
+				panic("unexpected invocation of MockLoginSourceFilesStore.List")
+			},
+		},
+		UpdateFunc: &LoginSourceFilesStoreUpdateFunc{
+			defaultHook: func(*LoginSource) {
+				panic("unexpected invocation of MockLoginSourceFilesStore.Update")
+			},
+		},
+	}
+}
+
+// surrogateMockLoginSourceFilesStore is a copy of the loginSourceFilesStore
+// interface (from the package gogs.io/gogs/internal/db). It is redefined
+// here as it is unexported in the source package.
+type surrogateMockLoginSourceFilesStore interface {
+	GetByID(int64) (*LoginSource, error)
+	Len() int
+	List(ListLoginSourceOpts) []*LoginSource
+	Update(*LoginSource)
+}
+
+// NewMockLoginSourceFilesStoreFrom creates a new mock of the
+// MockLoginSourceFilesStore interface. All methods delegate to the given
+// implementation, unless overwritten.
+func NewMockLoginSourceFilesStoreFrom(i surrogateMockLoginSourceFilesStore) *MockLoginSourceFilesStore {
+	return &MockLoginSourceFilesStore{
+		GetByIDFunc: &LoginSourceFilesStoreGetByIDFunc{
+			defaultHook: i.GetByID,
+		},
+		LenFunc: &LoginSourceFilesStoreLenFunc{
+			defaultHook: i.Len,
+		},
+		ListFunc: &LoginSourceFilesStoreListFunc{
+			defaultHook: i.List,
+		},
+		UpdateFunc: &LoginSourceFilesStoreUpdateFunc{
+			defaultHook: i.Update,
+		},
+	}
+}
+
+// LoginSourceFilesStoreGetByIDFunc describes the behavior when the GetByID
+// method of the parent MockLoginSourceFilesStore instance is invoked.
+type LoginSourceFilesStoreGetByIDFunc struct {
+	defaultHook func(int64) (*LoginSource, error)
+	hooks       []func(int64) (*LoginSource, error)
+	history     []LoginSourceFilesStoreGetByIDFuncCall
+	mutex       sync.Mutex
+}
+
+// GetByID delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourceFilesStore) GetByID(v0 int64) (*LoginSource, error) {
+	r0, r1 := m.GetByIDFunc.nextHook()(v0)
+	m.GetByIDFunc.appendCall(LoginSourceFilesStoreGetByIDFuncCall{v0, r0, r1})
+	return r0, r1
+}
+
+// SetDefaultHook sets function that is called when the GetByID method of
+// the parent MockLoginSourceFilesStore instance is invoked and the hook
+// queue is empty.
+func (f *LoginSourceFilesStoreGetByIDFunc) SetDefaultHook(hook func(int64) (*LoginSource, error)) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// GetByID method of the parent MockLoginSourceFilesStore instance invokes
+// the hook at the front of the queue and discards it. After the queue is
+// empty, the default hook function is invoked for any future action.
+func (f *LoginSourceFilesStoreGetByIDFunc) PushHook(hook func(int64) (*LoginSource, error)) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourceFilesStoreGetByIDFunc) SetDefaultReturn(r0 *LoginSource, r1 error) {
+	f.SetDefaultHook(func(int64) (*LoginSource, error) {
+		return r0, r1
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourceFilesStoreGetByIDFunc) PushReturn(r0 *LoginSource, r1 error) {
+	f.PushHook(func(int64) (*LoginSource, error) {
+		return r0, r1
+	})
+}
+
+func (f *LoginSourceFilesStoreGetByIDFunc) nextHook() func(int64) (*LoginSource, error) {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourceFilesStoreGetByIDFunc) appendCall(r0 LoginSourceFilesStoreGetByIDFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourceFilesStoreGetByIDFuncCall
+// objects describing the invocations of this function.
+func (f *LoginSourceFilesStoreGetByIDFunc) History() []LoginSourceFilesStoreGetByIDFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourceFilesStoreGetByIDFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourceFilesStoreGetByIDFuncCall is an object that describes an
+// invocation of method GetByID on an instance of MockLoginSourceFilesStore.
+type LoginSourceFilesStoreGetByIDFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 int64
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 *LoginSource
+	// Result1 is the value of the 2nd result returned from this method
+	// invocation.
+	Result1 error
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourceFilesStoreGetByIDFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourceFilesStoreGetByIDFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0, c.Result1}
+}
+
+// LoginSourceFilesStoreLenFunc describes the behavior when the Len method
+// of the parent MockLoginSourceFilesStore instance is invoked.
+type LoginSourceFilesStoreLenFunc struct {
+	defaultHook func() int
+	hooks       []func() int
+	history     []LoginSourceFilesStoreLenFuncCall
+	mutex       sync.Mutex
+}
+
+// Len delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourceFilesStore) Len() int {
+	r0 := m.LenFunc.nextHook()()
+	m.LenFunc.appendCall(LoginSourceFilesStoreLenFuncCall{r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the Len method of the
+// parent MockLoginSourceFilesStore instance is invoked and the hook queue
+// is empty.
+func (f *LoginSourceFilesStoreLenFunc) SetDefaultHook(hook func() int) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// Len method of the parent MockLoginSourceFilesStore instance invokes the
+// hook at the front of the queue and discards it. After the queue is empty,
+// the default hook function is invoked for any future action.
+func (f *LoginSourceFilesStoreLenFunc) PushHook(hook func() int) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourceFilesStoreLenFunc) SetDefaultReturn(r0 int) {
+	f.SetDefaultHook(func() int {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourceFilesStoreLenFunc) PushReturn(r0 int) {
+	f.PushHook(func() int {
+		return r0
+	})
+}
+
+func (f *LoginSourceFilesStoreLenFunc) nextHook() func() int {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourceFilesStoreLenFunc) appendCall(r0 LoginSourceFilesStoreLenFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourceFilesStoreLenFuncCall objects
+// describing the invocations of this function.
+func (f *LoginSourceFilesStoreLenFunc) History() []LoginSourceFilesStoreLenFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourceFilesStoreLenFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourceFilesStoreLenFuncCall is an object that describes an
+// invocation of method Len on an instance of MockLoginSourceFilesStore.
+type LoginSourceFilesStoreLenFuncCall struct {
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 int
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourceFilesStoreLenFuncCall) Args() []interface{} {
+	return []interface{}{}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourceFilesStoreLenFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// LoginSourceFilesStoreListFunc describes the behavior when the List method
+// of the parent MockLoginSourceFilesStore instance is invoked.
+type LoginSourceFilesStoreListFunc struct {
+	defaultHook func(ListLoginSourceOpts) []*LoginSource
+	hooks       []func(ListLoginSourceOpts) []*LoginSource
+	history     []LoginSourceFilesStoreListFuncCall
+	mutex       sync.Mutex
+}
+
+// List delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourceFilesStore) List(v0 ListLoginSourceOpts) []*LoginSource {
+	r0 := m.ListFunc.nextHook()(v0)
+	m.ListFunc.appendCall(LoginSourceFilesStoreListFuncCall{v0, r0})
+	return r0
+}
+
+// SetDefaultHook sets function that is called when the List method of the
+// parent MockLoginSourceFilesStore instance is invoked and the hook queue
+// is empty.
+func (f *LoginSourceFilesStoreListFunc) SetDefaultHook(hook func(ListLoginSourceOpts) []*LoginSource) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// List method of the parent MockLoginSourceFilesStore instance invokes the
+// hook at the front of the queue and discards it. After the queue is empty,
+// the default hook function is invoked for any future action.
+func (f *LoginSourceFilesStoreListFunc) PushHook(hook func(ListLoginSourceOpts) []*LoginSource) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourceFilesStoreListFunc) SetDefaultReturn(r0 []*LoginSource) {
+	f.SetDefaultHook(func(ListLoginSourceOpts) []*LoginSource {
+		return r0
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourceFilesStoreListFunc) PushReturn(r0 []*LoginSource) {
+	f.PushHook(func(ListLoginSourceOpts) []*LoginSource {
+		return r0
+	})
+}
+
+func (f *LoginSourceFilesStoreListFunc) nextHook() func(ListLoginSourceOpts) []*LoginSource {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourceFilesStoreListFunc) appendCall(r0 LoginSourceFilesStoreListFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourceFilesStoreListFuncCall objects
+// describing the invocations of this function.
+func (f *LoginSourceFilesStoreListFunc) History() []LoginSourceFilesStoreListFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourceFilesStoreListFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourceFilesStoreListFuncCall is an object that describes an
+// invocation of method List on an instance of MockLoginSourceFilesStore.
+type LoginSourceFilesStoreListFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 ListLoginSourceOpts
+	// Result0 is the value of the 1st result returned from this method
+	// invocation.
+	Result0 []*LoginSource
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourceFilesStoreListFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourceFilesStoreListFuncCall) Results() []interface{} {
+	return []interface{}{c.Result0}
+}
+
+// LoginSourceFilesStoreUpdateFunc describes the behavior when the Update
+// method of the parent MockLoginSourceFilesStore instance is invoked.
+type LoginSourceFilesStoreUpdateFunc struct {
+	defaultHook func(*LoginSource)
+	hooks       []func(*LoginSource)
+	history     []LoginSourceFilesStoreUpdateFuncCall
+	mutex       sync.Mutex
+}
+
+// Update delegates to the next hook function in the queue and stores the
+// parameter and result values of this invocation.
+func (m *MockLoginSourceFilesStore) Update(v0 *LoginSource) {
+	m.UpdateFunc.nextHook()(v0)
+	m.UpdateFunc.appendCall(LoginSourceFilesStoreUpdateFuncCall{v0})
+	return
+}
+
+// SetDefaultHook sets function that is called when the Update method of the
+// parent MockLoginSourceFilesStore instance is invoked and the hook queue
+// is empty.
+func (f *LoginSourceFilesStoreUpdateFunc) SetDefaultHook(hook func(*LoginSource)) {
+	f.defaultHook = hook
+}
+
+// PushHook adds a function to the end of hook queue. Each invocation of the
+// Update method of the parent MockLoginSourceFilesStore instance invokes
+// the hook at the front of the queue and discards it. After the queue is
+// empty, the default hook function is invoked for any future action.
+func (f *LoginSourceFilesStoreUpdateFunc) PushHook(hook func(*LoginSource)) {
+	f.mutex.Lock()
+	f.hooks = append(f.hooks, hook)
+	f.mutex.Unlock()
+}
+
+// SetDefaultReturn calls SetDefaultHook with a function that returns the
+// given values.
+func (f *LoginSourceFilesStoreUpdateFunc) SetDefaultReturn() {
+	f.SetDefaultHook(func(*LoginSource) {
+		return
+	})
+}
+
+// PushReturn calls PushHook with a function that returns the given values.
+func (f *LoginSourceFilesStoreUpdateFunc) PushReturn() {
+	f.PushHook(func(*LoginSource) {
+		return
+	})
+}
+
+func (f *LoginSourceFilesStoreUpdateFunc) nextHook() func(*LoginSource) {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if len(f.hooks) == 0 {
+		return f.defaultHook
+	}
+
+	hook := f.hooks[0]
+	f.hooks = f.hooks[1:]
+	return hook
+}
+
+func (f *LoginSourceFilesStoreUpdateFunc) appendCall(r0 LoginSourceFilesStoreUpdateFuncCall) {
+	f.mutex.Lock()
+	f.history = append(f.history, r0)
+	f.mutex.Unlock()
+}
+
+// History returns a sequence of LoginSourceFilesStoreUpdateFuncCall objects
+// describing the invocations of this function.
+func (f *LoginSourceFilesStoreUpdateFunc) History() []LoginSourceFilesStoreUpdateFuncCall {
+	f.mutex.Lock()
+	history := make([]LoginSourceFilesStoreUpdateFuncCall, len(f.history))
+	copy(history, f.history)
+	f.mutex.Unlock()
+
+	return history
+}
+
+// LoginSourceFilesStoreUpdateFuncCall is an object that describes an
+// invocation of method Update on an instance of MockLoginSourceFilesStore.
+type LoginSourceFilesStoreUpdateFuncCall struct {
+	// Arg0 is the value of the 1st argument passed to this method
+	// invocation.
+	Arg0 *LoginSource
+}
+
+// Args returns an interface slice containing the arguments of this
+// invocation.
+func (c LoginSourceFilesStoreUpdateFuncCall) Args() []interface{} {
+	return []interface{}{c.Arg0}
+}
+
+// Results returns an interface slice containing the results of this
+// invocation.
+func (c LoginSourceFilesStoreUpdateFuncCall) Results() []interface{} {
+	return []interface{}{}
+}

+ 3 - 2
internal/db/models.go

@@ -5,6 +5,7 @@
 package db
 
 import (
+	"context"
 	"database/sql"
 	"fmt"
 	"os"
@@ -208,7 +209,7 @@ type Statistic struct {
 	}
 }
 
-func GetStatistic() (stats Statistic) {
+func GetStatistic(ctx context.Context) (stats Statistic) {
 	stats.Counter.User = CountUsers()
 	stats.Counter.Org = CountOrganizations()
 	stats.Counter.PublicKey, _ = x.Count(new(PublicKey))
@@ -223,7 +224,7 @@ func GetStatistic() (stats Statistic) {
 	stats.Counter.Follow, _ = x.Count(new(Follow))
 	stats.Counter.Mirror, _ = x.Count(new(Mirror))
 	stats.Counter.Release, _ = x.Count(new(Release))
-	stats.Counter.LoginSource = LoginSources.Count()
+	stats.Counter.LoginSource = LoginSources.Count(ctx)
 	stats.Counter.Webhook, _ = x.Count(new(Webhook))
 	stats.Counter.Milestone, _ = x.Count(new(Milestone))
 	stats.Counter.Label, _ = x.Count(new(Label))

+ 4 - 1
internal/db/users.go

@@ -5,6 +5,7 @@
 package db
 
 import (
+	"context"
 	"fmt"
 	"strings"
 	"time"
@@ -80,6 +81,8 @@ func (err ErrLoginSourceMismatch) Error() string {
 }
 
 func (db *users) Authenticate(login, password string, loginSourceID int64) (*User, error) {
+	ctx := context.TODO()
+
 	login = strings.ToLower(login)
 
 	var query *gorm.DB
@@ -127,7 +130,7 @@ func (db *users) Authenticate(login, password string, loginSourceID int64) (*Use
 		createNewUser = true
 	}
 
-	source, err := LoginSources.GetByID(authSourceID)
+	source, err := LoginSources.GetByID(ctx, authSourceID)
 	if err != nil {
 		return nil, errors.Wrap(err, "get login source")
 	}

+ 1 - 1
internal/route/admin/admin.go

@@ -119,7 +119,7 @@ func Dashboard(c *context.Context) {
 	c.Data["BuildTime"] = conf.BuildTime
 	c.Data["BuildCommit"] = conf.BuildCommit
 
-	c.Data["Stats"] = db.GetStatistic()
+	c.Data["Stats"] = db.GetStatistic(c.Req.Context())
 	// FIXME: update periodically
 	updateSystemStatus()
 	c.Data["SysStatus"] = sysStatus

+ 17 - 15
internal/route/admin/auths.go

@@ -35,13 +35,13 @@ func Authentications(c *context.Context) {
 	c.PageIs("AdminAuthentications")
 
 	var err error
-	c.Data["Sources"], err = db.LoginSources.List(db.ListLoginSourceOpts{})
+	c.Data["Sources"], err = db.LoginSources.List(c.Req.Context(), db.ListLoginSourceOpts{})
 	if err != nil {
 		c.Error(err, "list login sources")
 		return
 	}
 
-	c.Data["Total"] = db.LoginSources.Count()
+	c.Data["Total"] = db.LoginSources.Count(c.Req.Context())
 	c.Success(AUTHS)
 }
 
@@ -159,13 +159,15 @@ func NewAuthSourcePost(c *context.Context, f form.Authentication) {
 		return
 	}
 
-	source, err := db.LoginSources.Create(db.CreateLoginSourceOpts{
-		Type:      auth.Type(f.Type),
-		Name:      f.Name,
-		Activated: f.IsActive,
-		Default:   f.IsDefault,
-		Config:    config,
-	})
+	source, err := db.LoginSources.Create(c.Req.Context(),
+		db.CreateLoginSourceOpts{
+			Type:      auth.Type(f.Type),
+			Name:      f.Name,
+			Activated: f.IsActive,
+			Default:   f.IsDefault,
+			Config:    config,
+		},
+	)
 	if err != nil {
 		if db.IsErrLoginSourceAlreadyExist(err) {
 			c.FormErr("Name")
@@ -177,7 +179,7 @@ func NewAuthSourcePost(c *context.Context, f form.Authentication) {
 	}
 
 	if source.IsDefault {
-		err = db.LoginSources.ResetNonDefault(source)
+		err = db.LoginSources.ResetNonDefault(c.Req.Context(), source)
 		if err != nil {
 			c.Error(err, "reset non-default login sources")
 			return
@@ -198,7 +200,7 @@ func EditAuthSource(c *context.Context) {
 	c.Data["SecurityProtocols"] = securityProtocols
 	c.Data["SMTPAuths"] = smtp.AuthTypes
 
-	source, err := db.LoginSources.GetByID(c.ParamsInt64(":authid"))
+	source, err := db.LoginSources.GetByID(c.Req.Context(), c.ParamsInt64(":authid"))
 	if err != nil {
 		c.Error(err, "get login source by ID")
 		return
@@ -216,7 +218,7 @@ func EditAuthSourcePost(c *context.Context, f form.Authentication) {
 
 	c.Data["SMTPAuths"] = smtp.AuthTypes
 
-	source, err := db.LoginSources.GetByID(c.ParamsInt64(":authid"))
+	source, err := db.LoginSources.GetByID(c.Req.Context(), c.ParamsInt64(":authid"))
 	if err != nil {
 		c.Error(err, "get login source by ID")
 		return
@@ -255,13 +257,13 @@ func EditAuthSourcePost(c *context.Context, f form.Authentication) {
 	source.IsActived = f.IsActive
 	source.IsDefault = f.IsDefault
 	source.Provider = provider
-	if err := db.LoginSources.Save(source); err != nil {
+	if err := db.LoginSources.Save(c.Req.Context(), source); err != nil {
 		c.Error(err, "update login source")
 		return
 	}
 
 	if source.IsDefault {
-		err = db.LoginSources.ResetNonDefault(source)
+		err = db.LoginSources.ResetNonDefault(c.Req.Context(), source)
 		if err != nil {
 			c.Error(err, "reset non-default login sources")
 			return
@@ -276,7 +278,7 @@ func EditAuthSourcePost(c *context.Context, f form.Authentication) {
 
 func DeleteAuthSource(c *context.Context) {
 	id := c.ParamsInt64(":authid")
-	if err := db.LoginSources.DeleteByID(id); err != nil {
+	if err := db.LoginSources.DeleteByID(c.Req.Context(), id); err != nil {
 		if db.IsErrLoginSourceInUse(err) {
 			c.Flash.Error(c.Tr("admin.auths.still_in_used"))
 		} else {

+ 4 - 4
internal/route/admin/users.go

@@ -46,7 +46,7 @@ func NewUser(c *context.Context) {
 
 	c.Data["login_type"] = "0-0"
 
-	sources, err := db.LoginSources.List(db.ListLoginSourceOpts{})
+	sources, err := db.LoginSources.List(c.Req.Context(), db.ListLoginSourceOpts{})
 	if err != nil {
 		c.Error(err, "list login sources")
 		return
@@ -62,7 +62,7 @@ func NewUserPost(c *context.Context, f form.AdminCrateUser) {
 	c.Data["PageIsAdmin"] = true
 	c.Data["PageIsAdminUsers"] = true
 
-	sources, err := db.LoginSources.List(db.ListLoginSourceOpts{})
+	sources, err := db.LoginSources.List(c.Req.Context(), db.ListLoginSourceOpts{})
 	if err != nil {
 		c.Error(err, "list login sources")
 		return
@@ -127,7 +127,7 @@ func prepareUserInfo(c *context.Context) *db.User {
 	c.Data["User"] = u
 
 	if u.LoginSource > 0 {
-		c.Data["LoginSource"], err = db.LoginSources.GetByID(u.LoginSource)
+		c.Data["LoginSource"], err = db.LoginSources.GetByID(c.Req.Context(), u.LoginSource)
 		if err != nil {
 			c.Error(err, "get login source by ID")
 			return nil
@@ -136,7 +136,7 @@ func prepareUserInfo(c *context.Context) *db.User {
 		c.Data["LoginSource"] = &db.LoginSource{}
 	}
 
-	sources, err := db.LoginSources.List(db.ListLoginSourceOpts{})
+	sources, err := db.LoginSources.List(c.Req.Context(), db.ListLoginSourceOpts{})
 	if err != nil {
 		c.Error(err, "list login sources")
 		return nil

+ 1 - 1
internal/route/api/v1/admin/user.go

@@ -22,7 +22,7 @@ func parseLoginSource(c *context.APIContext, u *db.User, sourceID int64, loginNa
 		return
 	}
 
-	source, err := db.LoginSources.GetByID(sourceID)
+	source, err := db.LoginSources.GetByID(c.Req.Context(), sourceID)
 	if err != nil {
 		if db.IsErrLoginSourceNotExist(err) {
 			c.ErrorStatus(http.StatusUnprocessableEntity, err)

+ 2 - 2
internal/route/user/auth.go

@@ -102,7 +102,7 @@ func Login(c *context.Context) {
 	}
 
 	// Display normal login page
-	loginSources, err := db.LoginSources.List(db.ListLoginSourceOpts{OnlyActivated: true})
+	loginSources, err := db.LoginSources.List(c.Req.Context(), db.ListLoginSourceOpts{OnlyActivated: true})
 	if err != nil {
 		c.Error(err, "list activated login sources")
 		return
@@ -149,7 +149,7 @@ func afterLogin(c *context.Context, u *db.User, remember bool) {
 func LoginPost(c *context.Context, f form.SignIn) {
 	c.Title("sign_in")
 
-	loginSources, err := db.LoginSources.List(db.ListLoginSourceOpts{OnlyActivated: true})
+	loginSources, err := db.LoginSources.List(c.Req.Context(), db.ListLoginSourceOpts{OnlyActivated: true})
 	if err != nil {
 		c.Error(err, "list activated login sources")
 		return