Capture ImportTestRun specifically

aiidateam · May 27, 2024 · 7f377fc · 7f377fc
1 parent 6291acc
commit 7f377fc
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/aiida/cmdline/commands/cmd_archive.py b/src/aiida/cmdline/commands/cmd_archive.py
@@ -480,6 +480,7 @@ def _import_archive_and_migrate(
 
     from aiida.common.folders import SandboxFolder
     from aiida.tools.archive.abstract import get_format
+    from aiida.tools.archive.exceptions import ImportTestRun
     from aiida.tools.archive.imports import import_archive as _import_archive
 
     archive_format = get_format()
@@ -521,6 +522,12 @@ def _import_archive_and_migrate(
                     )
             else:
                 _echo_exception(f'an exception occurred while trying to import the archive {archive}', exception)
+        except ImportTestRun:
+            echo.echo_success(
+                f'Import dry-run of archive {archive} terminated successfully. Profile storage unmodified.'
+            )
+            return
+
         except Exception as exception:
             _echo_exception(f'an exception occurred while trying to import the archive {archive}', exception)