1
0
Fork 0
mirror of https://github.com/binwiederhier/ntfy.git synced 2025-06-23 04:48:03 +02:00

More tests

This commit is contained in:
binwiederhier 2022-12-31 10:16:14 -05:00
parent cf7f118784
commit 0bb3c84b9e
3 changed files with 112 additions and 0 deletions

View file

@ -22,6 +22,10 @@ const (
userTokenExpiryDuration = 72 * time.Hour
)
var (
errNoTokenProvided = errors.New("no token provided")
)
// Manager-related queries
const (
createTablesQueriesNoTx = `
@ -239,6 +243,9 @@ func (a *Manager) CreateToken(user *User) (*Token, error) {
// ExtendToken sets the new expiry date for a token, thereby extending its use further into the future.
func (a *Manager) ExtendToken(user *User) (*Token, error) {
if user.Token == "" {
return nil, errNoTokenProvided
}
newExpires := time.Now().Add(userTokenExpiryDuration)
if _, err := a.db.Exec(updateTokenExpiryQuery, newExpires.Unix(), user.Name, user.Token); err != nil {
return nil, err