add metadata, description support

Author: alexluck-sift

python/lib/sift_client/_internal/low_level_wrappers/test_results.py

@@ -320,6 +320,14 @@ def simulate_update_test_measurement_response(
                 updates["string_expected_value"] = proto.string_bounds.expected_value
             else:
                 updates["string_expected_value"] = None
+        if "description" in update_mask_paths:
+            updates["description"] = proto.description if proto.description else None
+        if "metadata" in update_mask_paths:
+            from sift_client.util.metadata import metadata_proto_to_dict
+
+            updates["metadata"] = metadata_proto_to_dict(proto.metadata) if proto.metadata else None  # type: ignore[arg-type]
+        if "channel_names" in update_mask_paths:
+            updates["channel_names"] = list(proto.channel_names) if proto.channel_names else None
 
         return existing.model_copy(update=updates)
 
@@ -1259,6 +1267,9 @@ def _measurement_create_from_simulated(
             unit=simulated.unit,
             numeric_bounds=simulated.numeric_bounds,
             string_expected_value=simulated.string_expected_value,
+            description=simulated.description,
+            metadata=simulated.metadata,
+            channel_names=simulated.channel_names,
         )
 
 

python/lib/sift_client/_tests/resources/test_test_results.py

@@ -63,6 +63,9 @@ def compare_test_measurement_fields(simulated: TestMeasurement, actual: TestMeas
     assert simulated.boolean_value == actual.boolean_value
     assert simulated.passed == actual.passed
     assert simulated.timestamp == actual.timestamp
+    assert simulated.description == actual.description
+    assert simulated.metadata == actual.metadata
+    assert simulated.channel_names == actual.channel_names
 
 
 def test_client_binding(sift_client):
@@ -280,6 +283,9 @@ def test_create_test_measurements(self, sift_client, tmp_path):
             unit="Celsius",
             passed=True,
             timestamp=step1.start_time,
+            description="Expected nominal: 25.0C",
+            metadata={"sensor": "thermocouple_a", "channel_index": 1},
+            channel_names=["temperature_celsius"],
         )
 
         # Create simulated measurement first
@@ -358,6 +364,9 @@ def test_update_test_measurements(self, sift_client, tmp_path):
             "passed": False,
             "string_expected_value": "1.10.4",
             "unit": "C",
+            "description": "Updated note after recalibration",
+            "metadata": {"part_number": "PN-002"},
+            "channel_names": ["firmware_version_channel"],
         }
 
         # Test update with log_file first
@@ -383,6 +392,9 @@ def test_update_test_measurements(self, sift_client, tmp_path):
         assert measurement2.passed == False
         assert measurement2.string_expected_value == "1.10.4"
         assert measurement2.unit == "C"
+        assert measurement2.description == "Updated note after recalibration"
+        assert measurement2.metadata == {"part_number": "PN-002"}
+        assert measurement2.channel_names == ["firmware_version_channel"]
         # Update the measurement using class function.
         measurement4 = measurement4.update(
             {

python/lib/sift_client/_tests/sift_types/test_results.py

@@ -3,15 +3,21 @@
 from __future__ import annotations
 
 import tempfile
+import warnings
 from datetime import datetime, timedelta, timezone
 from unittest.mock import MagicMock, call
 
 import pytest
+from sift.test_reports.v1.test_reports_pb2 import (
+    TestMeasurement as TestMeasurementProto,
+)
 
+from sift_client.sift_types.channel import Channel, ChannelDataType
 from sift_client.sift_types.test_report import (
     ErrorInfo,
     NumericBounds,
     TestMeasurement,
+    TestMeasurementCreate,
     TestMeasurementType,
     TestReport,
     TestStatus,
@@ -84,6 +90,9 @@ def mock_test_measurement(mock_client):
         unit="Celsius",
         passed=True,
         timestamp=simulated_time,
+        description="Expected nominal: 25.0C",
+        metadata={"part_number": "PN-001", "serial_number": "SN-42"},
+        channel_names=["temperature_celsius"],
     )
     test_measurement._apply_client_to_instance(mock_client)
     return test_measurement
@@ -271,3 +280,135 @@ def test_upload_attachment(self, mock_test_report, mock_test_step, mock_client):
                 ),
             ]
         )
+
+    def test_measurement_description_truncates_with_warning(self):
+        """Description over the server limit is truncated and a UserWarning is raised."""
+        over_limit = "x" * 2001
+        with pytest.warns(UserWarning, match="exceeds 2000 characters"):
+            create = TestMeasurementCreate(
+                test_step_id="step_123",
+                name="m",
+                passed=True,
+                timestamp=datetime.now(timezone.utc),
+                numeric_value=1.0,
+                description=over_limit,
+            )
+        assert create.description is not None
+        assert len(create.description) == 2000
+
+    def test_measurement_description_at_limit_is_not_truncated(self):
+        """A description exactly at the limit should not warn or truncate."""
+        at_limit = "x" * 2000
+        with warnings.catch_warnings():
+            warnings.simplefilter("error")  # any warning fails the test
+            create = TestMeasurementCreate(
+                test_step_id="step_123",
+                name="m",
+                passed=True,
+                timestamp=datetime.now(timezone.utc),
+                numeric_value=1.0,
+                description=at_limit,
+            )
+        assert create.description == at_limit
+
+    def test_measurement_channel_names_accepts_strings(self):
+        """channel_names accepts a homogeneous list of channel name strings."""
+        create = TestMeasurementCreate(
+            test_step_id="step_123",
+            name="m",
+            passed=True,
+            timestamp=datetime.now(timezone.utc),
+            numeric_value=1.0,
+            channel_names=["temperature_celsius", "pressure_psi"],
+        )
+        assert create.channel_names == ["temperature_celsius", "pressure_psi"]
+        proto = create.to_proto()
+        assert list(proto.channel_names) == ["temperature_celsius", "pressure_psi"]
+
+    def test_measurement_channel_names_accepts_channels(self):
+        """channel_names accepts a homogeneous list of Channel instances; names are extracted at serialization."""
+        now = datetime.now(timezone.utc)
+
+        def _channel(name: str) -> Channel:
+            return Channel(
+                proto=MagicMock(),
+                id_=f"channel_{name}",
+                name=name,
+                data_type=ChannelDataType.DOUBLE,
+                description="",
+                unit="",
+                bit_field_elements=[],
+                enum_types={},
+                asset_id="asset_1",
+                created_date=now,
+                modified_date=now,
+                created_by_user_id="user1",
+                modified_by_user_id="user1",
+            )
+
+        create = TestMeasurementCreate(
+            test_step_id="step_123",
+            name="m",
+            passed=True,
+            timestamp=now,
+            numeric_value=1.0,
+            channel_names=[_channel("temperature_celsius"), _channel("pressure_psi")],
+        )
+        proto = create.to_proto()
+        assert list(proto.channel_names) == ["temperature_celsius", "pressure_psi"]
+
+    def test_measurement_create_to_proto_writes_new_fields(self):
+        """to_proto carries description, metadata, and channel_names onto the proto."""
+        create = TestMeasurementCreate(
+            test_step_id="step_123",
+            name="m",
+            passed=True,
+            timestamp=datetime.now(timezone.utc),
+            numeric_value=1.0,
+            description="note",
+            metadata={"pn": "PN-001", "count": 3, "flag": True},
+            channel_names=["chan_a", "chan_b"],
+        )
+        proto = create.to_proto()
+        assert proto.description == "note"
+        assert list(proto.channel_names) == ["chan_a", "chan_b"]
+        proto_keys = {m.key.name for m in proto.metadata}
+        assert proto_keys == {"pn", "count", "flag"}
+
+    def test_measurement_from_proto_round_trips_new_fields(self):
+        """A proto with the new fields populated round-trips into TestMeasurement."""
+        ts = datetime.now(timezone.utc)
+        source = TestMeasurementCreate(
+            test_step_id="step_123",
+            name="m",
+            passed=True,
+            timestamp=ts,
+            numeric_value=1.0,
+            description="note",
+            metadata={"pn": "PN-001", "count": 3},
+            channel_names=["chan_a"],
+        ).to_proto()
+        source.measurement_id = "measurement_456"
+        source.test_report_id = "report_789"
+
+        measurement = TestMeasurement._from_proto(source)
+
+        assert measurement.description == "note"
+        assert measurement.metadata == {"pn": "PN-001", "count": 3}
+        assert measurement.channel_names == ["chan_a"]
+
+    def test_measurement_from_proto_handles_absent_new_fields(self):
+        """Proto with unset description/metadata/channel_names yields None on the model."""
+        proto = TestMeasurementProto(
+            measurement_id="measurement_abc",
+            measurement_type=TestMeasurementType.DOUBLE.value,
+            name="m",
+            test_step_id="step_123",
+            test_report_id="report_789",
+            passed=True,
+        )
+        proto.timestamp.FromDatetime(datetime.now(timezone.utc))
+        measurement = TestMeasurement._from_proto(proto)
+        assert measurement.description is None
+        assert measurement.metadata is None
+        assert measurement.channel_names is None

python/lib/sift_client/sift_types/test_report.py

@@ -1,9 +1,11 @@
 from __future__ import annotations
 
+import warnings
 from datetime import datetime, timezone
 from enum import Enum
 from typing import TYPE_CHECKING, ClassVar
 
+from pydantic import field_validator
 from sift.test_reports.v1.test_reports_pb2 import (
     CreateTestReportRequest as CreateTestReportRequestProto,
 )
@@ -34,8 +36,11 @@
     ModelUpdate,
 )
 from sift_client.sift_types._mixins.file_attachments import FileAttachmentsMixin
+from sift_client.sift_types.channel import Channel
 from sift_client.util.metadata import metadata_dict_to_proto, metadata_proto_to_dict
 
+_MEASUREMENT_DESCRIPTION_MAX_LEN = 2000
+
 if TYPE_CHECKING:
     from pathlib import Path
 
@@ -254,6 +259,19 @@ class TestMeasurementBase(ModelCreateUpdateBase):
     numeric_bounds: NumericBounds | None = None
     string_expected_value: str | None = None
     measurement_type: TestMeasurementType | None = None
+    description: str | None = None
+    metadata: dict[str, str | float | bool] | None = None
+    channel_names: list[str] | list[Channel] | None = None
+
+    _to_proto_helpers: ClassVar[dict[str, MappingHelper]] = {
+        "string_expected_value": MappingHelper(
+            proto_attr_path="string_bounds.expected_value", update_field="string_bounds"
+        ),
+        "unit": MappingHelper(proto_attr_path="unit.abbreviated_name", update_field="unit"),
+        "metadata": MappingHelper(
+            proto_attr_path="metadata", update_field="metadata", converter=metadata_dict_to_proto
+        ),
+    }
 
     def _get_proto_class(self) -> type[TestMeasurementProto]:
         return TestMeasurementProto
@@ -268,6 +286,18 @@ def _resolve_measurement_type(self) -> TestMeasurementType:
         else:
             raise ValueError("No measurement value provided")
 
+    @field_validator("description", mode="after")
+    @classmethod
+    def _truncate_description(cls, value: str | None) -> str | None:
+        if value is None or len(value) <= _MEASUREMENT_DESCRIPTION_MAX_LEN:
+            return value
+        warnings.warn(
+            f"TestMeasurement description exceeds {_MEASUREMENT_DESCRIPTION_MAX_LEN} "
+            f"characters ({len(value)}); truncating to fit the server limit.",
+            stacklevel=2,
+        )
+        return value[:_MEASUREMENT_DESCRIPTION_MAX_LEN]
+
 
 class TestMeasurementUpdate(TestMeasurementBase, ModelUpdate[TestMeasurementProto]):
     """Update model for TestMeasurement."""
@@ -276,13 +306,6 @@ class TestMeasurementUpdate(TestMeasurementBase, ModelUpdate[TestMeasurementProt
     passed: bool | None = None
     timestamp: datetime | None = None
 
-    _to_proto_helpers: ClassVar[dict[str, MappingHelper]] = {
-        "string_expected_value": MappingHelper(
-            proto_attr_path="string_bounds.expected_value", update_field="string_bounds"
-        ),
-        "unit": MappingHelper(proto_attr_path="unit.abbreviated_name", update_field="unit"),
-    }
-
     def _add_resource_id_to_proto(self, proto_msg: TestMeasurementProto):
         if self._resource_id is None:
             raise ValueError("Resource ID must be set before adding to proto")
@@ -327,6 +350,17 @@ def to_proto(self) -> TestMeasurementProto:
                 StringBoundsProto(expected_value=self.string_expected_value)
             )
 
+        if self.description:
+            proto.description = self.description
+
+        if self.metadata:
+            proto.metadata.extend(metadata_dict_to_proto(self.metadata))
+
+        if self.channel_names:
+            proto.channel_names.extend(
+                c.name if isinstance(c, Channel) else c for c in self.channel_names
+            )
+
         return proto
 
 
@@ -345,6 +379,9 @@ class TestMeasurement(BaseType[TestMeasurementProto, "TestMeasurement"]):
     string_expected_value: str | None = None
     passed: bool
     timestamp: datetime
+    description: str | None = None
+    metadata: dict[str, str | float | bool] | None = None
+    channel_names: list[str] | None = None
 
     @classmethod
     def _from_proto(
@@ -380,6 +417,9 @@ def _from_proto(
             else None,
             passed=proto.passed,
             timestamp=proto.timestamp.ToDatetime(tzinfo=timezone.utc),
+            description=proto.description if proto.description else None,
+            metadata=metadata_proto_to_dict(proto.metadata) if proto.metadata else None,  # type: ignore[arg-type]
+            channel_names=list(proto.channel_names) if proto.channel_names else None,
             _client=sift_client,
         )
 
@@ -410,6 +450,15 @@ def _to_proto(self) -> TestMeasurementProto:
                 StringBoundsProto(expected_value=self.string_expected_value)
             )
 
+        if self.description:
+            proto.description = self.description
+
+        if self.metadata:
+            proto.metadata.extend(metadata_dict_to_proto(self.metadata))
+
+        if self.channel_names:
+            proto.channel_names.extend(self.channel_names)
+
         return proto
 
     def update(