Class TestEnvironmentVariableHandling

tests/unit/test_all_embedders.py:324–368

Test embedder selection via environment variables.

class TestEnvironmentVariableHandling:
    """Test embedder selection via environment variables."""
    
    def test_embedder_type_env_var(self, embedder_type=None):
        """Test embedder selection via DEEPWIKI_EMBEDDER_TYPE environment variable."""
        import importlib
        import api.config
        
        if embedder_type:
            # Test specific embedder type
            self._test_single_embedder_type(embedder_type)
        else:
            # Test all embedder types
            for et in ['openai', 'google', 'ollama', 'bedrock']:
                self._test_single_embedder_type(et)
    
    def _test_single_embedder_type(self, embedder_type):
        """Test a single embedder type."""
        import importlib
        import api.config
        
        # Save original value
        original_value = os.environ.get('DEEPWIKI_EMBEDDER_TYPE')
        
        try:
            # Set environment variable
            os.environ['DEEPWIKI_EMBEDDER_TYPE'] = embedder_type
            
            # Reload config to pick up new env var
            importlib.reload(api.config)
            
            from api.config import EMBEDDER_TYPE, get_embedder_type
            
            assert EMBEDDER_TYPE == embedder_type, f"EMBEDDER_TYPE should be {embedder_type}"
            assert get_embedder_type() == embedder_type, f"get_embedder_type() should return {embedder_type}"
            
        finally:
            # Restore original value
            if original_value is not None:
                os.environ['DEEPWIKI_EMBEDDER_TYPE'] = original_value
            elif 'DEEPWIKI_EMBEDDER_TYPE' in os.environ:
                del os.environ['DEEPWIKI_EMBEDDER_TYPE']
            
            # Reload config to restore original state
            importlib.reload(api.config)


class TestIssuesIdentified: