Merge pull request #1284 from tomoasleep/fix-registry-store-delete

Make sure deleted object's entries are removed from object_types

Author: lsegal

lib/yard/registry_store.rb

@@ -72,7 +72,12 @@ def put(key, value)
     # Deletes an object at a given path
     # @param [#to_sym] key the key to delete
     # @return [void]
-    def delete(key) @store.delete(key.to_sym) end
+    def delete(key)
+      if @store[key.to_sym]
+        @object_types[@store[key.to_sym].type].delete(key.to_s)
+        @store.delete(key.to_sym)
+      end
+    end
 
     # Gets all path names from the store. Loads the entire database
     # if +reload+ is +true+

spec/registry_store_spec.rb

@@ -304,6 +304,17 @@ def saves_to_multidb
     end
   end
 
+  describe "#delete" do
+    it "deletes the given object from store" do
+      @store.put(:YARD, @foo)
+      expect(@store.get(:YARD)).to be @foo
+      expect(@store.paths_for_type(:method)).to eq ["YARD"]
+      @store.delete(:YARD)
+      expect(@store.get(:YARD)).to be nil
+      expect(@store.paths_for_type(:method)).to eq []
+    end
+  end
+
   describe "#locale" do
     it "loads ./po/LOCALE_NAME.po" do
       fr_locale = I18n::Locale.new("fr")