Add worktrees.prune

ofir-petrushka · ofir-petrushka · commit 820c7b7f2705 · 2020-08-11T11:23:19.000+10:00
Signed-off-by: Ofir Petrushka &lt;hatkyinc@gmail.com&gt;
diff --git a/lib/git/lib.rb b/lib/git/lib.rb
@@ -337,6 +337,10 @@ def worktree_remove(dir)
       command('worktree', ['remove', dir])
     end
 
+    def worktree_prune
+      command('worktree', ['prune'])
+    end
+
     def list_files(ref_dir)
       dir = File.join(@git_dir, 'refs', ref_dir)
       files = []
diff --git a/lib/git/worktrees.rb b/lib/git/worktrees.rb
@@ -39,5 +39,9 @@ def to_s
       end
       out
     end
+
+    def prune
+      @base.lib.worktree_prune
+    end
   end
 end
diff --git a/tests/units/test_worktree.rb b/tests/units/test_worktree.rb
@@ -1,5 +1,5 @@
 #!/usr/bin/env ruby
-
+require 'fileutils'
 require File.dirname(__FILE__) + '/../test_helper'
 
 SAMPLE_LAST_COMMIT = '5e53019b3238362144c2766f02a2c00d91fcc023'
@@ -47,4 +47,16 @@ def test_worktree_add_and_remove
 
     assert_equal(@git.worktrees.size, 2)
   end
+
+  def test_worktree_prune
+    assert_equal(2, @git.worktrees.size)
+
+    @git.worktree('/tmp/pp1').add
+    assert_equal(3, @git.worktrees.size)
+    @git.worktrees.prune
+    assert_equal(2, @git.worktrees.size)
+    FileUtils.rm_rf('/tmp/pp1')
+    @git.worktrees.prune
+    assert_equal(1, @git.worktrees.size)
+  end
 end
