from math import e
import unittest

from src.interfaces.agents.agent import Agent
from src.modules.pdf.pdfAgent import PDFAgent
from src.modules.pdf.tools.metadataPDFTool import PDFMetadataTool 

class TestPDFAgent(unittest.TestCase):
    def setUp(self):
        self.agent: Agent = PDFAgent()
        return super().setUp()
    
    
    def test_createAgent(self):
        self.assertIsInstance(self.agent, Agent)
        self.assertEqual(self.agent.name, "PDF Agent")
        self.assertEqual(self.agent.description, "Agent to handle PDF operations")
        
        self.assertIsInstance(self.agent.tools, list)
        
    def test_list_tools(self):
        self.assertEqual(self.agent.list_tools(), [])
        
    def test_create_tools(self):
        metadataTool = PDFMetadataTool()
        self.assertIsInstance(metadataTool, PDFMetadataTool)
        self.assertEqual(metadataTool.name, "PDF Metadata Tool")
        
        self.agent.add_tool(metadataTool)
        tools = self.agent.get_tools()
        
        self.assertEqual(tools, [metadataTool])
        self.assertEqual(len(self.agent.tools), 1)
        self.assertEqual(self.agent.tools[0].name, "PDF Metadata Tool")
        
    # def test_