def test_embedded(self) -> None:
"""Test embedded method to check JSON-serialization and full join."""
assert Recommendation.from_id(1).to_json(join=True) == {
'id':
1,
'group_id':
1,
'tag_id':
1,
'time':
'2020-10-24 20:02:00' +
('' if config.backend == 'sqlite' else '-04:00'),
'priority':
1,
'object_id':
1,
'facility_id':
1,
'user_id':
2,
'forecast_id':
1,
'predicted_observation_id':
11,
'observation_id':
19,
'accepted':
True,
'rejected':
False,
'data': {},
'group':
RecommendationGroup.from_id(1).to_json(join=True),
'tag':
RecommendationTag.from_id(1).to_json(join=True),
'user':
User.from_id(2).to_json(join=True),
'facility':
Facility.from_id(1).to_json(join=True),
'object':
Object.from_id(1).to_json(join=True),
'forecast':
Forecast.from_id(1).to_json(join=True),
'predicted':
Observation.from_id(11).to_json(join=True),
'observed':
Observation.from_id(19).to_json(join=True),
}
def test_relationship_observed(self, testdata: TestData) -> None:
"""Test observation foreign key relationship on recommendation."""
for i, record in enumerate(testdata['recommendation']):
assert Recommendation.from_id(
i + 1).observed.id == record['observation_id']
def test_relationship_forecast(self, testdata: TestData) -> None:
"""Test forecast foreign key relationship on recommendation."""
for i, record in enumerate(testdata['recommendation']):
assert Recommendation.from_id(
i + 1).forecast.id == record['forecast_id']
def test_relationship_group(self, testdata: TestData) -> None:
"""Test recommendation_group foreign key relationship on recommendation."""
for i, record in enumerate(testdata['recommendation']):
assert Recommendation.from_id(i + 1).group.id == record['group_id']
def test_id_missing(self) -> None:
"""Test exception on missing recommendation `id`."""
with pytest.raises(NotFound):
Recommendation.from_id(-1)
def test_from_id(self, testdata: TestData) -> None:
"""Test loading recommendation from `id`."""
# NOTE: `id` not set until after insert
for i, record in enumerate(testdata['recommendation']):
assert Recommendation.from_id(i + 1).id == i + 1