def test_delete_point_cloud_feature(celery_task_always_eager, projects_fixture,
point_cloud_fixture,
lidar_las1pt2_file_fixture):
PointCloudService.fromFileObj(point_cloud_fixture.id,
lidar_las1pt2_file_fixture,
lidar_las1pt2_file_fixture.name)
point_cloud = db_session.query(PointCloud).get(1)
feature_asset_path = get_asset_path(point_cloud.feature.assets[0].path)
FeaturesService.delete(point_cloud.feature.id)
assert db_session.query(PointCloud).count() == 1
assert db_session.query(PointCloud).get(1).feature is None
assert db_session.query(Feature).count() == 0
assert db_session.query(FeatureAsset).count() == 0
assert os.path.exists(
get_asset_path(point_cloud.path, PointCloudService.ORIGINAL_FILES_DIR))
assert not os.path.exists(feature_asset_path)
def post(self, projectId: int, pointCloudId: int):
"""
:raises InvalidCoordinateReferenceSystem: in case file missing coordinate reference system
"""
f = request.files['file']
file_name = secure_filename(f.filename)
logger.info("Add a file to a point cloud:{} in project:{} for user:{}: {}".format(
pointCloudId, projectId, request.current_user.username, file_name))
pc_task = PointCloudService.fromFileObj(pointCloudId, f, file_name)
return pc_task
def test_add_point_cloud_file(projects_fixture, point_cloud_fixture,
lidar_las1pt2_file_fixture,
convert_to_potree_mock, check_point_cloud_mock,
get_point_cloud_info_mock):
filename = os.path.basename(lidar_las1pt2_file_fixture.name)
task = PointCloudService.fromFileObj(point_cloud_fixture.id,
lidar_las1pt2_file_fixture, filename)
assert task.status == "RUNNING"
assert point_cloud_fixture.task_id == task.id
# load updated point cloud
point_cloud = db_session.query(PointCloud).get(1)
las_files = os.listdir(
get_asset_path(point_cloud.path, PointCloudService.ORIGINAL_FILES_DIR))
assert len(las_files) == 1
assert las_files[0] == os.path.basename(filename)
original_file_size = os.fstat(lidar_las1pt2_file_fixture.fileno()).st_size
asset_file_path = os.path.join(
get_asset_path(point_cloud.path, PointCloudService.ORIGINAL_FILES_DIR),
filename)
assert os.path.getsize(asset_file_path) == original_file_size
# run conversion tool (that we had mocked)
_, convert_kwargs = convert_to_potree_mock.apply_async.call_args
assert projects_fixture.id == convert_kwargs['args'][0]
convert_to_potree(projects_fixture.id)
# load updated point cloud
point_cloud = db_session.query(PointCloud).get(1)
assert point_cloud.task.status == "FINISHED"
assert db_session.query(Feature).count() == 1
assert db_session.query(FeatureAsset).count() == 1
assert len(os.listdir(get_project_asset_dir(point_cloud.project_id))) == 2
assert len(os.listdir(get_asset_path(point_cloud.feature.assets[0].path))
) == 5 # index.html, preview.html, pointclouds, libs, logo
assert os.path.isfile(
os.path.join(get_asset_path(point_cloud.feature.assets[0].path),
"preview.html"))
with open(
os.path.join(get_asset_path(point_cloud.feature.assets[0].path),
"preview.html"), 'r+') as f:
preview = f.read()
assert "nsf_logo" not in preview
assert "$('.potree_menu_toggle').hide()" in preview