0
@@ -52,11 +52,12 @@ command :track do |remote, user|
0
   die "Already tracking #{user}" if helper.tracking?(user)
0
   repo = @helper.project if repo.nil?
0
   repo.chomp!(".git")
0
+  remote ||= user
0
 
0
   if options[:private] || options[:ssh]
0
-    git "remote add #{user} #{helper.private_url_for_user_and_repo(user, repo)}"
0
+    git "remote add #{remote} #{helper.private_url_for_user_and_repo(user, repo)}"
0
   else
0
-    git "remote add #{user} #{helper.public_url_for_user_and_repo(user, repo)}"
0
+    git "remote add #{remote} #{helper.public_url_for_user_and_repo(user, repo)}"
0
   end
0
 end
0
 

0
@@ -139,6 +139,24 @@ EOF
0
     end
0
   end
0
 
0
+  specify "track origin defunkt/github-gem should track defunkt/github-gem as the origin remote" do
0
+    running :track, "origin", "defunkt/github-gem" do
0
+      @helper.stub!(:url_for).with(:origin).and_return ""
0
+      @helper.stub!(:tracking?).and_return false
0
+      @command.should_receive(:git).with("remote add origin git://github.com/defunkt/github-gem.git")
0
+      stderr.should_not =~ /^Error/
0
+    end
0
+  end
0
+
0
+  specify "track --private origin defunkt/github-gem should track defunkt/github-gem as the origin remote using ssh" do
0
+    running :track, "--private", "origin", "defunkt/github-gem" do
0
+      @helper.stub!(:url_for).with(:origin).and_return ""
0
+      @helper.stub!(:tracking?).and_return false
0
+      @command.should_receive(:git).with("remote add origin git@github.com:defunkt/github-gem.git")
0
+      stderr.should_not =~ /^Error/
0
+    end
0
+  end
0
+
0
   # -- pull --
0
   specify "pull should die with no args" do
0
     running :pull do