0
@@ -5,14 +5,16 @@ module Piston
0
     class LockUnlock < Piston::Commands::Base
0
       attr_reader :options
0
 
0
-      def run(lock)
0
-        working_copy = Piston::WorkingCopy.guess(options[:wcdir])
0
+      def run(wcdir, lock)
0
+        working_copy = Piston::WorkingCopy.guess(wcdir)
0
         raise Piston::WorkingCopy::NotWorkingCopy if !working_copy.exist? || !working_copy.pistonized?
0
 
0
         values = working_copy.recall
0
         values["lock"] = lock
0
         working_copy.remember(values, values["handler"])
0
         working_copy.finalize
0
+
0
+        logger.info "Locked #{working_copy} against automatic updates"
0
       end
0
     end
0
   end

0
@@ -2,17 +2,28 @@ require File.dirname(__FILE__) + "/../test_helper"
0
 
0
 class TestLockUnlock < Test::Unit::TestCase
0
   def setup
0
-    @values = {:lock => "false", "handler" => { :repository => "repository" }}
0
+    @values = {"lock" => false}
0
+    @wcdir = "tmp/wcdir"
0
     @wc = mock("WorkingCopy")
0
   end
0
   
0
-  def test_run
0
-    run_and_verify do
0
+  def test_lock_working_copy
0
+    run_and_verify(true) do
0
       @wc.expects(:exist?).returns(true)
0
       @wc.expects(:pistonized?).returns(true)
0
       @wc.expects(:recall).returns(@values)
0
       @wc.expects(:finalize).returns(@values)
0
-      @wc.expects(:remember).with(@values, @values["handler"]).returns(@values)
0
+      @wc.expects(:remember).with(@values.merge("lock" => true), @values["handler"]).returns(@values)
0
+    end
0
+  end
0
+
0
+  def test_unlock_working_copy
0
+    run_and_verify(false) do
0
+      @wc.expects(:exist?).returns(true)
0
+      @wc.expects(:pistonized?).returns(true)
0
+      @wc.expects(:recall).returns(@values)
0
+      @wc.expects(:finalize).returns(@values)
0
+      @wc.expects(:remember).with(@values.merge("lock" => false), @values["handler"]).returns(@values)
0
     end
0
   end
0
   
0
@@ -34,14 +45,14 @@ class TestLockUnlock < Test::Unit::TestCase
0
   end
0
 
0
   private
0
-  def run_and_verify
0
+  def run_and_verify(lock=true)
0
     yield
0
-    piston_lock.run(true)
0
+    lock_unlock_command.run(@wcdir, lock)
0
   end
0
 
0
-  def piston_lock
0
-    Piston::WorkingCopy.expects(:guess).with("directory").returns(@wc)                                   
0
-    Piston::Commands::LockUnlock.new(:wcdir => "directory", :verbose => "verbose",
0
+  def lock_unlock_command
0
+    Piston::WorkingCopy.expects(:guess).with(@wcdir).returns(@wc)
0
+    Piston::Commands::LockUnlock.new(:verbose => "verbose",
0
                                      :quiet => "quiet", :force => "force")
0
   end
0
 end