Fix stubbed repo

2 files changed, 4 insertions, 8 deletions

diff --git a/app/models/repository.rb b/app/models/repository.rb
index 048795b22ab..be6502ebf5c 100644
--- a/app/models/repository.rb
+++ b/app/models/repository.rb
@@ -24,11 +24,11 @@ class Repository
   end
 
   def method_missing(m, *args, &block)
-    @raw_repository.send(m, *args, &block)
+    raw_repository.send(m, *args, &block)
   end
 
   def respond_to?(method)
-    return true if @raw_repository.respond_to?(method)
+    return true if raw_repository.respond_to?(method)
 
     super
   end
diff --git a/spec/support/stubbed_repository.rb b/spec/support/stubbed_repository.rb
index 68c380e6f4f..3dfdb3539fb 100644
--- a/spec/support/stubbed_repository.rb
+++ b/spec/support/stubbed_repository.rb
@@ -10,7 +10,7 @@ class Project
     if path == "empty" || !path
       nil
     else
-      GitLabTestRepo.new(path_with_namespace)
+      GitLabTestRepo.new(Rails.root.join('tmp', 'repositories', 'gitlabhq'), 'master')
     end
   end
 
@@ -39,11 +39,7 @@ class MergeRequest
   end
 end
 
-class GitLabTestRepo < Gitlab::Git::Repository
-  def repo
-    @repo ||= Grit::Repo.new(Rails.root.join('tmp', 'repositories', 'gitlabhq'))
-  end
-
+class GitLabTestRepo < Repository
   # patch repo size (in mb)
   def size
     12.45