diff --git a/tests/test_helpers.py b/tests/test_helpers.py
new file mode 100644
index 0000000..eee7492
--- /dev/null
+++ b/tests/test_helpers.py
@@ -0,0 +1,20 @@
+"""Unit tests module for the helper functions"""
+import unittest
+from tpv.commands.test import mock_galaxy
+from tpv.core.helpers import get_dataset_attributes
+
+
+class TestHelpers(unittest.TestCase):
+    """Tests for helper functions"""
+    def test_get_dataset_attributes(self):
+        """Test that the function returns a dictionary with the correct attributes"""
+        job = mock_galaxy.Job()
+        job.add_input_dataset(
+            mock_galaxy.DatasetAssociation(
+                "test",
+                mock_galaxy.Dataset("test.txt", file_size=7*1024**3, object_store_id="files1")
+                )
+            )
+        dataset_attributes = get_dataset_attributes(job.input_datasets)
+        expected_result = {0: {'object_store_id': 'files1', 'size': 7*1024**3}}
+        self.assertEqual(dataset_attributes, expected_result)
diff --git a/tpv/commands/test/mock_galaxy.py b/tpv/commands/test/mock_galaxy.py
index 08deb69..be3a82f 100644
--- a/tpv/commands/test/mock_galaxy.py
+++ b/tpv/commands/test/mock_galaxy.py
@@ -37,11 +37,12 @@ def __init__(self, name, dataset):
 class Dataset:
     counter = 0
 
-    def __init__(self, file_name, file_size):
+    def __init__(self, file_name, file_size, object_store_id=None):
         self.id = self.counter
         self.counter += 1
         self.file_name = file_name
         self.file_size = file_size
+        self.object_store_id = object_store_id
 
     def get_size(self, calculate_size=False):
         return self.file_size