"""Test Snowflake helper utilities."""
from dbdeploy.util.snowflake import helpers


class TestSnoflakeConfig:
    """Test snowflake config helper."""

    def test_get_sf_config(self, mock_base_config):
        """Test correct snowflake config generated."""
        test_sf_config = helpers.get_sf_config('orchard')
        test_sf_config_keys = {'role', 'warehouse', 'db', 'schema', 'user', 'account', 'private_key'}  # noqa: E501
        assert isinstance(test_sf_config['private_key'], bytes)
        assert isinstance(test_sf_config, dict)
        assert test_sf_config_keys == set(test_sf_config.keys())