from unittest import TestCase
from pygeai.health.clients import HealthClient

class TestHealthIntegration(TestCase): 
    __required_attrs = [
        'running', 'timestamp', 'version'
    ]  

    def setUp(self):
        self.healthClient = HealthClient()

    def __get_status(self):
        return self.healthClient.check_api_status()

    def test_get_status(self):
        status = self.__get_status()
        for attr in self.__required_attrs:
            self.assertIn(
                attr, status,
                f"Status response should have an '{attr}' attribute"
            ) 
 