datajunction 0.0.262__tar.gz → 0.0.264__tar.gz

{datajunction-0.0.262 → datajunction-0.0.264}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: datajunction
-Version: 0.0.262
+Version: 0.0.264
 Summary: DataJunction client library for connecting to a DataJunction server
 Project-URL: repository, https://github.com/DataJunction/dj
 Author-email: DataJunction Authors <yian.shang@gmail.com>

{datajunction-0.0.262 → datajunction-0.0.264}/datajunction/__about__.py

@@ -2,4 +2,4 @@
 Version for Hatch
 """
 
-__version__ = "0.0.262"
+__version__ = "0.0.264"

{datajunction-0.0.262 → datajunction-0.0.264}/datajunction/cli.py

@@ -46,6 +46,7 @@ class DJCLI:
         namespace: str | None = None,
         verbose: bool = False,
         force: bool = False,
+        force_nodes: list[str] | None = None,
         allow_empty: bool = False,
         format: str = "text",
     ):
@@ -57,6 +58,7 @@ class DJCLI:
             namespace=namespace,
             verbose=verbose,
             force=force,
+            force_nodes=force_nodes,
             allow_empty=allow_empty,
             format=format,
         )
@@ -904,6 +906,16 @@ class DJCLI:
             action="store_true",
             help="Force update all nodes even if they are unchanged",
         )
+        push_parser.add_argument(
+            "--force-node",
+            action="append",
+            dest="force_nodes",
+            metavar="NODE",
+            help=(
+                "Force update only this node (full name) even if it is unchanged. "
+                "Repeat for several nodes"
+            ),
+        )
         push_parser.add_argument(
             "--allow-empty",
             action="store_true",
@@ -1524,6 +1536,7 @@ class DJCLI:
                     namespace=args.namespace,
                     verbose=args.verbose,
                     force=args.force,
+                    force_nodes=args.force_nodes,
                     allow_empty=args.allow_empty,
                     format=args.format,
                 )

{datajunction-0.0.262 → datajunction-0.0.264}/datajunction/deployment.py

@@ -139,6 +139,7 @@ class DeploymentService:
         console: Console | None = None,
         verbose: bool = False,
         force: bool = False,
+        force_nodes: list[str] | None = None,
         allow_empty: bool = False,
         format: str = "text",
     ):
@@ -196,6 +197,8 @@ class DeploymentService:
                     )
         if force:
             deployment_spec["force"] = True
+        if force_nodes:
+            deployment_spec["force_nodes"] = list(force_nodes)
         if allow_empty:
             deployment_spec["allow_empty"] = True
         deployment_data = self.client.deploy(deployment_spec)

{datajunction-0.0.262 → datajunction-0.0.264}/tests/test_cli.py

@@ -1526,6 +1526,46 @@ class TestImpactAnalysis:
         _, kwargs = mock_push.call_args
         assert kwargs.get("allow_empty") is False
 
+    def test_push_force_node_flag_passed_to_service(self, tmp_path):
+        """--force-node on `dj push` is repeatable and reaches the service as a list."""
+        from datajunction.cli import DJCLI
+
+        cli = DJCLI(builder_client=mock.MagicMock())
+        with patch.object(cli.deployment_service, "push") as mock_push:
+            mock_push.return_value = None
+            with patch.object(
+                sys,
+                "argv",
+                [
+                    "dj",
+                    "push",
+                    str(tmp_path),
+                    "--force-node",
+                    "ns.first",
+                    "--force-node",
+                    "ns.second",
+                ],
+            ):
+                cli.run()
+
+        mock_push.assert_called_once()
+        _, kwargs = mock_push.call_args
+        assert kwargs.get("force_nodes") == ["ns.first", "ns.second"]
+        assert kwargs.get("force") is False
+
+    def test_push_without_force_node_flag_defaults_none(self, tmp_path):
+        """Omitting --force-node on `dj push` must pass force_nodes=None."""
+        from datajunction.cli import DJCLI
+
+        cli = DJCLI(builder_client=mock.MagicMock())
+        with patch.object(cli.deployment_service, "push") as mock_push:
+            mock_push.return_value = None
+            with patch.object(sys, "argv", ["dj", "push", str(tmp_path)]):
+                cli.run()
+
+        _, kwargs = mock_push.call_args
+        assert kwargs.get("force_nodes") is None
+
 
 class TestDJCLIClientCreation:
     """Tests for DJCLI client creation from environment variables."""

{datajunction-0.0.262 → datajunction-0.0.264}/tests/test_deploy.py

@@ -1043,6 +1043,51 @@ def test_push_without_force_does_not_set_flag(monkeypatch, tmp_path):
     assert "force" not in spec_sent
 
 
+def test_push_force_nodes_sets_list_in_spec(monkeypatch, tmp_path):
+    """push(force_nodes=[...]) must send the names as `force_nodes` in the spec."""
+    (tmp_path / "dj.yaml").write_text(yaml.safe_dump({"namespace": "foo"}))
+    (tmp_path / "bar.yaml").write_text(yaml.safe_dump({"name": "foo.bar"}))
+
+    client = MagicMock()
+    client.deploy.return_value = {
+        "uuid": "abc",
+        "status": "success",
+        "results": [],
+        "namespace": "foo",
+    }
+
+    svc = DeploymentService(client, console=Console(file=io.StringIO()))
+    monkeypatch.setattr(time, "sleep", lambda _: None)
+
+    svc.push(tmp_path, force_nodes=["foo.bar"])
+
+    spec_sent = client.deploy.call_args[0][0]
+    assert spec_sent.get("force_nodes") == ["foo.bar"]
+    assert "force" not in spec_sent
+
+
+def test_push_without_force_nodes_does_not_set_list(monkeypatch, tmp_path):
+    """push() without force_nodes must not include it in the spec."""
+    (tmp_path / "dj.yaml").write_text(yaml.safe_dump({"namespace": "foo"}))
+    (tmp_path / "bar.yaml").write_text(yaml.safe_dump({"name": "foo.bar"}))
+
+    client = MagicMock()
+    client.deploy.return_value = {
+        "uuid": "abc",
+        "status": "success",
+        "results": [],
+        "namespace": "foo",
+    }
+
+    svc = DeploymentService(client, console=Console(file=io.StringIO()))
+    monkeypatch.setattr(time, "sleep", lambda _: None)
+
+    svc.push(tmp_path)
+
+    spec_sent = client.deploy.call_args[0][0]
+    assert "force_nodes" not in spec_sent
+
+
 def test_push_times_out(monkeypatch, tmp_path):
     # minimal project structure so _reconstruct_deployment_spec works
     (tmp_path / "dj.yaml").write_text(yaml.safe_dump({"namespace": "foo"}))