def test_command_multiple_shell_failing(self):
@command(command_type=Type.SHELL)
def foo(*args, **kwargs):
return [['foo'], ['bar']]
args = ['foo']
main = Main(args)
with patch('clinner.run.base.Popen') as popen_mock:
popen_mock.return_value.returncode = 1
main.run()
self.assertEqual(popen_mock.call_count, 1)
self.assertEqual(popen_mock.call_args_list[0][1]['args'], ['foo'])
def test_command_multiple_shell_failing(self, cli):
@command(command_type=Type.SHELL)
def foo(*args, **kwargs):
return [["foo"], ["bar"]]
args = ["foo"]
main = Main(args)
with patch("clinner.run.base.Popen") as popen_mock:
popen_mock.return_value.returncode = 1
main.run()
assert popen_mock.call_count == 1
assert popen_mock.call_args_list[0][1]["args"] == ["foo"]
del command.register["foo"]
def test_command_wrong_type(self):
@command(command_type='foo')
def foo(*args, **kwargs):
pass
args = ['foo']
self.assertRaises(CommandTypeError, Main(args).run, args)
def test_command_wrong_type(self, cli):
@command(command_type="foo")
def foo(*args, **kwargs):
pass
args = ["foo"]
with pytest.raises(CommandTypeError):
Main(args).run(args)
del command.register["foo"]
def test_command_with_wrong_args_number(self, cli):
@command(command_type=Type.PYTHON, args=(((1, 2, 3),))) # wrong args number
def foo(*args, **kwargs):
kwargs["q"].put(kwargs["bar"])
args = ["foo"]
with pytest.raises(CommandArgParseError):
Main(args)
del command.register["foo"]
def test_command_python(self):
@command(command_type=Type.PYTHON)
def foo(*args, **kwargs):
kwargs['q'].put(42)
queue = Queue()
args = ['foo']
Main(args).run(q=queue)
self.assertEqual(queue.get(), 42)
def test_command(self):
@command
def foo(*args, **kwargs):
kwargs['q'].put(42)
queue = Queue()
args = ['foo']
Main(args).run(q=queue)
self.assertEqual(queue.get(), 42)
def test_command_with_unknown_args(self):
@command # args
def foo(*args, **kwargs):
kwargs['q'].put(args[0])
queue = Queue()
args = ['foo']
Main(args).run('foobar', q=queue)
self.assertEqual(queue.get(), 'foobar')
def test_command_with_args_and_opts(self):
@command(command_type=Type.PYTHON,
args=((('-b', '--bar'), {'type': int, 'help': 'bar argument'}),)) # args and opts
def foo(*args, **kwargs):
kwargs['q'].put(kwargs['bar'])
queue = Queue()
args = ['foo']
Main(args).run(bar=3, q=queue)
self.assertEqual(queue.get(), 3)
def test_command_with_args(self):
@command(command_type=Type.PYTHON,
args=((('-b', '--bar'),),)) # args
def foo(*args, **kwargs):
kwargs['q'].put(kwargs['bar'])
queue = Queue()
args = ['foo']
Main(args).run(bar='foobar', q=queue)
self.assertEqual(queue.get(), 'foobar')
def test_command(self, cli):
@command
def foo(*args, **kwargs):
kwargs["q"].put(42)
queue = Queue()
args = ["foo"]
Main(args).run(q=queue)
assert queue.get() == 42
del command.register["foo"]
def test_command_with_unknown_args(self, cli):
@command # args
def foo(*args, **kwargs):
kwargs["q"].put(args[0])
queue = Queue()
args = ["foo"]
Main(args).run("foobar", q=queue)
assert queue.get() == "foobar"
del command.register["foo"]
def test_command_with_args(self, cli):
@command(command_type=Type.PYTHON, args=((("-b", "--bar"),),)) # args
def foo(*args, **kwargs):
kwargs["q"].put(kwargs["bar"])
queue = Queue()
args = ["foo"]
Main(args).run(bar="foobar", q=queue)
assert queue.get() == "foobar"
del command.register["foo"]
def test_command_python_async(self, cli):
@command(command_type=Type.PYTHON)
async def foo(*args, **kwargs):
kwargs["q"].put(42)
queue = Queue()
args = ["foo"]
Main(args).run(q=queue)
assert queue.get() == 42
del command.register["foo"]
def test_command_with_callable_args(self):
def add_arguments(parser):
parser.add_argument('-b', '--bar', type=int, help='bar argument')
@command(args=add_arguments)
def foo(*args, **kwargs):
kwargs['q'].put(kwargs['bar'])
queue = Queue()
args = ['foo']
Main(args).run(bar=3, q=queue)
self.assertEqual(queue.get(), 3)
def test_command_with_args_and_opts(self, cli):
@command(
command_type=Type.PYTHON, args=((("-b", "--bar"), {"type": int, "help": "bar argument"}),)
) # args and opts
def foo(*args, **kwargs):
kwargs["q"].put(kwargs["bar"])
queue = Queue()
args = ["foo"]
Main(args).run(bar=3, q=queue)
assert queue.get() == 3
del command.register["foo"]
def test_command_with_callable_args(self, cli):
def add_arguments(parser):
parser.add_argument("-b", "--bar", type=int, help="bar argument")
@command(args=add_arguments)
def foo(*args, **kwargs):
kwargs["q"].put(kwargs["bar"])
queue = Queue()
args = ["foo"]
Main(args).run(bar=3, q=queue)
assert queue.get() == 3
del command.register["foo"]
condition = Condition()
if kwargs["unlock"]:
unlock_thread = Thread(name="unlock", target=unlock, args=(condition,))
unlock_thread.start()
socat_cmd = shlex.split("socat tcp-listen:8000,reuseaddr,fork tcp:localhost:9980")
siad_cmd = shlex.split(f"siad --sia-directory {os.getcwd()}") + list(args)
try:
with subprocess.Popen(socat_cmd) as socat_process:
with subprocess.Popen(
siad_cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, bufsize=1, universal_newlines=True
) as siad_process:
redirect_output_thread = Thread(
name="redirect_output", target=redirect_output, args=(siad_process, condition)
)
redirect_output_thread.start()
siad_process.wait()
socat_process.terminate()
except KeyboardInterrupt:
pass
finally:
logger.info("Siad command finished with code %d", siad_process.returncode)
logger.info("Socat command finished with code %d", socat_process.returncode)
if __name__ == "__main__":
sys.exit(Main().run())