refactor to improve database operability
This commit is contained in:
@@ -17,7 +17,7 @@ func expectNotRevoked(mock sqlmock.Sqlmock, jti any) {
|
||||
}
|
||||
|
||||
func TestValidateAccess_Success(t *testing.T) {
|
||||
gen, mock, cleanup := newGeneratorWithMockDB(t)
|
||||
gen, db, mock, cleanup := newGeneratorWithMockDB(t)
|
||||
defer cleanup()
|
||||
|
||||
tokenStr, _, err := gen.NewAccess(42, true, false)
|
||||
@@ -26,7 +26,7 @@ func TestValidateAccess_Success(t *testing.T) {
|
||||
// We don't know the JTI beforehand; match any arg
|
||||
expectNotRevoked(mock, sqlmock.AnyArg())
|
||||
|
||||
tx, err := gen.db.Begin()
|
||||
tx, err := db.Begin()
|
||||
require.NoError(t, err)
|
||||
defer tx.Rollback()
|
||||
|
||||
@@ -53,7 +53,7 @@ func TestValidateAccess_NoDB(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestValidateRefresh_Success(t *testing.T) {
|
||||
gen, mock, cleanup := newGeneratorWithMockDB(t)
|
||||
gen, db, mock, cleanup := newGeneratorWithMockDB(t)
|
||||
defer cleanup()
|
||||
|
||||
tokenStr, _, err := gen.NewRefresh(42, false)
|
||||
@@ -61,7 +61,7 @@ func TestValidateRefresh_Success(t *testing.T) {
|
||||
|
||||
expectNotRevoked(mock, sqlmock.AnyArg())
|
||||
|
||||
tx, err := gen.db.Begin()
|
||||
tx, err := db.Begin()
|
||||
require.NoError(t, err)
|
||||
defer tx.Rollback()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user