From 04c44fdd1b9875374c353b36a27dbf976305f9ff Mon Sep 17 00:00:00 2001
From: Sankalp Sanand <sankalp@agnostiq.ai>
Date: Fri, 24 Nov 2023 07:22:05 -0500
Subject: [PATCH] fixed tests

---
 .../covalent_dispatcher_tests/_dal/asset_test.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/tests/covalent_dispatcher_tests/_dal/asset_test.py b/tests/covalent_dispatcher_tests/_dal/asset_test.py
index 678bd9463..2f3975525 100644
--- a/tests/covalent_dispatcher_tests/_dal/asset_test.py
+++ b/tests/covalent_dispatcher_tests/_dal/asset_test.py
@@ -53,7 +53,7 @@ def test_asset_load_data():
         temppath = temp.name
         key = os.path.basename(temppath)
 
-    storage_path = "/tmp"
+    storage_path = temppath[: -len(key)]
 
     rec = get_asset_record(storage_path, key)
     a = Asset(None, rec)
@@ -65,7 +65,9 @@ def test_asset_store_data():
     with tempfile.NamedTemporaryFile("w", delete=True, suffix=".txt") as temp:
         temppath = temp.name
         key = os.path.basename(temppath)
-    storage_path = "/tmp"
+
+    storage_path = temppath[: -len(key)]
+
     rec = get_asset_record(storage_path, key)
     a = Asset(None, rec)
     a.store_data("Hello\n")
@@ -80,7 +82,8 @@ def test_upload_asset():
     with tempfile.NamedTemporaryFile("w", delete=True, suffix=".txt") as temp:
         src_path = temp.name
         src_key = os.path.basename(src_path)
-    storage_path = "/tmp"
+
+    storage_path = src_path[: -len(src_key)]
 
     rec = get_asset_record(storage_path, src_key)
     a = Asset(None, rec)
@@ -101,10 +104,12 @@ def test_download_asset():
     with tempfile.NamedTemporaryFile("w", delete=True, suffix=".txt") as temp:
         src_path = temp.name
         src_key = os.path.basename(src_path)
+
     with open(src_path, "w") as f:
         f.write("Hello\n")
 
-    storage_path = "/tmp"
+    storage_path = src_path[: -len(src_key)]
+
     with tempfile.NamedTemporaryFile("w", delete=True, suffix=".txt") as temp:
         dest_path = temp.name
         dest_key = os.path.basename(dest_path)
@@ -123,10 +128,11 @@ def test_copy_asset():
     with tempfile.NamedTemporaryFile("w", delete=True, suffix=".txt") as temp:
         src_path = temp.name
         src_key = os.path.basename(src_path)
+
     with open(src_path, "w") as f:
         f.write("Hello\n")
 
-    storage_path = "/tmp"
+    storage_path = src_path[: -len(src_key)]
     rec = get_asset_record(storage_path, src_key)
     src_asset = Asset(None, rec)