push

Author: hassiebp

tests/test_prompt.py

@@ -1505,135 +1505,3 @@ def test_environment_variable_override_prompt_cache_ttl():
     # Clean up environment variable
     if LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS in os.environ:
         del os.environ[LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS]
-
-
-@patch.object(PromptCacheItem, "get_epoch_seconds")
-def test_default_ttl_when_environment_variable_not_set(mock_time):
-    """Test that default 60-second TTL is used when LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS is not set."""
-
-    # Ensure environment variable is not set
-    if LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS in os.environ:
-        del os.environ[LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS]
-
-    # Set initial time to 0
-    mock_time.return_value = 0
-
-    # Create a new Langfuse instance to pick up the default TTL
-    langfuse = Langfuse(
-        public_key="test-public-key",
-        secret_key="test-secret-key",
-        host="https://mock-host.com",
-    )
-    langfuse.api = Mock()
-
-    prompt_name = "test_default_ttl"
-    prompt = Prompt_Text(
-        name=prompt_name,
-        version=1,
-        prompt="Test prompt with default TTL",
-        type="text",
-        labels=[],
-        config={},
-        tags=[],
-    )
-    prompt_client = TextPromptClient(prompt)
-
-    mock_server_call = langfuse.api.prompts.get
-    mock_server_call.return_value = prompt
-
-    # First call - should cache the prompt
-    result1 = langfuse.get_prompt(prompt_name)
-    assert mock_server_call.call_count == 1
-    assert result1 == prompt_client
-
-    # Check that prompt is cached
-    cached_item = langfuse._resources.prompt_cache.get(
-        langfuse._resources.prompt_cache.generate_cache_key(
-            prompt_name, version=None, label=None
-        )
-    )
-    assert cached_item is not None
-    assert cached_item.value == prompt_client
-
-    # Set time to just before default TTL expiration
-    mock_time.return_value = DEFAULT_PROMPT_CACHE_TTL_SECONDS - 1
-
-    # Second call - should still use cache
-    result2 = langfuse.get_prompt(prompt_name)
-    assert mock_server_call.call_count == 1  # No new server call
-    assert result2 == prompt_client
-
-    # Set time to just after default TTL expiration to trigger cache expiry
-    # Use the actual DEFAULT_PROMPT_CACHE_TTL_SECONDS value that was imported
-    mock_time.return_value = DEFAULT_PROMPT_CACHE_TTL_SECONDS + 1
-
-    # Third call - should still use cache (stale cache behavior)
-    result3 = langfuse.get_prompt(prompt_name)
-    assert result3 == prompt_client
-
-    # Wait for background refresh to complete
-    while True:
-        if langfuse._resources.prompt_cache._task_manager.active_tasks() == 0:
-            break
-        sleep(0.1)
-
-    # Should have made a new server call for refresh
-    assert mock_server_call.call_count == 2
-
-    # Set time to just after default TTL expiration
-    mock_time.return_value = DEFAULT_PROMPT_CACHE_TTL_SECONDS + 1
-
-    # Fourth call - should use refreshed cache
-    result4 = langfuse.get_prompt(prompt_name)
-    assert result4 == prompt_client
-
-
-def test_clear_prompt_cache(langfuse):
-    """Test clearing the entire prompt cache."""
-    prompt_name = create_uuid()
-
-    # Mock the API calls
-    mock_prompt = Prompt_Text(
-        name=prompt_name,
-        version=1,
-        prompt="test prompt",
-        type="text",
-        labels=["production"],
-        config={},
-        tags=[],
-    )
-
-    # Mock the create_prompt API call
-    langfuse.api.prompts.create.return_value = mock_prompt
-
-    # Mock the get_prompt API call
-    langfuse.api.prompts.get.return_value = mock_prompt
-
-    # Create a prompt and cache it
-    prompt_client = langfuse.create_prompt(
-        name=prompt_name,
-        prompt="test prompt",
-        labels=["production"],
-    )
-
-    # Get the prompt to cache it
-    cached_prompt = langfuse.get_prompt(prompt_name)
-    assert cached_prompt.name == prompt_name
-
-    # Verify that the prompt is in the cache
-    cache_key = f"{prompt_name}-label:production"
-    assert (
-        langfuse._resources.prompt_cache.get(cache_key) is not None
-    ), "Prompt should be in cache before clearing"
-
-    # Clear the entire prompt cache
-    langfuse.clear_prompt_cache()
-
-    # Verify cache is completely cleared
-    assert (
-        langfuse._resources.prompt_cache.get(cache_key) is None
-    ), "Prompt should be removed from cache after clearing"
-
-    # Verify data integrity
-    assert prompt_client.name == prompt_name
-    assert cached_prompt.name == prompt_name