update

tests/test_dotfiles.py

@@ -5,7 +5,7 @@ from pathlib import Path
 
 import pytest
 
-from flow.commands.dotfiles import _discover_packages, _walk_package
+from flow.commands.dotfiles import _discover_packages, _resolve_edit_target, _walk_package
 from flow.core.config import AppConfig, FlowContext
 from flow.core.console import ConsoleLogger
 from flow.core.platform import PlatformInfo
@@ -64,3 +64,17 @@ def test_walk_package(dotfiles_tree):
     targets = {str(t) for _, t in pairs}
     assert str(home / ".zshrc") in targets
     assert str(home / ".zshenv") in targets
+
+
+def test_resolve_edit_target_package(dotfiles_tree):
+    target = _resolve_edit_target("zsh", dotfiles_dir=dotfiles_tree)
+    assert target == dotfiles_tree / "common" / "zsh"
+
+
+def test_resolve_edit_target_repo_path(dotfiles_tree):
+    target = _resolve_edit_target("common/zsh/.zshrc", dotfiles_dir=dotfiles_tree)
+    assert target == dotfiles_tree / "common" / "zsh" / ".zshrc"
+
+
+def test_resolve_edit_target_missing_returns_none(dotfiles_tree):
+    assert _resolve_edit_target("does-not-exist", dotfiles_dir=dotfiles_tree) is None