mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-09-30 10:36:41 +00:00
38 lines
1.7 KiB
Python
38 lines
1.7 KiB
Python
 
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
import unittest
|
||
|
|
from unittest.mock import MagicMock, patch
|
||
|
|
|
||
|
|
SCRIPTS = Path(__file__).resolve().parents[1]
|
||
|
|
sys.path.insert(0, str(SCRIPTS))
|
||
|
|
import lxc_apps
|
||
|
|
|
||
|
|
|
||
|
|
class AdguardSetupTests(unittest.TestCase):
|
||
|
|
def test_non_adguard_is_not_probed(self):
|
||
|
|
with patch.object(lxc_apps, '_oci_instance_meta', return_value={'template_id': 'image-frigate'}), \
|
||
|
|
patch.object(lxc_apps.subprocess, 'run') as run:
|
||
|
|
self.assertFalse(lxc_apps.oci_adguard_setup_available(190))
|
||
|
|
run.assert_not_called()
|
||
|
|
|
||
|
|
def test_setup_returns_true_only_for_http_response(self):
|
||
|
|
process = MagicMock(stdout='192.168.0.42\n')
|
||
|
|
connection = MagicMock()
|
||
|
|
connection.__enter__.return_value.recv.return_value = b'HTTP/1.1 302 Found\r\n'
|
||
|
|
with patch.object(lxc_apps, '_oci_instance_meta', return_value={'template_id': 'image-adguard-home'}), \
|
||
|
|
patch.object(lxc_apps.subprocess, 'run', return_value=process), \
|
||
|
|
patch.object(lxc_apps.socket, 'create_connection', return_value=connection) as connect:
|
||
|
|
self.assertTrue(lxc_apps.oci_adguard_setup_available(190))
|
||
|
|
connect.assert_called_once_with(('192.168.0.42', 3000), timeout=1)
|
||
|
|
|
||
|
|
def test_closed_setup_port_is_not_offered(self):
|
||
|
|
process = MagicMock(stdout='192.168.0.42\n')
|
||
|
|
with patch.object(lxc_apps, '_oci_instance_meta', return_value={'template_id': 'image-adguard-home'}), \
|
||
|
|
patch.object(lxc_apps.subprocess, 'run', return_value=process), \
|
||
|
|
patch.object(lxc_apps.socket, 'create_connection', side_effect=OSError):
|
||
|
|
self.assertFalse(lxc_apps.oci_adguard_setup_available(190))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|