FlexMeasures / flexmeasures / 29730736875

81%

main: 82%

push

github

Backport #2209, #2072 and #2309 to the 0.33 line, for a v0.34.0rc1 pre-release (#2320)

* 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: ... (continued)

266 of 287 new or added lines in 13 files covered. (92.68%)

13271 of 16428 relevant lines covered (80.78%)

0.81 hits per line