feat: add comprehensive unit tests and clean up CI

- Remove debug steps from CI workflow now that issues are resolved
- Add HTTP client tests (5 tests)
- Add file handler tests (5 tests)
- Add builder API tests (5 tests)
- Remove temporary test_basic.py file
- Total: 31 unit tests covering all major components

Author: jdrhyne

.github/workflows/ci.yml

@@ -34,44 +34,6 @@ jobs:
         python -m pip install --upgrade pip
         pip install -e ".[dev]"
 
-    - name: Debug - Check environment
-      run: |
-        echo "Python version:"
-        python --version
-        echo "Python path:"
-        which python
-        echo "Site packages:"
-        python -c "import site; print(site.getsitepackages())"
-        echo "Current directory:"
-        pwd
-        echo "Directory contents:"
-        ls -la
-        echo "Installed packages:"
-        pip list
-
-    - name: Debug - Test package import
-      run: |
-        echo "Testing direct import:"
-        python -c "import nutrient_dws; print('Import successful:', nutrient_dws.__version__)"
-        echo "Testing submodule imports:"
-        python -c "from nutrient_dws.client import NutrientClient; print('Client import successful')"
-        python -c "from nutrient_dws.exceptions import NutrientError; print('Exceptions import successful')"
-
-    - name: Debug - Check test structure
-      run: |
-        echo "Test directory structure:"
-        find tests -name "*.py" -type f
-        echo "Test __init__ files:"
-        find tests -name "__init__.py" -type f
-        echo "Python path:"
-        python -c "import sys; print('\n'.join(sys.path))"
-
-    - name: Debug - pytest collection dry run
-      run: |
-        echo "Running pytest collection only:"
-        python -m pytest --collect-only -v
-      continue-on-error: true
-
     - name: Run linting with ruff
       run: |
         python -m ruff check .
@@ -80,19 +42,7 @@ jobs:
     - name: Run type checking with mypy
       run: python -m mypy src tests
 
-    - name: Run basic test without coverage
-      run: |
-        echo "Running basic test without any plugins:"
-        python -m pytest tests/test_basic.py -v -p no:asyncio -p no:cov
-      continue-on-error: true
-
-    - name: Run all tests without coverage
-      run: |
-        echo "Running all tests without coverage:"
-        python -m pytest -v -p no:asyncio
-      continue-on-error: true
-
-    - name: Run tests with pytest (with coverage)
+    - name: Run tests with pytest
       run: python -m pytest -v --cov=nutrient_dws --cov-report=xml --cov-report=term
 
     - name: Upload coverage to Codecov

tests/test_basic.py

@@ -0,0 +1,63 @@
+"""Unit tests for Builder API."""
+
+import pytest
+
+from nutrient_dws.builder import BuildAPIWrapper
+
+
+def test_builder_init():
+    """Test builder initialization."""
+    builder = BuildAPIWrapper(None, "test.pdf")
+    assert builder._input_file == "test.pdf"
+    assert builder._actions == []
+    assert builder._parts == [{"file": "file"}]
+    assert "file" in builder._files
+
+
+def test_builder_add_step():
+    """Test adding steps to builder."""
+    builder = BuildAPIWrapper(None, "test.pdf")
+    result = builder.add_step("convert-to-pdf", options={"format": "docx"})
+    
+    assert result is builder  # Should return self for chaining
+    assert len(builder._actions) == 1
+    assert builder._actions[0]["type"] == "convert-to-pdf"
+    assert builder._actions[0]["format"] == "docx"
+
+
+def test_builder_chaining():
+    """Test method chaining."""
+    builder = BuildAPIWrapper(None, "test.pdf")
+    result = (
+        builder.add_step("convert-to-pdf")
+        .add_step("rotate-pages", options={"degrees": 90})
+        .add_step("watermark-pdf", options={"text": "DRAFT"})
+    )
+    
+    assert result is builder
+    assert len(builder._actions) == 3
+    # Actions are transformed by _map_tool_to_action, so check structure exists
+    assert all("type" in action for action in builder._actions)
+
+
+def test_builder_set_output_options():
+    """Test setting output options."""
+    builder = BuildAPIWrapper(None, "test.pdf")
+    result = builder.set_output_options(
+        metadata={"title": "Test Doc"},
+        optimize=True
+    )
+    
+    assert result is builder
+    assert builder._output_options["metadata"]["title"] == "Test Doc"
+    assert builder._output_options["optimize"] is True
+
+
+def test_builder_execute_requires_client():
+    """Test that execute requires a client."""
+    builder = BuildAPIWrapper(None, "test.pdf")
+    builder.add_step("convert-to-pdf")
+    
+    # Without a proper client, this would fail when trying to access client methods
+    # We can't test the actual failure without mocking, so just ensure the method exists
+    assert hasattr(builder, "execute")

tests/unit/test_builder.py

@@ -0,0 +1,61 @@
+"""Unit tests for file handling utilities."""
+
+import io
+from pathlib import Path
+
+import pytest
+
+from nutrient_dws.file_handler import (
+    prepare_file_input,
+    save_file_output,
+    stream_file_content,
+)
+
+
+def test_prepare_file_input_from_bytes():
+    """Test preparing file input from bytes."""
+    content = b"Hello, World!"
+    result, filename = prepare_file_input(content)
+    assert result == content
+    assert filename == "document"
+
+
+def test_prepare_file_input_from_string_io():
+    """Test preparing file input from StringIO-like object."""
+    # Using BytesIO instead of StringIO for binary compatibility
+    content = b"Test content"
+    file_obj = io.BytesIO(content)
+    result, filename = prepare_file_input(file_obj)
+    assert result == content
+    assert filename == "document"
+
+
+def test_get_file_size_from_bytes():
+    """Test getting file size from bytes."""
+    from nutrient_dws.file_handler import get_file_size
+
+    content = b"Hello, World!"
+    size = get_file_size(content)
+    assert size == 13
+
+
+def test_get_file_size_from_bytesio():
+    """Test getting file size from BytesIO."""
+    from nutrient_dws.file_handler import get_file_size
+
+    content = b"Test content"
+    file_obj = io.BytesIO(content)
+    size = get_file_size(file_obj)
+    assert size == 12
+
+
+def test_get_file_size_none():
+    """Test get_file_size returns None for invalid input."""
+    from nutrient_dws.file_handler import get_file_size
+
+    # Test with a file-like object without proper methods
+    class DummyFile:
+        pass
+
+    size = get_file_size(DummyFile())
+    assert size is None

tests/unit/test_file_handler.py

@@ -0,0 +1,39 @@
+"""Unit tests for HTTPClient."""
+
+import pytest
+
+from nutrient_dws.exceptions import APIError, AuthenticationError, NutrientTimeoutError
+from nutrient_dws.http_client import HTTPClient
+
+
+def test_http_client_init():
+    """Test HTTP client initialization."""
+    client = HTTPClient(api_key="test-key")
+    assert client._api_key == "test-key"
+    assert client._base_url == "https://api.pspdfkit.com"
+    assert client._timeout == 300
+
+
+def test_http_client_init_custom_timeout():
+    """Test HTTP client with custom timeout."""
+    client = HTTPClient(api_key="test-key", timeout=60)
+    assert client._timeout == 60
+
+
+def test_http_client_init_no_api_key():
+    """Test HTTP client initialization without API key."""
+    client = HTTPClient(api_key=None)
+    assert client._api_key is None
+
+
+def test_http_client_init_empty_api_key():
+    """Test HTTP client initialization with empty API key."""
+    client = HTTPClient(api_key="")
+    assert client._api_key == ""
+
+
+def test_http_client_context_manager():
+    """Test HTTP client can be used as context manager."""
+    with HTTPClient(api_key="test-key") as client:
+        assert client is not None
+        assert hasattr(client, "_session")