@@ -36,7 +36,7 @@ func TestRegenerateVapidKeypair(t *testing.T) {
36
36
37
37
db := database .New (sqlDB )
38
38
// Ensure there is no existing VAPID keypair.
39
- rows , err := db .GetNotificationVAPIDKeys (ctx )
39
+ rows , err := db .GetWebpushVAPIDKeys (ctx )
40
40
require .NoError (t , err )
41
41
require .Empty (t , rows )
42
42
@@ -48,13 +48,13 @@ func TestRegenerateVapidKeypair(t *testing.T) {
48
48
clitest .Start (t , inv )
49
49
50
50
pty .ExpectMatchContext (ctx , "Regenerating VAPID keypair..." )
51
- pty .ExpectMatchContext (ctx , "This will delete all existing push notification subscriptions." )
51
+ pty .ExpectMatchContext (ctx , "This will delete all existing webpush subscriptions." )
52
52
pty .ExpectMatchContext (ctx , "Are you sure you want to continue? (y/N)" )
53
53
pty .WriteLine ("y" )
54
54
pty .ExpectMatchContext (ctx , "VAPID keypair regenerated successfully." )
55
55
56
56
// Ensure the VAPID keypair was created.
57
- keys , err := db .GetNotificationVAPIDKeys (ctx )
57
+ keys , err := db .GetWebpushVAPIDKeys (ctx )
58
58
require .NoError (t , err )
59
59
require .NotEmpty (t , keys .VapidPublicKey )
60
60
require .NotEmpty (t , keys .VapidPrivateKey )
@@ -79,7 +79,7 @@ func TestRegenerateVapidKeypair(t *testing.T) {
79
79
u := dbgen .User (t , db , database.User {})
80
80
// Insert a few fake push subscriptions for each user.
81
81
for j := 0 ; j < 10 ; j ++ {
82
- _ = dbgen .NotificationPushSubscription (t , db , database.InsertNotificationPushSubscriptionParams {
82
+ _ = dbgen .WebpushSubscription (t , db , database.InsertWebpushSubscriptionParams {
83
83
UserID : u .ID ,
84
84
})
85
85
}
@@ -93,20 +93,20 @@ func TestRegenerateVapidKeypair(t *testing.T) {
93
93
clitest .Start (t , inv )
94
94
95
95
pty .ExpectMatchContext (ctx , "Regenerating VAPID keypair..." )
96
- pty .ExpectMatchContext (ctx , "This will delete all existing push notification subscriptions." )
96
+ pty .ExpectMatchContext (ctx , "This will delete all existing webpush subscriptions." )
97
97
pty .ExpectMatchContext (ctx , "Are you sure you want to continue? (y/N)" )
98
98
pty .WriteLine ("y" )
99
99
pty .ExpectMatchContext (ctx , "VAPID keypair regenerated successfully." )
100
100
101
101
// Ensure the VAPID keypair was created.
102
- keys , err := db .GetNotificationVAPIDKeys (ctx )
102
+ keys , err := db .GetWebpushVAPIDKeys (ctx )
103
103
require .NoError (t , err )
104
104
require .NotEmpty (t , keys .VapidPublicKey )
105
105
require .NotEmpty (t , keys .VapidPrivateKey )
106
106
107
107
// Ensure the push subscriptions were deleted.
108
108
var count int64
109
- rows , err := sqlDB .QueryContext (ctx , "SELECT COUNT(*) FROM notification_push_subscriptions " )
109
+ rows , err := sqlDB .QueryContext (ctx , "SELECT COUNT(*) FROM webpush_subscriptions " )
110
110
require .NoError (t , err )
111
111
t .Cleanup (func () {
112
112
_ = rows .Close ()
0 commit comments