def test_cli_connections_list(self):
with redirect_stdout(io.StringIO()) as stdout:
connection_command.connections_list(self.parser.parse_args(["connections", "list"]))
stdout = stdout.getvalue()
lines = stdout.split("\n")
for conn_id, conn_type in self.EXPECTED_CONS:
self.assertTrue(any(conn_id in line and conn_type in line for line in lines))
def test_cli_connections_filter_conn_id(self):
args = self.parser.parse_args(
["connections", "list", "--output", "json", '--conn-id', 'http_default']
)
with redirect_stdout(io.StringIO()) as stdout:
connection_command.connections_list(args)
stdout = stdout.getvalue()
self.assertIn("http_default", stdout)
def test_cli_connections_list_as_json(self):
args = self.parser.parse_args(["connections", "list", "--output", "json"])
with redirect_stdout(io.StringIO()) as stdout:
connection_command.connections_list(args)
print(stdout.getvalue())
stdout = stdout.getvalue()
for conn_id, conn_type in self.EXPECTED_CONS:
self.assertIn(conn_type, stdout)
self.assertIn(conn_id, stdout)
def test_cli_connections_filter_conn_id(self):
args = self.parser.parse_args(["connections", "list", "--output", "tsv", '--conn-id', 'http_default'])
with redirect_stdout(io.StringIO()) as stdout:
connection_command.connections_list(args)
stdout = stdout.getvalue()
lines = stdout.split("\n")
conn_ids = [line.split("\t", 2)[0].strip() for line in lines[1:] if line]
self.assertEqual(conn_ids, ['http_default'])
def test_cli_connections_include_secrets(self):
args = self.parser.parse_args([
"connections",
"list",
"--output",
"tsv",
'--include-secrets',
])
with self.assertRaises(SystemExit):
connection_command.connections_list(args)
def test_cli_connections_list(self):
with mock.patch('sys.stdout', new_callable=io.StringIO) as mock_stdout:
connection_command.connections_list(
self.parser.parse_args(['connections', 'list']))
stdout = mock_stdout.getvalue()
conns = [[x.strip("'") for x in re.findall(r"'\w+'", line)[:2]]
for ii, line in enumerate(stdout.split('\n')) if ii % 2 == 1]
conns = [conn for conn in conns if len(conn) > 0]
# Assert that some of the connections are present in the output as
# expected:
self.assertIn(['aws_default', 'aws'], conns)
self.assertIn(['hive_cli_default', 'hive_cli'], conns)
self.assertIn(['emr_default', 'emr'], conns)
self.assertIn(['mssql_default', 'mssql'], conns)
self.assertIn(['mysql_default', 'mysql'], conns)
self.assertIn(['postgres_default', 'postgres'], conns)
self.assertIn(['wasb_default', 'wasb'], conns)
self.assertIn(['segment_default', 'segment'], conns)
def test_cli_connections_list_with_args(self):
args = self.parser.parse_args(
['connections', 'list', '--output', 'tsv'])
connection_command.connections_list(args)
