spec/cli/update_spec.rb in heirloom-0.4.0 vs spec/cli/update_spec.rb in heirloom-0.4.1rc1

@@ -10,21 +10,27 @@
                 :attribute      => 'att',
                 :updated_value  => 'val' }
     @logger_stub = stub :debug => true
     @config_mock = mock 'config'
     @archive_mock = mock 'archive'
+    @config_mock.stub :logger => @logger_mock
     Trollop.stub(:options).and_return options
     Heirloom::HeirloomLogger.should_receive(:new).with(:log_level => 'info').
                              and_return @logger_stub
     Heirloom::CLI::Update.any_instance.should_receive(:load_config).
                           with(:logger => @logger_stub,
                                :opts   => options).
                           and_return @config_mock
     Heirloom::Archive.should_receive(:new).
+                      with(:name => 'archive_name',
+                           :config => @config_mock).
+                      and_return @archive_mock
+    Heirloom::Archive.should_receive(:new).
                       with(:name   => 'archive_name',
                            :id     => '1.0.0',
                            :config => @config_mock).
                       and_return @archive_mock
+    @archive_mock.should_receive(:domain_exists?).and_return true
     @cli_update = Heirloom::CLI::Update.new
   end
 
   it "should update an attribute for a given id" do
     @archive_mock.should_receive(:update).