From 18aa1d9d42c21e41c4753b47f94489abb993eee4 Mon Sep 17 00:00:00 2001
From: Christopher Sherman <sherman27@llnl.gov>
Date: Wed, 10 Apr 2024 12:34:09 -0700
Subject: [PATCH] Allowing integrated tests to run without baselines

---
 geos_ats_package/geos_ats/baseline_io.py | 11 +++++------
 geos_ats_package/geos_ats/main.py        |  7 ++++++-
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/geos_ats_package/geos_ats/baseline_io.py b/geos_ats_package/geos_ats/baseline_io.py
index 73fbf46..99d07b9 100644
--- a/geos_ats_package/geos_ats/baseline_io.py
+++ b/geos_ats_package/geos_ats/baseline_io.py
@@ -130,7 +130,7 @@ def collect_baselines( bucket_name: str,
                 file_download_progress( {}, f"{bucket_name}/{blob_tar}", archive_name )
             except Exception as e:
                 logger.error( f'Failed to download baseline from URL ({bucket_name}/{blob_tar})' )
-                logger.error( str( e ) )
+                logger.error( repr( e ) )
         else:
             # Download from GCP
             try:
@@ -140,7 +140,7 @@ def collect_baselines( bucket_name: str,
                 blob.download_to_filename( archive_name )
             except Exception as e:
                 logger.error( f'Failed to download baseline from GCP ({bucket_name}/{blob_tar})' )
-                logger.error( str( e ) )
+                logger.error( repr( e ) )
 
     if os.path.isfile( archive_name ):
         # Unpack new baselines
@@ -149,11 +149,10 @@ def collect_baselines( bucket_name: str,
             shutil.unpack_archive( archive_name, baseline_path, format='gztar' )
             logger.info( 'Finished fetching baselines!' )
         except Exception as e:
-            logger.error( str( e ) )
+            logger.error( repr( e ) )
             raise Exception( f'Failed to unpack baselines: {archive_name}' )
 
     else:
-        logger.error( str( e ) )
         raise Exception( f'Could not find baseline files to unpack: expected={archive_name}' )
 
 
@@ -195,7 +194,7 @@ def pack_baselines( archive_name: str, baseline_path: str, log_path: str = '' ):
         logger.info( f'Created {archive_name}.tar.gz' )
     except Exception as e:
         logger.error( 'Failed to create baseline archive' )
-        logger.error( str( e ) )
+        logger.error( repr( e ) )
 
 
 def upload_baselines( bucket_name: str, archive_name: str ):
@@ -221,7 +220,7 @@ def upload_baselines( bucket_name: str, archive_name: str ):
 
     except Exception as e:
         logger.error( 'Failed to upload baselines!' )
-        logger.error( str( e ) )
+        logger.error( repr( e ) )
 
 
 def manage_baselines( options ):
diff --git a/geos_ats_package/geos_ats/main.py b/geos_ats_package/geos_ats/main.py
index 5de47bc..c859853 100644
--- a/geos_ats_package/geos_ats/main.py
+++ b/geos_ats_package/geos_ats/main.py
@@ -292,7 +292,12 @@ def main():
     os.chdir( ats_root_dir )
     os.makedirs( options.workingDir, exist_ok=True )
     create_log_directory( options )
-    baseline_io.manage_baselines( options )
+    try:
+        baseline_io.manage_baselines( options )
+    except Exception as e:
+        logger.error( 'Failed to download/locate baselines... Continuing to run tests without them' )
+        logger.error( repr( e ) )
+        os.makedirs( options.baselineDir, exist_ok=True )
 
     # Check the test configuration
     from geos_ats import configuration_record