github-mirror/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2025-01-19 04:52:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Refactor of PushSubscriptionPusher.

Browse Source
This commit is contained in:
Guo Xiang Tan 2018-05-09 08:14:14 +08:00
parent bbc85258c9
commit 2eb2f273a8
2 changed files with 23 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
app/services/push_notification_pusher.rb
View File

@ -33,13 +33,17 @@ class PushNotificationPusher
end
def self.subscribe(user, subscription, send_confirmation)
subscriptions = PushSubscription.where(user: user, data: subscription.to_json)
if subscriptions.length > 1
data = subscription.to_json
subscriptions = PushSubscription.where(user: user, data: data)
subscriptions_count = subscriptions.count
if subscriptions_count > 1
subscriptions.destroy_all
PushSubscription.create user: user, data: subscription.to_json
elsif subscriptions.length == 0
PushSubscription.create user: user, data: subscription.to_json
PushSubscription.create!(user: user, data: data)
elsif subscriptions_count == 0
PushSubscription.create!(user: user, data: data)
end
if send_confirmation == "true"
message = {
title: I18n.t("discourse_push_notifications.popup.confirm_title",
@ -55,7 +59,7 @@ class PushNotificationPusher
end
def self.unsubscribe(user, subscription)
PushSubscription.find_by(user: user, data: subscription.to_json)&.destroy
PushSubscription.find_by(user: user, data: subscription.to_json)&.destroy!
end
protected

36
spec/requests/push_notification_controller_spec.rb
View File

@ -61,29 +61,19 @@ describe PushNotificationController do
end
it "should not create duplicate subscriptions" do
post '/push_notifications/subscribe.json', params: {
username: user.username,
subscription: {
endpoint: "endpoint",
keys: {
p256dh: "256dh",
auth: "auth"
}
},
send_confirmation: false
}
post '/push_notifications/subscribe.json', params: {
username: user.username,
subscription: {
endpoint: "endpoint",
keys: {
p256dh: "256dh",
auth: "auth"
}
},
send_confirmation: false
}
2.times do
post '/push_notifications/subscribe.json', params: {
username: user.username,
subscription: {
endpoint: "endpoint",
keys: {
p256dh: "256dh",
auth: "auth"
}
},
send_confirmation: false
}
end
expect(response.status).to eq(200)
expect(user.push_subscriptions.count).to eq(1)