[Repo] Allow to specify a branch for new repos.

lib/cocoapods/command/repo.rb

@@ -6,7 +6,7 @@ module Pod
       def self.banner
 %{Managing spec-repos:
 
-    $ pod repo add NAME URL
+    $ pod repo add NAME URL [BRANCH]
 
       Clones `URL' in the local spec-repos directory at `~/.cocoapods'. The
       remote can later be referred to by `NAME'.
@@ -26,6 +26,7 @@ module Pod
           unless (@name = argv.arguments[1]) && (@url = argv.arguments[2])
             raise Informative, "#{@action == 'add' ? 'Adding' : 'Updating the remote of'} a repo needs a `name' and a `url'."
           end
+          @branch = argv.arguments[3]
         when 'update'
           @name = argv.arguments[1]
         else
@@ -45,6 +46,7 @@ module Pod
         puts "Cloning spec repo `#{@name}' from `#{@url}'" unless config.silent?
         config.repos_dir.mkpath
         Dir.chdir(config.repos_dir) { git("clone '#{@url}' #{@name}") }
+        Dir.chdir(dir) { git("checkout #{@branch}") } if @branch
         check_versions(dir)
       end
 

spec/functional/command/repo_spec.rb

@@ -21,6 +21,12 @@ describe "Pod::Command::Repo" do
       git_config('private', 'remote.origin.url').should == fixture('spec-repos/master').to_s
     end
 
+    it "adds a spec-repo with on a specified branch" do
+      repo = command( 'repo' ,'add', 'new', fixture('spec-repos/master'), '0.6')
+      repo.run
+      Dir.chdir(repo.dir) { `git symbolic-ref HEAD` }.should.include? '0.6'
+    end
+
     it "updates a spec-repo" do
       repo1 = add_repo('repo1', fixture('spec-repos/master'))
       repo2 = add_repo('repo2', repo1.dir)

spec/spec_helper/temporary_repos.rb

@@ -30,7 +30,10 @@ module SpecHelper
     def add_repo(name, from)
       command = command('repo', 'add', name, from)
       command.run
-      Dir.chdir(command.dir) { `git checkout -b test >/dev/null 2>&1` }
+      Dir.chdir(command.dir) do
+        `git checkout -b test >/dev/null 2>&1`
+        `git branch --set-upstream test origin/test >/dev/null 2>&1`
+      end
       command
     end