ProxMenux/oci/tests/test_custom_multiapp_stack.py

"""Custom stacks keep each app's endpoint and actual shared storage identity."""
from pathlib import Path
import sys
import unittest
from unittest.mock import patch

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / 'src'))
from proxmenux_oci.casaos import convert_casaos_compose
from proxmenux_oci.stack import apply_stack_support, assess, build_stack, DefaultsUI, service_template
from proxmenux_oci.installer import DEFAULT_MODE, ADVANCED_MODE

COMPOSE = '''name: arr-test
services:
  sonarr:
    image: lscr.io/linuxserver/sonarr:latest
    ports: ['8989:8989']
    environment: {PUID: '1000', PGID: '1000'}
    volumes: [sonarr-config:/config, shared-data:/data]
  radarr:
    image: lscr.io/linuxserver/radarr:latest
    ports: ['7878:7878']
    environment: {PUID: '1000', PGID: '1000'}
    volumes: [radarr-config:/config, shared-data:/data]
volumes:
  sonarr-config:
  radarr-config:
  shared-data:
x-casaos:
  main: sonarr
  port_map: '8989'
  scheme: http
  index: /
  title: {en_US: Arr Test}
'''


def template(custom=True, text=COMPOSE):
    result = convert_casaos_compose(text, 'local', '', 'local/arr-test/docker-compose.yml', '')
    result['proxmox']['custom_compose_stack'] = custom
    apply_stack_support(result)
    return result


class UI(DefaultsUI):
    def __init__(self):
        self.asks = []
    def ask(self, text, default=None, **kwargs):
        self.asks.append(text)
        return default or ''


class CustomMultiAppStackTests(unittest.TestCase):
    def test_custom_opt_in_supports_two_web_apps_but_catalog_remains_guarded(self):
        self.assertEqual(assess(template()), [])
        self.assertTrue(assess(template(custom=False)))

    def test_custom_background_app_without_port_is_only_checked_running(self):
        result = template(text=COMPOSE.replace("    ports: ['7878:7878']\n", ''))
        self.assertEqual(assess(result), [])

    def test_service_image_contract_preserves_tag_and_digest(self):
        parent = template()
        service = parent['compose_stack']['services'][0]
        for reference, repository, tag, digest in (
            ('ghcr.io/example/app:2.4', 'ghcr.io/example/app', '2.4', None),
            ('ghcr.io/example/app:2.4@sha256:'+'a'*64, 'ghcr.io/example/app', '2.4', 'sha256:'+'a'*64),
        ):
            with self.subTest(reference=reference):
                service['image'] = reference
                service['compose']['image'] = reference
                image = service_template(parent, service)['container_contract']['image']
                self.assertEqual((image['reference'], image['repository'], image['tag'], image['digest']),
                                 (reference, repository, tag, digest))
                self.assertEqual(image['registry'], 'ghcr.io')

    def test_compose_extension_metadata_is_accepted_only_for_custom(self):
        text = 'x-common-environment: &common {PUID: \'1000\'}\n' + COMPOSE
        self.assertEqual(assess(template(text=text)), [])
        self.assertTrue(assess(template(custom=False, text=text)))

    def test_shared_owners_must_match(self):
        result = template(text=COMPOSE.replace("environment: {PUID: '1000', PGID: '1000'}", "environment: {PUID: '1001', PGID: '1000'}", 1))
        self.assertTrue(any('shared' in e for e in assess(result)))

    def test_advanced_addresses_are_kept_per_service(self):
        ui = UI()
        supplied = {'sonarr': '192.0.2.21/24', 'radarr': '192.0.2.22/24'}
        with patch('proxmenux_oci.installer.ask_storage', return_value='local-zfs'), \
             patch('proxmenux_oci.installer.ask_bridge', return_value='vmbr0'), \
             patch('proxmenux_oci.host.timezone', return_value='UTC'), \
             patch('proxmenux_oci.custom_mounts.ask_stack_custom_mounts'), \
             patch('proxmenux_oci.extra_devices.ask_stack_extra_devices'), \
             patch('proxmenux_oci.network.ask_addresses', return_value=(supplied, '192.0.2.1')):
            plan = build_stack(template(), ui, ADVANCED_MODE)
        self.assertEqual({s['name']:s['frontend_ipv4'] for s in plan['services']}, supplied)
        self.assertEqual(plan['network']['frontend_gateway'], '192.0.2.1')

    def test_advanced_custom_resources_are_selected_per_service(self):
        class ResourceUI(UI):
            def ask(self,text,default=None,**kwargs):
                if text=='radarr: Memory in MB': return '768'
                if text=='radarr: CPU cores': return '3'
                if text=='Memory in MB': return '2048'
                if text=='CPU cores': return '2'
                return super().ask(text,default,**kwargs)
        with patch('proxmenux_oci.installer.ask_storage',return_value='local-zfs'), \
             patch('proxmenux_oci.installer.ask_bridge',return_value='vmbr0'), \
             patch('proxmenux_oci.host.timezone',return_value='UTC'), \
             patch('proxmenux_oci.custom_mounts.ask_stack_custom_mounts'), \
             patch('proxmenux_oci.extra_devices.ask_stack_extra_devices'), \
             patch('proxmenux_oci.network.ask_addresses',return_value=({'sonarr':'dhcp','radarr':'dhcp'},None)):
            plan=build_stack(template(),ResourceUI(),ADVANCED_MODE)
        resources={service['name']:service['deployment']['resources'] for service in plan['services']}
        self.assertEqual(resources['radarr']['memory_mb'],768)
        self.assertEqual(resources['radarr']['cores'],3)
        self.assertEqual(resources['sonarr']['memory_mb'],2048)
        self.assertEqual(resources['sonarr']['cores'],2)

    def test_build_assigns_distinct_endpoints_and_shared_bind_source(self):
        ui = UI()
        with patch('proxmenux_oci.installer.ask_storage', return_value='local-zfs'), \
             patch('proxmenux_oci.installer.ask_bridge', return_value='vmbr0'), \
             patch('proxmenux_oci.host.timezone', return_value='UTC'), \
             patch('proxmenux_oci.network.ask_addresses', side_effect=lambda ui, bridge, labels: ({name:'dhcp' for name in labels}, None)):
            plan = build_stack(template(), ui, DEFAULT_MODE)
        members = {s['name']:s for s in plan['services']}
        self.assertEqual(members['sonarr']['healthcheck']['endpoint']['port'], 8989)
        self.assertEqual(members['radarr']['healthcheck']['endpoint']['port'], 7878)
        for member in members.values():
            self.assertTrue(member['frontend'])
            self.assertEqual(member['frontend_ipv4'], 'dhcp')
        binds = [next(m for m in s['deployment']['mounts'] if m['container_path']=='/data') for s in members.values()]
        self.assertEqual(binds[0]['source'], binds[1]['source'])
        self.assertEqual(binds[0]['host_owner'], [101000, 101000])
        self.assertTrue(all(m['type']=='host-bind' and m['backup'] is False for m in binds))
        self.assertEqual(len([text for text in ui.asks if text.endswith('shared-data')]), 1)
        configs = [next(m for m in s['deployment']['mounts'] if m['container_path']=='/config') for s in members.values()]
        self.assertTrue(all(m['type']=='managed-volume' for m in configs))

    def test_shared_directory_without_puid_uses_imported_image_owner(self):
        text = COMPOSE.replace("    environment: {PUID: '1000', PGID: '1000'}\n", '')
        with patch('proxmenux_oci.installer.ask_storage', return_value='local-zfs'), \
             patch('proxmenux_oci.installer.ask_bridge', return_value='vmbr0'), \
             patch('proxmenux_oci.host.timezone', return_value='UTC'), \
             patch('proxmenux_oci.network.ask_addresses', side_effect=lambda ui, bridge, labels: ({name:'dhcp' for name in labels}, None)):
            plan = build_stack(template(text=text), UI(), DEFAULT_MODE)
        for service in plan['services']:
            shared = next(m for m in service['deployment']['mounts'] if m['container_path'] == '/data')
            self.assertNotIn('host_owner', shared)


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