Remove dead code

Author: b-long

tests/test_validate_otdf_python.py

@@ -23,7 +23,7 @@
 from tests.config_pydantic import CONFIG_TDF
 
 
-def _get_configuration() -> SDK:
+def _get_sdk() -> SDK:
     return (
         SDKBuilder()
         .set_platform_endpoint(
@@ -44,44 +44,37 @@ def _get_configuration() -> SDK:
     )
 
 
-def encrypt_file(input_path: Path, sdk=None) -> Path:
-    """Encrypt a file and return the path to the encrypted file."""
-    if sdk is None:
-        kas_url = os.environ.get("OPENTDF_KAS_URL", "https://default.kas.example.com")
+def _get_sdk_and_tdf_config() -> tuple:
+    sdk = _get_sdk()
 
-        # Build the SDK
-        sdk = _get_configuration()
+    # Create KASInfo without public key - let the SDK fetch it
+    kas_info = KASInfo(
+        url=CONFIG_TDF.OPENTDF_PLATFORM_URL,
+        default=True,
+    )
 
-        # Create KASInfo without public key - let the SDK fetch it
-        kas_info = KASInfo(url=kas_url)
+    tdf_config = sdk.new_tdf_config(
+        attributes=["attr1", "attr2"],
+        kas_info_list=[kas_info],  # KAS info without explicit public key
+    )
+    return sdk, tdf_config
 
-        # Create config with the KASInfo
-        config = sdk.new_tdf_config(
-            attributes=["attr1", "attr2"], kas_info_list=[kas_info]
-        )
-    else:
-        # If SDK is provided, we need to create a config with attributes and KASInfo
-        # Get the platform URL from the SDK
-        kas_url = sdk.get_platform_url() or "https://default.kas.example.com"
 
-        # Create KASInfo without public key - let the SDK fetch it
-        kas_info = KASInfo(url=kas_url)
+def encrypt_file(input_path: Path) -> Path:
+    """Encrypt a file and return the path to the encrypted file."""
 
-        # Create config with attributes and KASInfo
-        config = sdk.new_tdf_config(
-            attributes=["attr1", "attr2"], kas_info_list=[kas_info]
-        )
+    # Build the SDK
+    sdk, tdf_config = _get_sdk_and_tdf_config()
 
     output_path = input_path.with_suffix(input_path.suffix + ".tdf")
     with open(input_path, "rb") as infile, open(output_path, "wb") as outfile:
-        sdk.create_tdf(infile.read(), config, output_stream=outfile)
+        sdk.create_tdf(infile.read(), tdf_config, output_stream=outfile)
     return output_path
 
 
-def decrypt_file(encrypted_path: Path, sdk=None) -> Path:
+def decrypt_file(encrypted_path: Path) -> Path:
     """Decrypt a file and return the path to the decrypted file."""
-    if sdk is None:
-        sdk = _get_configuration()
+    sdk = _get_sdk()
 
     output_path = encrypted_path.with_suffix(".decrypted")
     with open(encrypted_path, "rb") as infile, open(output_path, "wb") as outfile:
@@ -106,7 +99,7 @@ def decrypt_file(encrypted_path: Path, sdk=None) -> Path:
 def verify_encrypt_str() -> None:
     print("Validating string encryption (local TDF)")
     try:
-        sdk = _get_configuration()
+        sdk = _get_sdk()
 
         payload = b"Hello from Python"
 
@@ -116,7 +109,7 @@ def verify_encrypt_str() -> None:
             default=True,
         )
 
-        config = sdk.new_tdf_config(
+        tdf_config = sdk.new_tdf_config(
             attributes=["attr1", "attr2"],
             kas_info_list=[kas_info],  # KAS info without explicit public key
         )
@@ -125,7 +118,7 @@ def verify_encrypt_str() -> None:
         from io import BytesIO
 
         output = BytesIO()
-        sdk.create_tdf(payload, config, output_stream=output)
+        sdk.create_tdf(payload, tdf_config, output_stream=output)
         manifest_bytes = output.getvalue()
         print(f"Manifest returned: {manifest_bytes[:60]}... (truncated)")
         assert manifest_bytes and len(manifest_bytes) > 0
@@ -160,6 +153,11 @@ def verify_encrypt_file() -> None:
         ) from e
 
 
+def test_verify_encrypt_file():
+    """Run the file encryption verification test."""
+    verify_encrypt_file()
+
+
 def verify_encrypt_decrypt_file() -> None:
     print("Validating encrypt/decrypt roundtrip (local TDF)")
     try:
@@ -168,17 +166,12 @@ def verify_encrypt_decrypt_file() -> None:
             input_file = tmpDir / "plain.txt"
             input_file.write_text("Secret message")
 
-            # Build the SDK
-            sdk = _get_configuration()
-
-            # Get public key from KAS
             try:
-                # Use the encrypt_file function which now gets the public key from KAS
-                encrypted_path = encrypt_file(input_file, sdk)
+                encrypted_path = encrypt_file(input_file)
                 print(f"Encrypted file at: {encrypted_path}")
 
                 # Decrypt the file using the same SDK
-                decrypted_path = decrypt_file(encrypted_path, sdk)
+                decrypted_path = decrypt_file(encrypted_path)
                 print(f"Decrypted file at: {decrypted_path}")
 
                 # Verify the result