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)