def test_get_connection_empty_schema(
self, mock_keyring_get_password: MagicMock,
mock_user_config_dir: MagicMock) -> None:
redshift_connector = RedshiftConnector()
mock_user_config_dir.assert_called_once_with('pypandasql')
mock_keyring_get_password.assert_called_once_with(
'pypandasql-redshift', 'test')
with self.assertRaises(AssertionError):
redshift_connector.get_connection(schema='')
def test_get_connection(self, mock_connect: MagicMock,
mock_keyring_get_password: MagicMock,
mock_user_config_dir: MagicMock) -> None:
redshift_connector = RedshiftConnector()
redshift_connector.get_connection(schema='test-schema')
mock_user_config_dir.assert_called_once_with('pypandasql')
mock_keyring_get_password.assert_called_once_with(
'pypandasql-redshift', 'test')
mock_connect.assert_called_once_with(dbname='test-schema',
host='test-host',
port=1234,
user='******',
password='******')