Make sure deleted object's entries are removed from object_types

Author: tomoasleep

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

@@ -305,7 +305,7 @@ def saves_to_multidb
   end
 
   describe "#delete" do
-    pending "deletes the given object from store" 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"]