def test_template_query_basic(mock_connect): """Test query with Pretty output.""" runner = CliRunner() with runner.isolated_filesystem(): test_fixtures.populate_isolated_filesystem( 'basic_config.yml', 'fake_template_query.sql' ) result = runner.invoke( cli.cli, ['-c', 'config.yml', '-o', 'pretty', 'template_query', '-v', 'last_name:Troxel', 'query.sql'] ) assert result.exit_code == 0 assert result.output in [ test_fixtures.get_file_contents( 'pretty_outputs/query_expected_pretty_output3' ), test_fixtures.get_file_contents( 'pretty_outputs/query_expected_pretty_output4' ) ] assert mock_connect.called_with( 'MY_MSSQL.example.com', 'a_user', 'a_password' )
def test_query_csv(mock_connect): """Test query with CSV output.""" runner = CliRunner() with runner.isolated_filesystem(): test_fixtures.populate_isolated_filesystem( 'basic_config.yml', 'fake_query.sql' ) result = runner.invoke( cli.cli, ['-c', 'config.yml', '-o', 'csv', 'query', 'query.sql'] ) assert result.exit_code == 0 assert result.output in [ test_fixtures.get_file_contents( 'csv_outputs/query_expected_csv_output.csv' ), test_fixtures.get_file_contents( 'csv_outputs/query_expected_csv_output_2.csv' ) ] assert mock_connect.called_with( 'MY_MSSQL.example.com', 'a_user', 'a_password' )
def test_query_basic(mock_connect): """Test query with basic options (json).""" runner = CliRunner() with runner.isolated_filesystem(): test_fixtures.populate_isolated_filesystem('basic_config.yml', 'fake_query.sql') result = runner.invoke(cli.cli, ['query', '-c', 'config.yml', 'query.sql']) assert result.exit_code == 0 assert json.loads(result.output) == json.loads( test_fixtures.get_file_contents( 'json_outputs/query_expected_json_output.json')) assert mock_connect.called_with('MY_MSSQL.example.com', 'a_user', 'a_password')
def test_query_extended_config(mock_connect): """Test query with extended config.""" runner = CliRunner() with runner.isolated_filesystem(): test_fixtures.populate_isolated_filesystem( 'extended_config.yml', 'fake_query.sql' ) result = runner.invoke( cli.cli, ['-c', 'config.yml', 'query', 'query.sql'] ) assert result.exit_code == 0 assert mock_connect.called_with( 'MY_MSSQL.example.com', 'a_user', 'a_password', port=2345, timeout=10 )