Pass optional and extras specification to rewritten dependencies (#38)

* Pass optional and extras specification to rewritten dependency

* Don't need to set groups argument

poetry_monoranger_plugin/path_rewriter.py

@@ -97,4 +97,10 @@ def _pin_dependency(self, poetry: Poetry, dependency: DirectoryDependency):
         else:
             raise ValueError(f"Invalid version rewrite rule: {self.plugin_conf.version_rewrite_rule}")
 
-        return Dependency(name, pinned_version, groups=dependency.groups)
+        return Dependency(
+            name,
+            pinned_version,
+            groups=dependency.groups,
+            optional=dependency.is_optional(),
+            extras=dependency.extras,
+        )

tests/conftest.py

@@ -3,7 +3,7 @@
 
 import pytest
 from poetry.core.packages.dependency import Dependency
-from poetry.core.packages.dependency_group import DependencyGroup
+from poetry.core.packages.dependency_group import MAIN_GROUP, DependencyGroup
 from poetry.core.packages.directory_dependency import DirectoryDependency
 from poetry.poetry import Poetry
 
@@ -15,9 +15,17 @@ def mock_event_gen():
     def _factory(command_cls: type[Command], disable_cache: bool):
         from cleo.events.console_command_event import ConsoleCommandEvent
 
-        main_grp = DependencyGroup("main")
-        main_grp.add_dependency(DirectoryDependency("packageB", Path("../packageB"), develop=True))
+        main_grp = DependencyGroup(MAIN_GROUP)
         main_grp.add_dependency(Dependency("numpy", "==1.5.0"))
+        main_grp.add_dependency(
+            DirectoryDependency(
+                "packageB",
+                Path("../packageB"),
+                develop=True,
+                optional=True,
+                extras=["fast"],
+            )
+        )
 
         mock_command = Mock(spec=command_cls)
         mock_command.poetry = Mock(spec=Poetry)

tests/test_path_rewriter.py

@@ -35,6 +35,8 @@ def test_executes_path_rewriting_for_build_command(mock_event_gen, disable_cache
     new_dependencies = sorted(new_dependencies.dependencies, key=lambda x: x.name)
     for i, dep in enumerate(new_dependencies):
         assert dep.name == original_dependencies[i].name
+        assert dep.is_optional() == original_dependencies[i].is_optional()
+        assert dep.extras == original_dependencies[i].extras
         if isinstance(original_dependencies[i], DirectoryDependency):
             assert dep.pretty_constraint == "0.1.0"
         else: