wehub-resource-sync
87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
# Copyright 2024 Google, LLC. This software is provided as-is, without
|
|
# warranty or representation for any use or purpose. Your use of it is
|
|
# subject to your agreement with Google.
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from agent import SwotAgentDeps, SwotAnalysis
|
|
from pydantic_ai import models
|
|
import pytest
|
|
|
|
# Prevent accidental model requests during testing
|
|
models.ALLOW_MODEL_REQUESTS = False
|
|
|
|
|
|
class TestSWOTAnalysis:
|
|
"""Test the SWOT Analysis data model"""
|
|
|
|
def test_swot_analysis_model(self) -> None:
|
|
"""Test that the SWOT Analysis model can be instantiated with valid data"""
|
|
analysis = SwotAnalysis(
|
|
strengths=["Strong brand"],
|
|
weaknesses=["High costs"],
|
|
opportunities=["New markets"],
|
|
threats=["Competition"],
|
|
analysis="Test analysis",
|
|
)
|
|
assert isinstance(analysis, SwotAnalysis)
|
|
assert len(analysis.strengths) == 1
|
|
assert len(analysis.weaknesses) == 1
|
|
assert len(analysis.opportunities) == 1
|
|
assert len(analysis.threats) == 1
|
|
assert analysis.analysis == "Test analysis"
|
|
|
|
|
|
class TestSwotAgentDeps:
|
|
"""Test the SWOT Agent Dependencies"""
|
|
|
|
@pytest.fixture
|
|
def agent_deps(self) -> SwotAgentDeps:
|
|
"""Create a basic SwotAgentDeps instance for testing"""
|
|
return SwotAgentDeps()
|
|
|
|
def test_agent_initialization(self, agent_deps: SwotAgentDeps) -> None:
|
|
"""Test that the agent dependencies are properly initialized"""
|
|
assert agent_deps.tool_history == []
|
|
assert agent_deps.request is None
|
|
assert agent_deps.update_status_func is None
|
|
|
|
def test_gemini_client_initialization(self, agent_deps: SwotAgentDeps) -> None:
|
|
"""Test Gemini client initialization"""
|
|
# Just verify that the Gemini client exists (could be None or initialized)
|
|
assert hasattr(agent_deps, "client")
|
|
|
|
def test_reddit_client_initialization(self, agent_deps: SwotAgentDeps) -> None:
|
|
"""Test Reddit client initialization"""
|
|
# Just verify that the reddit client exists (could be None or initialized)
|
|
assert hasattr(agent_deps, "reddit")
|
|
|
|
def test_mock_analysis(self, agent_deps: SwotAgentDeps) -> None:
|
|
"""Test that the required attributes exist for analysis"""
|
|
# Test SwotAnalysis model structure using model_fields
|
|
required_fields = {
|
|
"strengths",
|
|
"weaknesses",
|
|
"opportunities",
|
|
"threats",
|
|
"analysis",
|
|
}
|
|
model_fields = set(SwotAnalysis.model_fields.keys())
|
|
assert required_fields.issubset(
|
|
model_fields
|
|
), f"Missing fields: {required_fields - model_fields}"
|
|
|
|
# Test agent dependencies
|
|
assert hasattr(agent_deps, "client")
|
|
assert hasattr(agent_deps, "tool_history")
|