b38dea0597
Share Gaia assets and adapters, keep private profiles and development documents local, and update both public READMEs.
40 lines
2.1 KiB
Python
40 lines
2.1 KiB
Python
# SPDX-License-Identifier: MPL-2.0
|
|
import importlib.util
|
|
from pathlib import Path
|
|
import subprocess
|
|
import sys
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
TOOLS = Path(__file__).resolve().parents[1]/'tools'
|
|
sys.path.insert(0, str(TOOLS))
|
|
spec = importlib.util.spec_from_file_location('ssh_inventory', TOOLS/'collect-device-ssh.py')
|
|
inventory = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(inventory)
|
|
|
|
|
|
class SSHInventoryTests(unittest.TestCase):
|
|
def test_model_and_compatible_do_not_confuse_sargo_and_bonito(self):
|
|
self.assertEqual(inventory.identify('Google Pixel 3a\0', ''), 'sargo')
|
|
self.assertEqual(inventory.identify('', 'google,sargo\0qcom,sdm670\0'), 'sargo')
|
|
self.assertEqual(inventory.identify('Google Pixel 3a XL', 'google,sargo'), 'wrong-device-pixel-3a-xl')
|
|
self.assertEqual(inventory.identify('Qualcomm SDM670', ''), 'unverified')
|
|
self.assertEqual(inventory.identify('Google Inc. MSM sdm670 S4 PVT v1.0',
|
|
'google,b4s4-sdm670\0qcom,sdm670\0', 'sargo\n'), 'sargo')
|
|
self.assertEqual(inventory.identify('Google Pixel 3a', '', 'bonito'), 'wrong-device-pixel-3a-xl')
|
|
|
|
def test_remote_argument_is_quoted_and_fallback_connection_disabled(self):
|
|
base = inventory.ssh_base('/tmp/test-control', '10.15.19.82', 'droidian')
|
|
self.assertIn('ProxyCommand=false', base)
|
|
result = subprocess.CompletedProcess([], 0, '', '')
|
|
with patch.object(inventory.subprocess, 'run', return_value=result) as run:
|
|
inventory.capture(base, ['dpkg-query', '-f', '${Version}\n'])
|
|
self.assertEqual(run.call_args.args[0][-1], "dpkg-query -f '${Version}\n'")
|
|
self.assertNotIn('shell', run.call_args.kwargs)
|
|
|
|
def test_read_commands_have_no_privileged_or_mutating_entrypoints(self):
|
|
allowed = {'id', 'cat', 'uname', 'lsblk', 'ls', 'systemctl', 'dpkg-query', 'getprop'}
|
|
for command in inventory.READS.values():
|
|
self.assertIn(command[0], allowed)
|
|
if command[0] == 'systemctl':
|
|
self.assertEqual(command[1], 'list-units')
|