1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
#
import os
import sys
import tempfile
from later.unittest import TestCase
from nubia import argument, command
from tests.util import TestShell
class ReadStdinTest(TestCase):
async def test_read_from_stdin(self):
@command
@argument("arg")
def test_command(arg: str) -> int:
"""
Sample Docstring
"""
self.assertEqual("test_arg", arg)
return 22
command_file = tempfile.NamedTemporaryFile(
mode="w+", prefix="test_read_from_stdin", delete=True
)
command_file.write("test-command arg=test_arg")
command_file.flush()
os.lseek(command_file.fileno(), 0, os.SEEK_SET)
os.dup2(command_file.fileno(), sys.stdin.fileno())
shell = TestShell(commands=[test_command])
self.assertEqual(22, await shell.run_async(cli_args=["", "connect"]))
|
