Fix remote and upstream branch if upstream isn't <remote>/<branch>

Fixes #7 [Bump after signing cla]
Shopify · May 10, 2021 · b677645 · b677645
1 parent 8923666
commit b677645
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/git_chain/commands/push.rb b/lib/git_chain/commands/push.rb
@@ -33,13 +33,13 @@ def run(options)
             branch_remote, upstream_branch = upstream.split("/", 2)
             if remote && branch_remote != remote
               raise(Abort, "Multiple remotes detected: #{remote}, #{branch_remote}") if remote != branch_remote
-            else
+            elsif upstream_branch
               remote = branch_remote
             end
             upstreams[b] = upstream_branch
-          elsif options[:set_upstream]
-            upstreams[b] = b
           end
+
+          upstreams[b] ||= b if options[:set_upstream]
         end
 
         raise(Abort, "Nothing to push") if upstreams.empty?