Add test cases

laughingman7743 · Dec 17, 2022 · e47f0f6 · e47f0f6
1 parent 3b71e8e
commit e47f0f6
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/tests/filesystem/test_s3_object.py b/tests/filesystem/test_s3_object.py
@@ -35,3 +35,40 @@ def test_init(self):
         assert actual.type == S3ObjectType.S3_OBJECT_TYPE_FILE
         assert actual.storage_class == S3StorageClass.S3_STORAGE_CLASS_STANDARD
         assert actual.etag == "etag"
+
+    def test_to_dict(self):
+        actual = S3Object(
+            bucket="test-bucket",
+            key=None,
+            size=0,
+            type=S3ObjectType.S3_OBJECT_TYPE_DIRECTORY,
+            storage_class=S3StorageClass.S3_STORAGE_CLASS_DIRECTORY,
+            etag=None,
+        )
+        assert actual.to_dict() == {
+            "bucket": "test-bucket",
+            "key": None,
+            "name": "test-bucket",
+            "size": 0,
+            "type": S3ObjectType.S3_OBJECT_TYPE_DIRECTORY,
+            "storage_class": S3StorageClass.S3_STORAGE_CLASS_DIRECTORY,
+            "etag": None,
+        }
+
+        actual = S3Object(
+            bucket="test-bucket",
+            key="path/to/object",
+            size=100,
+            type=S3ObjectType.S3_OBJECT_TYPE_FILE,
+            storage_class=S3StorageClass.S3_STORAGE_CLASS_STANDARD,
+            etag="etag",
+        )
+        assert actual.to_dict() == {
+            "bucket": "test-bucket",
+            "key": "path/to/object",
+            "name": "test-bucket/path/to/object",
+            "size": 100,
+            "type": S3ObjectType.S3_OBJECT_TYPE_FILE,
+            "storage_class": S3StorageClass.S3_STORAGE_CLASS_STANDARD,
+            "etag": "etag",
+        }