UpstreamPR8010 Cmrd-senya fix archive edge cases

lib/archive_importer/entity_importer.rb

@@ -15,8 +15,11 @@ def import
     rescue DiasporaFederation::Entities::Signable::SignatureVerificationFailed,
            DiasporaFederation::Discovery::InvalidDocument,
            DiasporaFederation::Discovery::DiscoveryError,
+           DiasporaFederation::Federation::Fetcher::NotFetchable,
+           OwnRelayableImporter::NoParentError,
            ActiveRecord::RecordInvalid => e
       logger.warn "#{self}: #{e}"
+      self.persisted_object = nil
     end
 
     attr_reader :json

lib/archive_importer/own_entity_importer.rb

@@ -21,7 +21,9 @@ def substitute_author
     attr_reader :old_author_id
 
     def persisted_object
-      @persisted_object ||= (instance if real_author == old_author_id)
+      return @persisted_object if defined?(@persisted_object)
+
+      @persisted_object = (instance if real_author == old_author_id)
     end
 
     def real_author

lib/archive_importer/own_relayable_importer.rb

@@ -2,6 +2,8 @@
 
 class ArchiveImporter
   class OwnRelayableImporter < OwnEntityImporter
+    class NoParentError < RuntimeError; end
+
     def entity
       fetch_parent(symbolized_entity_data)
       entity_class.new(symbolized_entity_data)
@@ -19,6 +21,8 @@ def fetch_parent(data)
         break entity_class::PARENT_TYPE if entity_class.const_defined?(:PARENT_TYPE)
       }
       entity = Diaspora::Federation::Mappings.model_class_for(type).find_by(guid: data.fetch(:parent_guid))
+      raise NoParentError if entity.nil?
+
       data[:parent] = Diaspora::Federation::Entities.related_entity(entity)
     end
   end

spec/lib/archive_importer/own_relayable_importer_spec.rb

@@ -28,4 +28,27 @@
       ).to_json
     }
   end
+
+  context "with comment with unknown parent" do
+    let(:new_user) { FactoryGirl.create(:user) }
+    let(:instance) { described_class.new(entity_json, new_user) }
+    let(:entity_json) { JSON.parse(<<~JSON) }
+      {
+        "entity_data" : {
+           "created_at" : "2015-10-19T13:58:16Z",
+           "guid" : "#{UUID.generate(:compact)}",
+           "text" : "test post",
+           "author" : "author@example.com",
+           "parent_guid": "#{UUID.generate(:compact)}"
+        },
+        "entity_type": "comment"
+      }
+    JSON
+
+    it "doesn't raise error" do
+      expect {
+        instance.import
+      }.not_to raise_error
+    end
+  end
 end

spec/lib/archive_importer/post_importer_spec.rb

@@ -113,5 +113,39 @@
         end
       end
     end
+
+    context "with reshare" do
+      let!(:author) { FactoryGirl.create(:person, diaspora_handle: "author@example.com") }
+      let(:entity_json) { JSON.parse(<<~JSON) }
+        {
+          "entity_data" : {
+             "created_at" : "2015-10-19T13:58:16Z",
+             "guid" : "#{UUID.generate(:compact)}",
+             "text" : "test post",
+             "author" : "author@example.com",
+             "root_author": "root_author@remote-pod.com",
+             "root_guid":   "#{UUID.generate(:compact)}"
+          },
+          "entity_type": "reshare"
+        }
+      JSON
+
+      context "when a remote pod responds 403 to discovery requests" do
+        before do
+          stub_request(:get, "https://remote-pod.com/.well-known/webfinger?resource=acct:root_author@remote-pod.com")
+            .to_return(status: 403, body: "", headers: {})
+          stub_request(:get, "https://remote-pod.com/.well-known/host-meta")
+            .to_return(status: 403, body: "", headers: {})
+          stub_request(:get, "http://remote-pod.com/.well-known/host-meta")
+            .to_return(status: 403, body: "", headers: {})
+        end
+
+        it "doesn't raise error" do
+          expect {
+            instance.import
+          }.not_to raise_error
+        end
+      end
+    end
   end
 end