FlexMeasures / flexmeasures / 25832179460

88%

Build Type

push

github

Committed by web-flow

Commit Message

feat: add account_id filter for sources in search_beliefs (#2065)

* Initial plan

* data_sources: add account_id field to DataSource model

- Add account_id column (FK to account.id, nullable) to DataSource
- Add account relationship (backref 'data_sources' on Account)
- Update __table_args__ UniqueConstraint to include account_id
- Set self.account from user.account in __init__ when user is provided

* migration: add account_id to data_source table (9877450113f6)

- Add account_id column (nullable, FK to account.id)
- Data migration: populate account_id from user's account where user_id IS NOT NULL
- Update UniqueConstraint to include account_id column
- down_revision: 8b62f8129f34

* migration: replace N+1 loop with single UPDATE...FROM for account_id backfill

* tests/sensor_data: verify account_id is set on data source when posting sensor data

Context:
- DataSource.__init__ now sets self.account = user.account when a user is provided
- The POST sensor data endpoint calls get_or_create_source(current_user)
- Need to verify account_id is populated correctly on the resulting data source

Change:
- Added test_post_sensor_data_sets_account_id_on_data_source
- Posts sensor data as test_supplier_user_4@seita.nl
- Asserts data_source.account_id == user.account_id

* migration: use correlated subquery for account_id data migration; add changelog entry

Co-authored-by: Flix6x <30658763+Flix6x@users.noreply.github.com>
Agent-Logs-Url: https://github.com/FlexMeasures/flexmeasures/sessions/7969297e-88c2-4ace-99c9-ec74109f03c8

* data_sources: fix account_id assignment to avoid SAWarning and handle unflushed users

Co-authored-by: Flix6x <30658763+Flix6x@users.noreply.github.com>
Agent-Logs-Url: https://github.com/FlexMeasures/flexmeasures/sessions/7969297e-88c2-4ace-99c9-ec74109f03c8

* agents/architecture: add migration checklist and DataSource account_id invariants

Context:
- PR #2058 added account_id FK to DataSource table
- No migration checklist existe... (continued)

Coverage Stats

33 of 38 new or added lines in 7 files covered. (86.84%)

69 existing lines in 4 files now uncovered.

12579 of 15662 relevant lines covered (80.32%)

0.8 hits per line