def test_id_already_exists(self) -> None:
"""Test exception on recommendation `id` already exists."""
with pytest.raises(IntegrityError):
Recommendation.add({
'id': 1,
'group_id': 1,
'tag_id': 1,
'priority': 1,
'object_id': 1,
'facility_id': 1,
'user_id': 2
})
def test_for_user_with_group_id_2(self) -> None:
"""Test query for all recommendations for a given user and group."""
user_id = User.from_alias('tomb_raider').id
results = Recommendation.for_user(user_id, group_id=2)
assert len(results) == 4
for record in results:
assert record.user_id == user_id
assert record.group_id == 2
def test_next(self) -> None:
"""Test query for latest recommendation."""
user_id = User.from_alias('tomb_raider').id
response = Recommendation.next(user_id=user_id)
assert len(response) == 0 # NOTE: all accepted already
rec_id = Recommendation.for_user(user_id)[0].id
Recommendation.update(rec_id, accepted=False)
response = Recommendation.next(user_id=user_id)
assert len(response) == 1
Recommendation.update(rec_id, accepted=True)
response = Recommendation.next(user_id=user_id)
assert len(response) == 0
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_embedded_no_join(self, testdata: TestData) -> None:
"""Tests embedded method to check JSON-serialization."""
for data in testdata['recommendation']:
assert data == json_roundtrip(
Recommendation(**data).to_json(join=False))
def test_tuple(self, testdata: TestData) -> None:
"""Test tuple-conversion."""
for data in testdata['recommendation']:
recommendation = Recommendation.from_dict(data)
assert tuple(data.values()) == recommendation.to_tuple()
def test_dict(self, testdata: TestData) -> None:
"""Test round-trip of dict translations."""
for data in testdata['recommendation']:
recommendation = Recommendation.from_dict(data)
assert data == recommendation.to_dict()
def test_init(self, testdata: TestData) -> None:
"""Create recommendation instance and validate accessors."""
for data in testdata['recommendation']:
recommendation = Recommendation(**data)
for key, value in data.items():
assert getattr(recommendation, key) == value
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