Add spec for Pubsubhubbub::DistributionWorker. PuSH-deliver public items (#2954)

to all subscribers. IDN-normalize callback URLs for subscriptions on insert.
This commit is contained in:
Eugen Rochko
2017-05-10 02:55:43 +02:00
committed by GitHub
parent 29d8313b28
commit 3a38322a54
4 changed files with 67 additions and 15 deletions

View File

@ -6,9 +6,9 @@ class Pubsubhubbub::SubscribeService < BaseService
attr_reader :account, :callback, :secret, :lease_seconds
def call(account, callback, secret, lease_seconds)
@account = account
@callback = callback
@secret = secret
@account = account
@callback = Addressable::URI.parse(callback).normalize.to_s
@secret = secret
@lease_seconds = lease_seconds
process_subscribe
@ -52,7 +52,7 @@ class Pubsubhubbub::SubscribeService < BaseService
end
def blocked_domain?
DomainBlock.blocked? Addressable::URI.parse(callback).normalize.host
DomainBlock.blocked? Addressable::URI.parse(callback).host
end
def locate_subscription

View File

@ -1,11 +1,11 @@
# frozen_string_literal: true
class Pubsubhubbub::UnsubscribeService < BaseService
attr_reader :account, :callback_url
attr_reader :account, :callback
def call(account, callback_url)
@account = account
@callback_url = callback_url
def call(account, callback)
@account = account
@callback = Addressable::URI.parse(callback).normalize.to_s
process_unsubscribe
end
@ -26,6 +26,6 @@ class Pubsubhubbub::UnsubscribeService < BaseService
end
def subscription
@_subscription ||= Subscription.find_by(account: account, callback_url: callback_url)
@_subscription ||= Subscription.find_by(account: account, callback_url: callback)
end
end

View File

@ -10,15 +10,21 @@ class Pubsubhubbub::DistributionWorker
return if stream_entry.status&.direct_visibility?
account = stream_entry.account
payload = AtomSerializer.render(AtomSerializer.new.feed(account, [stream_entry]))
domains = account.followers_domains
@account = stream_entry.account
@payload = AtomSerializer.render(AtomSerializer.new.feed(@account, [stream_entry]))
@domains = @account.followers_domains
Subscription.where(account: account).active.select('id, callback_url').find_each do |subscription|
next unless domains.include?(Addressable::URI.parse(subscription.callback_url).host)
Pubsubhubbub::DeliveryWorker.perform_async(subscription.id, payload)
Subscription.where(account: @account).active.select('id, callback_url').find_each do |subscription|
next if stream_entry.hidden? && !allowed_to_receive?(subscription.callback_url)
Pubsubhubbub::DeliveryWorker.perform_async(subscription.id, @payload)
end
rescue ActiveRecord::RecordNotFound
true
end
private
def allowed_to_receive?(callback_url)
@domains.include?(Addressable::URI.parse(callback_url).host)
end
end