Use `getBranches` in more places, cleanup and test (#6556)

1 files changed, 3 insertions, 2 deletions

diff --git a/source/features/enable-file-links-in-compare-view.tsx b/source/features/enable-file-links-in-compare-view.tsx
index e64fe86d..e59dc4fb 100644
--- a/source/features/enable-file-links-in-compare-view.tsx
+++ b/source/features/enable-file-links-in-compare-view.tsx
@@ -6,16 +6,17 @@ import delegate, {DelegateEvent} from 'delegate-it';
 
 import features from '../feature-manager';
 import GitHubURL from '../github-helpers/github-url';
+import {getBranches} from '../github-helpers/pr-branches';
 
 /** Rebuilds the "View file" link because it points to the base repo and to the commit, instead of the head repo and its branch */
 function handlePRMenuOpening({delegateTarget: dropdown}: DelegateEvent): void {
 	dropdown.classList.add('rgh-actionable-link'); // Mark this as processed
 
-	const [nameWithOwner, headBranch] = select('.head-ref')!.title.split(':');
+	const {nameWithOwner, branch} = getBranches().head;
 	const filePath = dropdown.closest('[data-path]')!.getAttribute('data-path')!;
 
 	select('a[data-ga-click^="View file"]', dropdown)!
-		.pathname = [nameWithOwner, 'blob', headBranch, filePath].join('/'); // Do not replace with `GitHubURL`  #3152 #3111 #2595
+		.pathname = [nameWithOwner, 'blob', branch, filePath].join('/'); // Do not replace with `GitHubURL`  #3152 #3111 #2595
 }
 
 function handleCompareMenuOpening({delegateTarget: dropdown}: DelegateEvent): void {