test

Author: zpoint

examples/managed_job_with_storage.yaml

@@ -25,10 +25,11 @@ file_mounts:
     # Change this to your own globally unique bucket name.
     name: sky-output-bucket
     mode: MOUNT
+    store: azure
 
   /public-bucket:
     source: s3://fah-public-data-covid19-cryptic-pockets
-  
+
   # File mounts for folder
   /tmp/workdir: ~/tmp-workdir
 
@@ -50,7 +51,7 @@ run: |
   ls ~/sky_workdir/managed_job_with_storage.yaml
   ls ~/bucket_workdir/managed_job_with_storage.yaml
   ls -l /public-bucket
-  
+
 
   mkdir -p /data/logs
   touch /data/logs/test.log
@@ -61,6 +62,6 @@ run: |
 
   cat ~/tmpfile
   cat ~/a/b/c/tmpfile
-  
+
   # Write to a file in the mounted bucket
   echo "hello world!" > /output_path/output.txt

tests/smoke_tests/test_managed_job.py

@@ -947,7 +947,9 @@ def test_managed_jobs_storage(generic_cloud: str):
              f'{smoke_tests_utils.down_cluster_for_cloud_cmd(name)} || true'),
             # Increase timeout since sky jobs queue -r can be blocked by other spot tests.
             timeout=20 * 60,
-        )
+            env={
+                'SKYPILOT_CONFIG': 'tests/test_yamls/low_resource_sky_config.yaml'
+            })
         smoke_tests_utils.run_one_test(test)
 
 

tests/test_yamls/low_resource_sky_config.yaml

@@ -0,0 +1,10 @@
+jobs:
+  controller:
+    resources:
+      cpus: 2+
+      memory: 4+
+serve:
+  controller:
+    resources:
+      cpus: 2+
+      memory: 4+

tests/test_yamls/test_storage_mounting.yaml.j2

@@ -25,19 +25,22 @@ file_mounts:
     name: {{storage_name}}
     source: ~/tmp-workdir
     mode: {% if only_mount | default(false) %}MOUNT{% else %}COPY{% endif %}
+    store: azure
 
   # Mounting private buckets in COPY mode with a list of files as source
   /mount_private_copy_lof:
     name: {{storage_name}}
     source: ['~/tmp-workdir/tmp file', '~/tmp-workdir/tmp file2']
     mode: {% if only_mount | default(false) %}MOUNT{% else %}COPY{% endif %}
+    store: azure
 
   {% if include_private_mount | default(True) %}
   # Mounting private buckets in MOUNT mode
   /mount_private_mount:
     name: {{storage_name}}
     source: ~/tmp-workdir
     mode: MOUNT
+    store: azure
   {% endif %}
 
 run: |