mastofe

commit: e9810cbad61625771c4d0a8de25211795f03d82e
parent: 1027556614febb096f2186d3c678116b415c0b9b
Author: alpaca-tc <alpaca-tc@alpaca.tc>
Date:   Sat, 13 May 2017 22:56:12 +0900

Fixes NoMethodError: undefined method 'first' for nil:NilClass (#3036)


Diffstat:
M
app/models/status.rb
2
+-
M
spec/models/status_spec.rb
13
+++++++++++++

2 files changed, 14 insertions(+), 1 deletion(-)
diff --git a/app/models/status.rb b/app/models/status.rb
@@ -124,7 +124,7 @@ class Status < ApplicationRecord
   def ancestors(account = nil)
     ids      = Rails.cache.fetch("ancestors:#{id}") { (Status.find_by_sql(['WITH RECURSIVE search_tree(id, in_reply_to_id, path) AS (SELECT id, in_reply_to_id, ARRAY[id] FROM statuses WHERE id = ? UNION ALL SELECT statuses.id, statuses.in_reply_to_id, path || statuses.id FROM search_tree JOIN statuses ON statuses.id = search_tree.in_reply_to_id WHERE NOT statuses.id = ANY(path)) SELECT id FROM search_tree ORDER BY path DESC', id]) - [self]).pluck(:id) }
     statuses = Status.where(id: ids).group_by(&:id)
-    results  = ids.map { |id| statuses[id].first }
+    results  = ids.map { |id| statuses[id]&.first }.compact
     results  = results.reject { |status| filter_from_context?(status, account) }
 
     results
diff --git a/spec/models/status_spec.rb b/spec/models/status_spec.rb
@@ -179,6 +179,19 @@ RSpec.describe Status, type: :model do
     end
   end
 
+  describe '#ancestors' do
+    it 'ignores deleted records' do
+      first_status = Fabricate(:status, account: bob)
+      second_status = Fabricate(:status, thread: first_status, account: alice)
+
+      # Create cache and delete cached record
+      second_status.ancestors
+      first_status.destroy
+
+      expect(second_status.ancestors).to eq([])
+    end
+  end
+
   describe '#filter_from_context?' do
     pending
   end