From f2099c3811af8d0b8807145a7d2af28b9cf9d24a Mon Sep 17 00:00:00 2001 From: Sam Date: Tue, 14 Feb 2017 16:32:33 -0500 Subject: [PATCH] adjust API --- app/models/topic_list.rb | 4 ++-- spec/models/topic_list_spec.rb | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/app/models/topic_list.rb b/app/models/topic_list.rb index 43ec43f1fde..4a56c86d015 100644 --- a/app/models/topic_list.rb +++ b/app/models/topic_list.rb @@ -6,11 +6,11 @@ class TopicList cattr_accessor :preloaded_custom_fields self.preloaded_custom_fields = Set.new - def self.on_preload(blk) + def self.on_preload(&blk) (@preload ||= Set.new) << blk end - def self.cancel_preload(blk) + def self.cancel_preload(&blk) if @preload @preload.delete blk if @preload.length == 0 diff --git a/spec/models/topic_list_spec.rb b/spec/models/topic_list_spec.rb index fc1fe8f3dbb..29ea180008d 100644 --- a/spec/models/topic_list_spec.rb +++ b/spec/models/topic_list_spec.rb @@ -36,12 +36,12 @@ describe TopicList do preloaded_topic = true end - TopicList.on_preload(preloader) + TopicList.on_preload(&preloader) topic_list.topics expect(preloaded_topic).to eq(true) - TopicList.cancel_preload(preloader) + TopicList.cancel_preload(&preloader) end end