63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
import pytest
|
|
|
|
from unittest.mock import patch
|
|
|
|
from shlax import *
|
|
from shlax import proc
|
|
|
|
|
|
test_args_params = [
|
|
(
|
|
Localhost(Run('echo hi')),
|
|
[('sh', '-euc', 'echo hi')]
|
|
),
|
|
(
|
|
Localhost(Run('echo hi', user='jimi')),
|
|
[('sudo', '-u', 'jimi', 'sh', '-euc', 'echo hi')]
|
|
),
|
|
(
|
|
Localhost(Run('echo hi', user='root')),
|
|
[('sudo', 'sh', '-euc', 'echo hi')]
|
|
),
|
|
(
|
|
Ssh('host', Run('echo hi', user='root')),
|
|
[('ssh', 'host', 'sudo', 'sh', '-euc', 'echo hi')]
|
|
),
|
|
(
|
|
Buildah('alpine', Run('echo hi')),
|
|
[
|
|
('buildah', 'from', 'alpine'),
|
|
('buildah', 'mount', ''),
|
|
('buildah', 'run', '', '--', 'sh', '-euc', 'echo hi'),
|
|
('buildah', 'rm', ''),
|
|
]
|
|
),
|
|
(
|
|
Buildah('alpine', Run('echo hi', user='root')),
|
|
[
|
|
('buildah', 'from', 'alpine'),
|
|
('buildah', 'mount', ''),
|
|
('buildah', 'run', '--user', 'root', '', '--', 'sh', '-euc', 'echo hi'),
|
|
('buildah', 'rm', ''),
|
|
]
|
|
),
|
|
(
|
|
Ssh('host', Buildah('alpine', Run('echo hi', user='root'))),
|
|
[
|
|
('ssh', 'host', 'buildah', 'from', 'alpine'),
|
|
('ssh', 'host', 'buildah', 'mount', ''),
|
|
('ssh', 'host', 'buildah', 'run', '--user', 'root', '', '--', 'sh', '-euc', 'echo hi'),
|
|
('ssh', 'host', 'buildah', 'rm', ''),
|
|
]
|
|
),
|
|
]
|
|
@pytest.mark.parametrize(
|
|
'script,commands',
|
|
test_args_params
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_args(script, commands):
|
|
with Proc.mock():
|
|
await script()
|
|
assert commands == Proc.test
|