import unittest
import os
import yaml

class TestNoStopRequiredIntegrity(unittest.TestCase):
    def setUp(self):
        # Path to the roles directory
        self.roles_dir = os.path.abspath(
            os.path.join(os.path.dirname(__file__), '../../../roles')
        )

    def test_backup_no_stop_required_consistency(self):
        """
        Ensure that if `backup.no_stop_required: true` is set for any docker.services[*]:
          - it's a boolean value
          - the containing service dict has an `image` entry at the same level
        """
        for role in os.listdir(self.roles_dir):
            docker_config_path = os.path.join(
                self.roles_dir, role, 'config', 'main.yml'
            )
            if not os.path.isfile(docker_config_path):
                continue

            with open(docker_config_path, 'r') as f:
                try:
                    # Ensure config is at least an empty dict if YAML is empty or null
                    config = yaml.safe_load(f) or {}
                except yaml.YAMLError as e:
                    self.fail(f"YAML parsing failed for {docker_config_path}: {e}")
                    continue

            # Safely get services dict
            services = (config.get('docker', {}) or {}).get('services', {}) or {}

            for service_key, service in services.items():
                if not isinstance(service, dict):
                    continue
                backup_cfg = service.get('backup', {}) or {}
                # Check if no_stop_required is explicitly True
                if backup_cfg.get('no_stop_required') is True:
                    with self.subTest(role=role, service=service_key):
                        # Must be a boolean
                        self.assertIsInstance(
                            backup_cfg['no_stop_required'], bool,
                            f"`backup.no_stop_required` in role '{role}', service '{service_key}' must be a boolean."
                        )
                        # Must have `image` defined at the service level
                        self.assertIn(
                            'image', service,
                            f"`image` is required in role '{role}', service '{service_key}' when `backup.no_stop_required` is set to True."
                        )

if __name__ == '__main__':
    unittest.main()