From 5e32058c13f34b46c69b7cdee6ccc0b7fe3b6df3 Mon Sep 17 00:00:00 2001 From: Joe Chen Date: Sat, 11 Jun 2022 11:54:11 +0800 Subject: db: use `context` and go-mockgen for `TwoFactorsStore` (#7045) --- internal/route/lfs/route_test.go | 22 ++++++++++++---------- 1 file changed, 12 insertions(+), 10 deletions(-) (limited to 'internal/route/lfs') diff --git a/internal/route/lfs/route_test.go b/internal/route/lfs/route_test.go index 6202695b..ee668fc6 100644 --- a/internal/route/lfs/route_test.go +++ b/internal/route/lfs/route_test.go @@ -31,7 +31,7 @@ func Test_authenticate(t *testing.T) { name string header http.Header mockUsersStore func() db.UsersStore - mockTwoFactorsStore *db.MockTwoFactorsStore + mockTwoFactorsStore func() db.TwoFactorsStore mockAccessTokensStore func() db.AccessTokensStore expStatusCode int expHeader http.Header @@ -56,10 +56,10 @@ func Test_authenticate(t *testing.T) { mock.AuthenticateFunc.SetDefaultReturn(&db.User{}, nil) return mock }, - mockTwoFactorsStore: &db.MockTwoFactorsStore{ - MockIsUserEnabled: func(userID int64) bool { - return true - }, + mockTwoFactorsStore: func() db.TwoFactorsStore { + mock := db.NewMockTwoFactorsStore() + mock.IsUserEnabledFunc.SetDefaultReturn(true) + return mock }, expStatusCode: http.StatusBadRequest, expHeader: http.Header{}, @@ -98,10 +98,10 @@ func Test_authenticate(t *testing.T) { mock.AuthenticateFunc.SetDefaultReturn(&db.User{ID: 1, Name: "unknwon"}, nil) return mock }, - mockTwoFactorsStore: &db.MockTwoFactorsStore{ - MockIsUserEnabled: func(userID int64) bool { - return false - }, + mockTwoFactorsStore: func() db.TwoFactorsStore { + mock := db.NewMockTwoFactorsStore() + mock.IsUserEnabledFunc.SetDefaultReturn(false) + return mock }, expStatusCode: http.StatusOK, expHeader: http.Header{}, @@ -133,7 +133,9 @@ func Test_authenticate(t *testing.T) { if test.mockUsersStore != nil { db.SetMockUsersStore(t, test.mockUsersStore()) } - db.SetMockTwoFactorsStore(t, test.mockTwoFactorsStore) + if test.mockTwoFactorsStore != nil { + db.SetMockTwoFactorsStore(t, test.mockTwoFactorsStore()) + } if test.mockAccessTokensStore != nil { db.SetMockAccessTokensStore(t, test.mockAccessTokensStore()) } -- cgit v1.2.3