🐛 replace published path in traits after copy

and add some more tests

tests/client/ayon_core/plugins/publish/test_integrate_traits.py

@@ -237,6 +237,8 @@ def mock_context(
     return context
 
 
+
+
 def test_get_template_name(mock_context: pyblish.api.Context) -> None:
     """Test get_template_name.
 
@@ -252,6 +254,75 @@ def test_get_template_name(mock_context: pyblish.api.Context) -> None:
     assert template_name == "default"
 
 
+class TestGetSize:
+    @staticmethod
+    def get_size(file_path: Path) -> int:
+        """Get size of the file.
+
+        Args:
+            file_path (Path): File path.
+
+        Returns:
+            int: Size of the file.
+
+        """
+        return file_path.stat().st_size
+
+    @pytest.mark.parametrize(
+        "file_path, expected_size",
+        [
+            (Path("./test_file_1.txt"), 10),  # id: happy_path_small_file
+            (Path("./test_file_2.txt"), 1024),  # id: happy_path_medium_file
+            (Path("./test_file_3.txt"), 10485760)  # id: happy_path_large_file
+        ],
+        ids=["happy_path_small_file", "happy_path_medium_file", "happy_path_large_file"]
+    )
+    def test_get_size_happy_path(self, file_path: Path, expected_size: int, tmp_path: Path):
+        # Arrange
+        file_path = tmp_path / file_path
+        file_path.write_bytes(b"\0" * expected_size)
+
+        # Act
+        size = self.get_size(file_path)
+
+        # Assert
+        assert size == expected_size
+
+
+    @pytest.mark.parametrize(
+        "file_path, expected_size",
+        [
+            (Path("./test_file_empty.txt"), 0)  # id: edge_case_empty_file
+        ],
+        ids=["edge_case_empty_file"]
+    )
+    def test_get_size_edge_cases(self, file_path: Path, expected_size: int, tmp_path: Path):
+        # Arrange
+        file_path = tmp_path / file_path
+        file_path.touch() # Create an empty file
+
+        # Act
+        size = self.get_size(file_path)
+
+        # Assert
+        assert size == expected_size
+
+    @pytest.mark.parametrize(
+        "file_path, expected_exception",
+        [
+            (Path("./non_existent_file.txt"), FileNotFoundError),  # id: error_file_not_found
+            (123, TypeError) # id: error_invalid_input_type
+        ],
+        ids=["error_file_not_found", "error_invalid_input_type"]
+    )
+    def test_get_size_error_cases(self, file_path, expected_exception, tmp_path):
+
+        # Act & Assert
+        with pytest.raises(expected_exception):
+            file_path = tmp_path / file_path
+            self.get_size(file_path)
+
+
 def test_filter_lifecycle() -> None:
     """Test filter_lifecycle."""
     integrator = IntegrateTraits()