From 8ceed1b9cf0e83de462f7ebc3c972aee0fa6698a Mon Sep 17 00:00:00 2001
From: Jayjeet Chakraborty <jc.github@rediffmail.com>
Date: Wed, 9 Mar 2022 10:10:37 -0800
Subject: [PATCH] Update skyhook tests

---
 tests/test_skyhook_job.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/tests/test_skyhook_job.py b/tests/test_skyhook_job.py
index 133d2e6d27..6c4aa0c931 100644
--- a/tests/test_skyhook_job.py
+++ b/tests/test_skyhook_job.py
@@ -1,4 +1,5 @@
 import os
+import toml
 import uproot
 import awkward as ak
 from coffea import processor
@@ -7,6 +8,15 @@
 
 
 if __name__ == "__main__":
+    config_dict = {
+        "skyhook": {
+            "ceph_config_path": "/tmp/testskyhookjob/ceph.conf",
+            "ceph_data_pool": "cephfs_data"
+        }
+    }
+    with open('/root/.coffea.toml', 'w') as f:
+        toml.dump(config_dict, f)
+
     ak.to_parquet(
         uproot.lazy("tests/samples/nano_dy.root:Events"),
         "nano_dy.parquet",
@@ -45,7 +55,6 @@
     run = processor.Runner(
         executor=executor,
         use_skyhook=True,
-        skyhook_options={"ceph_config_path": "/tmp/testskyhookjob/ceph.conf"},
         format="parquet",
         schema=schemas.NanoAODSchema,
     )