mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-11-01 18:32:13 +00:00
154 lines
7.8 KiB
Python
154 lines
7.8 KiB
Python
"""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()
|