#!/usr/bin/env python3

import unittest

from jinja2 import Template


class TestPackagePreinstall(unittest.TestCase):
    def setUp(self):
        """Set up test fixtures."""
        # Create a simplified test template with just the packages section
        self.packages_template = Template("""
packages:
  - sudo
{% if performance_preinstall_packages | default(false) %}
  # Universal tools always needed by Algo (performance optimization)
  - git
  - screen
  - apparmor-utils
  - uuid-runtime
  - coreutils
  - iptables-persistent
  - cgroup-tools
{% endif %}
""")

    def test_preinstall_disabled_by_default(self):
        """Test that package pre-installation is disabled by default."""
        # Test with default config (performance_preinstall_packages not set)
        rendered = self.packages_template.render({})

        # Should only have sudo package
        self.assertIn('- sudo', rendered)
        self.assertNotIn('- git', rendered)
        self.assertNotIn('- screen', rendered)
        self.assertNotIn('- apparmor-utils', rendered)

    def test_preinstall_enabled(self):
        """Test that package pre-installation works when enabled."""
        # Test with pre-installation enabled
        rendered = self.packages_template.render({'performance_preinstall_packages': True})

        # Should have sudo and all universal packages
        self.assertIn('- sudo', rendered)
        self.assertIn('- git', rendered)
        self.assertIn('- screen', rendered)
        self.assertIn('- apparmor-utils', rendered)
        self.assertIn('- uuid-runtime', rendered)
        self.assertIn('- coreutils', rendered)
        self.assertIn('- iptables-persistent', rendered)
        self.assertIn('- cgroup-tools', rendered)

    def test_preinstall_disabled_explicitly(self):
        """Test that package pre-installation is disabled when set to false."""
        # Test with pre-installation explicitly disabled
        rendered = self.packages_template.render({'performance_preinstall_packages': False})

        # Should only have sudo package
        self.assertIn('- sudo', rendered)
        self.assertNotIn('- git', rendered)
        self.assertNotIn('- screen', rendered)
        self.assertNotIn('- apparmor-utils', rendered)

    def test_package_count(self):
        """Test that the correct number of packages are included."""
        # Default: should only have sudo (1 package)
        rendered_default = self.packages_template.render({})
        lines_default = [line.strip() for line in rendered_default.split('\n') if line.strip().startswith('- ')]
        self.assertEqual(len(lines_default), 1)

        # Enabled: should have sudo + 7 universal packages (8 total)
        rendered_enabled = self.packages_template.render({'performance_preinstall_packages': True})
        lines_enabled = [line.strip() for line in rendered_enabled.split('\n') if line.strip().startswith('- ')]
        self.assertEqual(len(lines_enabled), 8)


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