spec/gitmodel_spec.rb in gitmodel-0.0.4 vs spec/gitmodel_spec.rb in gitmodel-0.0.5

@@ -2,27 +2,27 @@
 
 describe GitModel do
 
   describe ".last_commit" do
     it "returns nil if there are no commits" do
-      GitModel.last_commit.should == nil
+      GitModel.last_commit(GitModel.default_branch).should == nil
     end
 
     it "returns the most recent commit on a branch" do
       index = Grit::Index.new(GitModel.repo)
       head = GitModel.repo.commits.first
       index.read_tree head.to_s
       index.add "foo", "foo"
       sha = index.commit nil, nil, nil, nil, 'master'
             
-      GitModel.last_commit.to_s.should == sha
+      GitModel.last_commit(GitModel.default_branch).to_s.should == sha
     end
   end
 
   describe ".current_tree" do
     it "returns nil if there are no commits" do
-      GitModel.current_tree.should == nil
+      GitModel.current_tree(GitModel.default_branch).should == nil
     end
 
     it "returns the tree for the most recent commit on a branch" do
       last_commit = mock('last_commit')
       last_commit.should_receive(:tree).and_return("yay, a tree!")
@@ -38,10 +38,10 @@
     end
 
     it "calls index! on each Persistable model class" do
       TestEntity.should_receive(:index!)
       TestEntity2.should_receive(:index!)
-      GitModel.index!
+      GitModel.index!(GitModel.default_branch)
     end
   end
 
 end