added update-ref

scott Chacon · scott Chacon · commit 3c1b3f01465a · 2007-11-16T11:31:08.000-08:00
diff --git a/lib/git/base.rb b/lib/git/base.rb
@@ -6,6 +6,8 @@ class Base
     @repository = nil
     @index = nil
 
+    @lib = nil
+    
     # opens a bare Git Repository - no working directory options
     def self.bare(git_dir)
       self.new :repository => git_dir
@@ -376,6 +378,10 @@ def write_and_commit_tree(opts = {})
       commit_tree(tree, opts)
     end
       
+    def update_ref(branch, commit)
+      branch(branch).update_ref(commit)
+    end
+    
     def ls_files
       self.lib.ls_files
     end
diff --git a/lib/git/branch.rb b/lib/git/branch.rb
@@ -75,6 +75,10 @@ def merge(branch = nil, message = nil)
       end
     end
     
+    def update_ref(commit)
+      @base.lib.update_ref(@full, commit)
+    end
+    
     def to_a
       [@full]
     end
diff --git a/lib/git/lib.rb b/lib/git/lib.rb
@@ -376,6 +376,11 @@ def commit_tree(tree, opts = {})
       command('commit-tree', arr_opts)
     end
     
+    def update_ref(branch, commit)
+      command('update-ref', [branch.to_s, commit.to_s])
+    end
+    
+    
     # creates an archive file
     #
     # options
diff --git a/tests/units/test_tree_ops.rb b/tests/units/test_tree_ops.rb
@@ -95,8 +95,15 @@ def test_read_tree
           assert(index['b3/test-file3'])
           g.commit('hi')
         end
-        
         assert(c.commit?)
+
+        files = g.ls_files
+        assert(!files['b1/example.txt'])
+        
+        g.branch('newbranch').update_ref(c)        
+        g.checkout('newbranch')
+        assert(!files['b1/example.txt'])
+        
         assert_equal('b40f7a9072cdec637725700668f8fdebe39e6d38', c.gtree.sha)
         
       end
