Add temporary deployment test

scripts/rubintv_worker.py

@@ -20,6 +20,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 import argparse
+import logging
 import os
 import pathlib
 from lsst.rubintv.analysis.service.data import DataCenter
@@ -34,12 +35,63 @@
 from lsst.rubintv.analysis.service.worker import Worker
 
 default_config = os.path.join(pathlib.Path(__file__).parent.absolute(), "config.yaml")
-
+log = logging.getLogger('lsst.rubintv.analysis.server.worker')
 
 class UniversalToVisit(DataMatch):
     def get_join(self):
         return
 
+def connection_test():
+    REPO_LOCATION = os.getenv(
+        'DAF_BUTLER_REPOSITORY_INDEX',
+        '/sdf/group/rubin/repo/ir2/butler.yaml'
+    )
+    REPO_COLLECTIONS = os.getenv(
+        'DAF_BUTLER_COLLECTIONS',
+        'LSSTCam/raw/all,LSSTCam/calib'
+    ).split('/')
+
+    try:
+        import lsst.afw.image as afwImage
+        import numpy as np
+        data = np.zeros((10, 10), dtype=np.float32)
+        img = afwImage.ImageF(data)
+        msg = 'Create an afwImage - stack import successful'
+        print(msg)
+        log.info(msg)
+    except Exception as e:
+        log.exception(f'Failed to create an afwImage: {e}')
+
+    try:
+        from lsst.daf.butler import Butler
+        butler = Butler(REPO_LOCATION,
+                        collections=REPO_COLLECTIONS)
+        msg = 'Created a full camera butler - repo mount sucessful'
+        print(msg)
+        log.info(msg)
+    except Exception as e:
+        log.exception(f'Failed to create the butler: {e}')
+
+    try:
+        import lsst.summit.utils.butlerUtils as butlerUtils
+        butler = butlerUtils.makeDefaultLatissButler(embargo=True)
+        msg = 'Created an embargo LATISS butler - embargo credentials working'
+        print(msg)
+        log.info(msg)
+    except Exception as e:
+        log.exception(f'Failed to create LATISS embargo butler: {e}')
+
+    try:
+        from lsst.daf.butler import DimensionRecord
+        where = "exposure.day_obs>=20230101 AND exposure.seq_num=1"
+        records = butler.registry.queryDimensionRecords('exposure', where=where, datasets='raw')
+        record = list(records)[0]
+        assert isinstance(record, DimensionRecord)
+        msg = 'Successfully accessed an exposure record - butler is really working'
+        print(msg)
+        log.info(msg)
+    except Exception as e:
+        log.exception(f'Failed to get a dimension record from the butler: {e}')
 
 def main():
     parser = argparse.ArgumentParser(description="Initialize a new RubinTV worker.")
@@ -52,8 +104,16 @@ def main():
     parser.add_argument(
         "-c", "--config", default=default_config, type=str, help="Location of the configuration file."
     )
+    parser.add_argument(
+        "-t", "--test", default=True, help="Run connection tests."
+    )
     args = parser.parse_args()
 
+    # Run the connection test.
+    # This code can be deleted after the initial deployment
+    if args.test:
+        connection_test()
+
     # Load the configuration file
     with open(args.config, "r") as file:
         config = yaml.safe_load(file)