spec/unit/berkshelf_spec.rb in berkshelf-0.4.0 vs spec/unit/berkshelf_spec.rb in berkshelf-0.5.0.rc1

@@ -58,14 +58,14 @@
           Berkshelf.instance_eval { @formatter = nil }
         end
 
         class CustomFormatter
           include Berkshelf::Formatters::AbstractFormatter
-          Berkshelf.formatters["custom"] = self
+          register_formatter :custom
         end
 
         before do
-          Berkshelf.set_format "custom"
+          Berkshelf.set_format :custom
         end
 
         it "should be the custom class" do
           Berkshelf.formatter.should be_an_instance_of(CustomFormatter)
         end