def test_assign_location_multiple_times(user: User, object: Object):
object_location_assignments = locations.get_object_location_assignments(
object.id)
assert object_location_assignments == []
location1 = locations.create_location("Location",
"This is an example location", None,
user.id)
location2 = locations.create_location("Location",
"This is an example location", None,
user.id)
locations.assign_location_to_object(
object.id, location1.id, None, user.id,
"This object is stored at this location")
locations.assign_location_to_object(
object.id, location2.id, None, user.id,
"This object is stored at another location")
assert len(locations.get_object_location_assignments(object.id)) == 2
object_location_assignment1, object_location_assignment2 = locations.get_object_location_assignments(
object.id)
assert object_location_assignment1.object_id == object.id
assert object_location_assignment1.location_id == location1.id
assert object_location_assignment1.user_id == user.id
assert object_location_assignment1.description == "This object is stored at this location"
assert object_location_assignment2.object_id == object.id
assert object_location_assignment2.location_id == location2.id
assert object_location_assignment2.user_id == user.id
assert object_location_assignment2.description == "This object is stored at another location"
def test_object_ids_for_location(user: User, action: Action):
data = {'name': {'_type': 'text', 'text': 'Object'}}
object1 = objects.create_object(user_id=user.id,
action_id=action.id,
data=data)
object2 = objects.create_object(user_id=user.id,
action_id=action.id,
data=data)
object3 = objects.create_object(user_id=user.id,
action_id=action.id,
data=data)
location1 = locations.create_location("Location",
"This is an example location", None,
user.id)
location2 = locations.create_location("Location",
"This is an example location", None,
user.id)
assert locations.get_object_ids_at_location(location1.id) == set()
assert locations.get_object_ids_at_location(location2.id) == set()
locations.assign_location_to_object(object1.id, location1.id, None,
user.id, "")
assert locations.get_object_ids_at_location(location1.id) == {object1.id}
assert locations.get_object_ids_at_location(location2.id) == set()
locations.assign_location_to_object(object2.id, location1.id, None,
user.id, "")
assert locations.get_object_ids_at_location(
location1.id) == {object1.id, object2.id}
assert locations.get_object_ids_at_location(location2.id) == set()
locations.assign_location_to_object(object3.id, location2.id, user.id,
user.id, "")
assert locations.get_object_ids_at_location(
location1.id) == {object1.id, object2.id}
assert locations.get_object_ids_at_location(location2.id) == {object3.id}
def test_get_location_tree(user: User):
child_location2 = locations.create_location("Location",
"This is an example location",
None, user.id)
parent_location = locations.create_location("Location",
"This is an example location",
None, user.id)
location = locations.create_location("Location",
"This is an example location",
parent_location.id, user.id)
child_location1 = locations.create_location("Location",
"This is an example location",
location.id, user.id)
locations.update_location(child_location2.id, "Location",
"This is an example location", location.id,
user.id)
locations_map, locations_tree = locations.get_locations_tree()
child_location2 = locations.get_location(child_location2.id)
assert locations_map == {
parent_location.id: parent_location,
location.id: location,
child_location1.id: child_location1,
child_location2.id: child_location2
}
assert locations_tree == {
parent_location.id: {
location.id: {
child_location1.id: {},
child_location2.id: {}
}
}
}
def test_update_location(user: User):
parent_location = locations.create_location("Parent Location",
"This is an example location",
None, user.id)
location = locations.create_location("Location",
"This is an example location", None,
user.id)
assert len(locations.get_locations()) == 2
locations.update_location(location.id, "Updated Location",
"This is a location description", None, user.id)
assert len(locations.get_locations()) == 2
location = locations.get_location(location.id)
assert location.name == "Updated Location"
assert location.description == "This is a location description"
assert location.parent_location_id is None
locations.update_location(location.id, "Updated Location",
"This is a location description",
parent_location.id, user.id)
location = locations.get_location(location.id)
assert location.parent_location_id == parent_location.id
user_log_entries = user_log.get_user_log_entries(user.id)
assert [
e for e in user_log_entries
if e.data.get('location_id', -1) == location.id
and e.type == UserLogEntryType.UPDATE_LOCATION
]
def test_create_location_with_invalid_parent_location(user: User):
parent_location = locations.create_location("Parent Location",
"This is an example location",
None, user.id)
assert len(locations.get_locations()) == 1
with pytest.raises(errors.LocationDoesNotExistError):
locations.create_location("Example Location",
"This is an example location",
parent_location.id + 1, user.id)
assert len(locations.get_locations()) == 1
def test_update_location_cyclic(user: User):
parent_location = locations.create_location("Parent Location",
"This is an example location",
None, user.id)
location = locations.create_location("Location",
"This is an example location",
parent_location.id, user.id)
with pytest.raises(errors.CyclicLocationError):
locations.update_location(parent_location.id, "Parent Location",
"This is an example location", location.id,
user.id)
def test_create_location(user: User):
assert len(locations.get_locations()) == 0
locations.create_location("Example Location",
"This is an example location", None, user.id)
assert len(locations.get_locations()) == 1
location = locations.get_locations()[0]
assert location.name == "Example Location"
assert location.description == "This is an example location"
assert location.parent_location_id is None
user_log_entries = user_log.get_user_log_entries(user.id)
assert user_log_entries[-1].type == UserLogEntryType.CREATE_LOCATION
assert user_log_entries[-1].data['location_id'] == location.id
def test_assign_location(user: User, object: Object):
object_location_assignment = locations.get_current_object_location_assignment(
object.id)
assert object_location_assignment is None
location = locations.create_location("Location",
"This is an example location", None,
user.id)
locations.assign_location_to_object(
object.id, location.id, None, user.id,
"This object is stored at this location")
object_location_assignment = locations.get_current_object_location_assignment(
object.id)
assert object_location_assignment.object_id == object.id
assert object_location_assignment.location_id == location.id
assert object_location_assignment.user_id == user.id
assert object_location_assignment.description == "This object is stored at this location"
user_log_entries = user_log.get_user_log_entries(user.id)
assert [
e for e in user_log_entries
if e.data.get('object_location_assignment_id',
-1) == object_location_assignment.id
and e.type == UserLogEntryType.ASSIGN_LOCATION
]
object_log_entries = object_log.get_object_log_entries(object.id)
assert [
e for e in object_log_entries
if e.data.get('object_location_assignment_id',
-1) == object_location_assignment.id
and e.type == ObjectLogEntryType.ASSIGN_LOCATION
]
def test_update_location_which_does_not_exist(user: User):
location = locations.create_location("Location",
"This is an example location", None,
user.id)
with pytest.raises(errors.LocationDoesNotExistError):
locations.update_location(location.id + 1, "Location",
"This is an example location", None, user.id)
def test_assign_location_to_object_which_does_not_exist(user: User):
location = locations.create_location("Location",
"This is an example location", None,
user.id)
with pytest.raises(errors.ObjectDoesNotExistError):
locations.assign_location_to_object(
42, location.id, None, user.id,
"This object is stored at this location")
def test_update_location_self_parent(user: User):
location = locations.create_location("Location",
"This is an example location", None,
user.id)
with pytest.raises(errors.CyclicLocationError):
locations.update_location(location.id, "Updated Location",
"This is a location description",
location.id, user.id)
def test_create_location_with_parent_location(user: User):
parent_location = locations.create_location("Parent Location",
"This is an example location",
None, user.id)
assert len(locations.get_locations()) == 1
user_log_entries = user_log.get_user_log_entries(user.id)
assert len(user_log_entries) == 1
location = locations.create_location("Example Location",
"This is an example location",
parent_location.id, user.id)
assert len(locations.get_locations()) == 2
location = locations.get_location(location.id)
assert location.name == "Example Location"
assert location.description == "This is an example location"
assert location.parent_location_id == parent_location.id
user_log_entries = user_log.get_user_log_entries(user.id)
assert len(user_log_entries) == 2
user_log_entry = [
e for e in user_log_entries
if e.data.get('location_id', -1) == location.id
][0]
assert user_log_entry.type == UserLogEntryType.CREATE_LOCATION
assert user_log_entry.data['location_id'] == location.id
