FlexMeasures / flexmeasures / 26845067286

81%

push

github

feat: support source filters in sensor references for flex-model and flex-context (#2209)

* feat: add source filters to sensor references in flex-model/flex-context (issue #1807)

Co-authored-by: Flix6x <30658763+Flix6x@users.noreply.github.com>

* tests/schemas: add source-filter deserialization tests for SensorReference

Add 8 tests covering:
- backward compatibility: plain {sensor: id} still yields a Sensor
- source-types deserialization → SensorReference with source_types set
- exclude-source-types deserialization → SensorReference with exclude_source_types set
- sources list deserialization → SensorReference with DataSource objects resolved
- 4 parametrized invalid-payload cases (ValidationError expected)

Fix: test_sensor_reference_with_sources adds db fixture and calls
db.session.flush() to ensure DataSources created by setup_sources
(which never commits) get DB-assigned primary keys before the test
accesses seita_source.id.

Co-authored-by: Flix6x <30658763+Flix6x@users.noreply.github.com>

* tests/planning: add integration tests for SensorReference source filters

Co-authored-by: Flix6x <30658763+Flix6x@users.noreply.github.com>

* feat/sensors: add source-account filter to SensorReference

- Add `source_account: list[Account] | None` field to the SensorReference dataclass
- Import Account from flexmeasures.data.models.user and AccountIdField from
  flexmeasures.data.schemas.account in sensors.py
- Add `source_account` field (data_key="source-account") to SensorReferenceSchema
- Extend `_SOURCE_FILTER_KEYS` with "source-account" so the backward-compat
  plain-Sensor path is preserved when no filter keys are present
- Extract source-filter deserialization into `_deserialize_source_filters` helper
  to keep `_deserialize_dict` within the C901 complexity limit (≤13)

Co-authored-by: Flix6x <30658763+Flix6x@users.noreply.github.com>

* feat/planning/utils: forward source_account_ids from SensorReference to TimedBelief.search

Pass `source_accoun... (continued)

133 of 139 new or added lines in 6 files covered. (95.68%)

13121 of 16248 relevant lines covered (80.75%)

0.81 hits per line