fix: improve templates

snakemake · Nov 23, 2023 · 920825f · 920825f
1 parent 75f37b5
commit 920825f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/poetry_snakemake_plugin/templates/storage-plugins/tests.py b/poetry_snakemake_plugin/templates/storage-plugins/tests.py
@@ -4,16 +4,18 @@
 from snakemake_interface_storage_plugins.settings import StorageProviderSettingsBase
 
 
-class TestStorageNoSettings(TestStorageBase):
+class TestStorage(TestStorageBase):
     __test__ = True
     retrieve_only = False  # set to True if the storage is read-only
+    store_only = False  # set to True if the storage is write-only
+    delete = True  # set to False if the storage does not support deletion
 
-    def get_query(self) -> str:
+    def get_query(self, tmp_path) -> str:
         # Return a query. If retrieve_only is True, this should be a query that
         # is present in the storage, as it will not be created.
         ...
 
-    def get_query_not_existing(self) -> str:
+    def get_query_not_existing(self, tmp_path) -> str:
         ...
 
     def get_storage_provider_cls(self) -> Type[StorageProviderBase]: