graphify-sf 0.3.2__tar.gz → 0.3.3__tar.gz

{graphify_sf-0.3.2 → graphify_sf-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphify-sf
-Version: 0.3.2
+Version: 0.3.3
 Summary: Turn any Salesforce SFDX project into a queryable knowledge graph — fully offline, no org connection required.
 Author: Ray Kuo
 License-Expression: MIT

{graphify_sf-0.3.2 → graphify_sf-0.3.3}/graphify_sf/extract/flow.py

@@ -122,15 +122,26 @@ def extract_flow(path: Path) -> dict:
             if sub_name:
                 edges.append(_make_edge(flow_nid, flow_id(sub_name), "invokes", "EXTRACTED", str_path))
 
-    # Object record operations
-    _record_tags = ("recordLookups", "recordCreates", "recordUpdates", "recordDeletes")
-    seen_objects: set[str] = set()
-    for tag in _record_tags:
+    # Object record operations. The relation stays "references" (a generic relation
+    # reused by many extractors — never change its semantics), but each edge carries an
+    # "operation" field (read/create/update/delete) so downstream can distinguish
+    # read-only access from writes. Dedup is by (object, operation) so a flow that both
+    # reads and updates the same object yields two edges, not one collapsed reference.
+    _record_op_tags = {
+        "recordLookups": "read",
+        "recordCreates": "create",
+        "recordUpdates": "update",
+        "recordDeletes": "delete",
+    }
+    seen_record_ops: set[tuple[str, str]] = set()
+    for tag, operation in _record_op_tags.items():
         for el in _find_all(root_el, tag, ns):
             obj = _find_text(el, "object", ns)
-            if obj and obj not in seen_objects:
-                seen_objects.add(obj)
-                edges.append(_make_edge(flow_nid, object_id(obj), "references", "EXTRACTED", str_path))
+            if obj and (obj, operation) not in seen_record_ops:
+                seen_record_ops.add((obj, operation))
+                edge = _make_edge(flow_nid, object_id(obj), "references", "EXTRACTED", str_path)
+                edge["operation"] = operation
+                edges.append(edge)
 
     # Subflows
     for subflow in _find_all(root_el, "subflows", ns):

{graphify_sf-0.3.2 → graphify_sf-0.3.3}/graphify_sf.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphify-sf
-Version: 0.3.2
+Version: 0.3.3
 Summary: Turn any Salesforce SFDX project into a queryable knowledge graph — fully offline, no org connection required.
 Author: Ray Kuo
 License-Expression: MIT

{graphify_sf-0.3.2 → graphify_sf-0.3.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "graphify-sf"
-version = "0.3.2"
+version = "0.3.3"
 description = "Turn any Salesforce SFDX project into a queryable knowledge graph — fully offline, no org connection required."
 readme = "README.md"
 requires-python = ">=3.10"

{graphify_sf-0.3.2 → graphify_sf-0.3.3}/tests/test_extract_flow.py

@@ -324,6 +324,122 @@ def test_extract_flow_action_call_subflow_creates_invokes_edge(tmp_path):
     assert "subflow" in invokes_edges[0]["target"].lower()
 
 
+# ---------------------------------------------------------------------------
+# Flow → Object record operations (read/create/update/delete)
+# ---------------------------------------------------------------------------
+
+READ_AND_UPDATE_SAME_OBJECT_XML = f"""\
+<?xml version="1.0" encoding="UTF-8"?>
+<Flow {_NS}>
+    <processType>AutoLaunchedFlow</processType>
+    <recordLookups>
+        <name>GetOpp</name>
+        <object>Opportunity</object>
+    </recordLookups>
+    <recordUpdates>
+        <name>UpdateOpp</name>
+        <object>Opportunity</object>
+    </recordUpdates>
+</Flow>
+"""
+
+PURE_UPDATE_XML = f"""\
+<?xml version="1.0" encoding="UTF-8"?>
+<Flow {_NS}>
+    <processType>AutoLaunchedFlow</processType>
+    <recordUpdates>
+        <name>UpdateOpp</name>
+        <object>Opportunity</object>
+    </recordUpdates>
+</Flow>
+"""
+
+ALL_FOUR_OPS_XML = f"""\
+<?xml version="1.0" encoding="UTF-8"?>
+<Flow {_NS}>
+    <processType>AutoLaunchedFlow</processType>
+    <recordLookups>
+        <name>GetAcc</name>
+        <object>Account</object>
+    </recordLookups>
+    <recordCreates>
+        <name>MakeContact</name>
+        <object>Contact</object>
+    </recordCreates>
+    <recordUpdates>
+        <name>UpdateAcc</name>
+        <object>Account</object>
+    </recordUpdates>
+    <recordDeletes>
+        <name>DeleteLead</name>
+        <object>Lead</object>
+    </recordDeletes>
+</Flow>
+"""
+
+
+def test_extract_flow_read_and_update_yields_two_edges(tmp_path):
+    """A flow that both reads and updates the same object yields TWO edges
+    (operation=read and operation=update), not one collapsed reference."""
+    from graphify_sf.extract.flow import extract_flow
+
+    f = tmp_path / "OppReadWrite.flow-meta.xml"
+    f.write_text(READ_AND_UPDATE_SAME_OBJECT_XML)
+
+    result = extract_flow(f)
+    opp_edges = [
+        e for e in result["edges"] if e.get("relation") == "references" and "opportunity" in e["target"].lower()
+    ]
+    assert len(opp_edges) == 2, "read+update on same object must not be deduped to one edge"
+    ops = {e.get("operation") for e in opp_edges}
+    assert ops == {"read", "update"}
+
+
+def test_extract_flow_pure_update_carries_operation(tmp_path):
+    """A recordUpdates element produces a references edge with operation='update'."""
+    from graphify_sf.extract.flow import extract_flow
+
+    f = tmp_path / "OppUpdate.flow-meta.xml"
+    f.write_text(PURE_UPDATE_XML)
+
+    result = extract_flow(f)
+    ref_edges = [e for e in result["edges"] if e.get("relation") == "references"]
+    assert len(ref_edges) == 1
+    assert ref_edges[0].get("operation") == "update"
+    assert "opportunity" in ref_edges[0]["target"].lower()
+
+
+def test_extract_flow_record_ops_relation_unchanged(tmp_path):
+    """Regression: record-op edges keep relation='references' (semantics unchanged);
+    only the new 'operation' field distinguishes them."""
+    from graphify_sf.extract.flow import extract_flow
+
+    f = tmp_path / "OppReadWrite.flow-meta.xml"
+    f.write_text(READ_AND_UPDATE_SAME_OBJECT_XML)
+
+    result = extract_flow(f)
+    record_op_edges = [e for e in result["edges"] if e.get("operation") is not None]
+    assert record_op_edges, "expected at least one record-op edge"
+    for e in record_op_edges:
+        assert e["relation"] == "references"
+        assert e["confidence"] == "EXTRACTED"
+
+
+def test_extract_flow_all_four_operations(tmp_path):
+    """All four record-op tags map to their operation: read/create/update/delete."""
+    from graphify_sf.extract.flow import extract_flow
+
+    f = tmp_path / "AllOps.flow-meta.xml"
+    f.write_text(ALL_FOUR_OPS_XML)
+
+    result = extract_flow(f)
+    by_op = {e.get("operation"): e["target"].lower() for e in result["edges"] if e.get("operation") is not None}
+    assert by_op.get("read") and "account" in by_op["read"]
+    assert by_op.get("create") and "contact" in by_op["create"]
+    assert by_op.get("update") and "account" in by_op["update"]
+    assert by_op.get("delete") and "lead" in by_op["delete"]
+
+
 # ---------------------------------------------------------------------------
 # Flow child element nodes (Decision, Screen, Loop, Assignment)
 # ---------------------------------------------------------------------------