Fix coverage

Author: PawelLipski

git_machete/client/traverse.py

@@ -60,9 +60,9 @@ def _update_worktrees_cache_after_checkout(self, checked_out_branch: LocalBranch
         # Remove the old branch entry for this worktree (if any)
         # We need to find which branch was previously checked out in this worktree
         for branch, path in self.__worktrees_cache.items():
-            if path == current_worktree_path:
+            if path == current_worktree_path:  # pragma: no branch
                 del self.__worktrees_cache[branch]
-                break  # pragma: no branch
+                break
 
         # Add the new branch entry for this worktree
         self.__worktrees_cache[checked_out_branch] = current_worktree_path

git_machete/git_operations.py

@@ -598,9 +598,8 @@ def get_worktrees(self) -> Dict[LocalBranchShortName, str]:
             elif line.startswith('branch '):
                 # Format: "branch refs/heads/<branch-name>"
                 branch_ref = line[len('branch '):]
-                if branch_ref.startswith('refs/heads/'):
-                    current_branch = LocalBranchShortName.of(branch_ref[len('refs/heads/'):])
-                    is_detached = False
+                current_branch = LocalBranchFullName.of(branch_ref).to_short_name()
+                is_detached = False
             elif line.startswith('detached'):
                 is_detached = True
                 current_branch = None
@@ -640,11 +639,11 @@ def get_main_worktree_path(self) -> str:
 
         # The first worktree in the list is always the main worktree
         for line in result.stdout.strip().splitlines():
-            if line.startswith('worktree '):
+            if line.startswith('worktree '):  # pragma: no branch
                 return line[len('worktree '):]
 
         # Fallback: if no worktrees found, return current root dir
-        return self.get_root_dir()
+        return self.get_root_dir()  # pragma: no cover
 
     def checkout(self, branch: LocalBranchShortName) -> None:
         self._run_git("checkout", "--quiet", branch, "--", flush_caches=True)