Feat: Add test case for base64 encoded private key

dlt-hub · Sep 14, 2023 · 5f8f03b · 5f8f03b
1 parent b40215d
commit 5f8f03b
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/tests/common/cases/secrets/encrypted-private-key-base64 b/tests/common/cases/secrets/encrypted-private-key-base64
@@ -0,0 +1 @@
+LS0tLS1CRUdJTiBFTkNSWVBURUQgUFJJVkFURSBLRVktLS0tLQpNSUlGTFRCWEJna3Foa2lHOXcwQkJRMHdTakFwQmdrcWhraUc5dzBCQlF3d0hBUUlsL0ErZlkrSVFMVUNBZ2dBCk1Bd0dDQ3FHU0liM0RRSUpCUUF3SFFZSllJWklBV1VEQkFFcUJCQ2krUnBJVmpEN1hBYURHLzNoa3F5dEJJSUUKMEtnODJvbnhqZnp5VFpFVVFtU3ZsVkw4cncwY0hTWk92RS96TkFmTFhwRjcva213OVVWWkxMLy9jbHcybWlpNgo2ZnNSRlVLd1RTNVJYYWtRcDh2ZWllTHpDT3g2bzFLd0lrMGFvOUE3c0ptd1BMejBsTXZoWEsvdHlGanM5dUhVCm4yWWNTSEo0WGdzUGRnL2xKc1RjbjRwNFFMNTJLREFFSUUvY2pSVktiZEljbGFYN3pQZHBWSzhXQ2NQVWNSK2cKOWxJbTNIQnl4ZDd3NklDRXh0WXFabGFCbE56S3NTeVdZNmNOR0J5bG91WHR2YktidHFpNEhnaThpaDdlc2ZFTgpqZU0zYmhudVpQSmViQVBHYWFxQ2cyMHlKNDQyOGpEWlpVSUlKbEx1cnZhZ2JwME11S0tTRy83YURlQzVNaUlMCktMRHY2UjJZKzNoK3ZhQy8xOWlwVHFtZXlUZ3NDQXFpSVdhWUtpTzVYRnE2UWwyZXNkSVhZUjFPQ0h3MUxNOXQKMHh2VnZyZmVaNHllWXA5SGo4MTFJQTJZV3ZUckVMMjA0VUwyUHIwcDByV1B4a1QrakhPdTFjNHlZSDI3MXNPVwpIQzdqZ2VwRUM4ckNVejZnbW82OFRPL04yZGhTVk51QjI2ZnF2dnZpVG5iK3p2SXp3ak41STA0eHpJWGlMdVVQCkZZS1BsWVVNOFpPN2htcmxWY1hWOTgrSDFsUFBPZVpOZU81RDFJODZSSkJ4SUpvdmxzazlqSW5CL2JsZTN4d3gKL1UzcU91UUZRRXBWUXJ0VktwVlc4VmJGZ0VWUHRTU2Q4eklzY1NjbTM4STcyMU1XT3E1cnJFOG03TFA5Y3lFNQplZzQyTUhPa0ZOcFd2YUt4Q3BSbDA0ckp1bnZZd0hzRjluRnZJV2hON1RWbmx2dVF6SnNMK29pWFArMTJzQzRjCi9jTjloWDZJNjBmNTBaV2FXNDlhVVNQdWRRcVc4Znd4MzVpREdHVmdIb0RQMzhoTUE2Y2hCT21pb3gyTG8rR0oKVHdUSjlBbGU2dVVtaWplV0NSMUEyZE1xS3dSenlsQlpwL0hIMGljcHhRdEhPTm44SDd0Y01JNWg3VnFiZjgvSwp6Q0ZZeHNZampja3lETlI1NElzV25iR0ZlK3ZBVTRTYTJmc1hPMHR6STN5SHArVExSeC9hVlZ4NEhuTURnQ2NUCmMyUEx2TWJsKzJzdHlwekxMQzl5MitEYjM5Wmd0WjZzNGpadTdZTUkxTStXWnByL0FMQ2t2NWZHVGYyZHoycEkKVkdEaTNXTHI5dWZMcnA2eHMreTVQaTRkTER3aDdZbTUwekhNOVNrdnMzSTBONDFlNEtDOGNrMmZEUmZrRjNkOQpsdHBRNXNDaDIvb3g4cjdXSk1XNjZXaUVzdkU4cUZLbCs0dHIxWE9Md1JwQTVZNTFZcXZjMGxqeWkrVFNqdVhpCitZRjFaK3hoUFhKUElZMFBOVWtDUmptNmhxVDJlRWIyT3pNazZEcE9EVkorK2NBcXBoV240SHI0eGhVVjVBYXEKQTBIZEJCcTczVDlHNVAyaU5XdUJib0hTTzNLUncxT3BiRHRDc1ZvN1h3V2NBdU9oOTg2ck0zelhPcDdsd29ERQphMHNMay9Kd1pndEVPTndqeGlXWk4zWHZOYWl4aVJ2MEt3NnNsUFBDYytkWlRSN0VIdHVZZXlGTXBtK2VZQkoxClhXbTBGMFo4RERRSkExYkZqbTUyeDNhR3hnVWFHYkVyZHNFb0ZpcHFoVURaWXhjeVhucythRFhuMURkZFJnbDMKZjFzQ09FSkFyNlpQVStlNkszMDE1NEFLcHZQMlJYTTZFNjVjR2lxUU1NKzdSV3czOGdrRlZPL1NqdEZmcGtuVQp0dmZjZFBYcTltVkQ4SXFNU0lkQVNka3Y5bmN1L25pTXNZS1crMG40djF0ZzJQeC9WaUVEOVhkSEtTUDNEbjExCkpvWnR0bkhUenFKQWl5emtUd0pNYjhVTHlIb2JsN2pEZFBmaTJkam1ZcnVnTEZDaDBOemRaTnBXNFdoeE5LejAKellqTTVuTnAzWXFOajVpQnAxTDVCUnBiSHdwbGhvenFTZGo5Y2Z1MWJRdWFBRnZya3NCMDhXa2puSkdSWFdSbApPUHMrRStyNGU0SFY3ZUxET2Y3MTFxRWhOVTJzNVVCODVpY3Fzc3AxWU52TgotLS0tLUVORCBFTkNSWVBURUQgUFJJVkFURSBLRVktLS0tLQo=
diff --git a/tests/load/snowflake/test_snowflake_configuration.py b/tests/load/snowflake/test_snowflake_configuration.py
@@ -15,7 +15,7 @@
 
 
 def test_connection_string_with_all_params() -> None:
-    url = "snowflake://user1:pass1@host1/db1?warehouse=warehouse1&role=role1&private_key=pk&private_key_passphrase=paphr"
+    url = "snowflake://user1:pass1@host1/db1?warehouse=warehouse1&role=role1&private_key=cGsK&private_key_passphrase=paphr"
 
     creds = SnowflakeCredentials()
     creds.parse_native_representation(url)
@@ -60,6 +60,31 @@ def test_to_connector_params() -> None:
         warehouse='warehouse1',
         role='role1',
     )
+
+    pkey_str = Path('./tests/common/cases/secrets/encrypted-private-key-base64').read_text('utf8')
+
+    creds = SnowflakeCredentials()
+    creds.private_key = pkey_str  # type: ignore[assignment]
+    creds.private_key_passphrase = '12345'  # type: ignore[assignment]
+    creds.username = 'user1'
+    creds.database = 'db1'
+    creds.host = 'host1'
+    creds.warehouse = 'warehouse1'
+    creds.role = 'role1'
+
+    params = creds.to_connector_params()
+
+    assert isinstance(params['private_key'], bytes)
+    params.pop('private_key')
+
+    assert params == dict(
+        user='user1',
+        database='db1',
+        account='host1',
+        password=None,
+        warehouse='warehouse1',
+        role='role1',
+    )
 
 
 def test_snowflake_credentials_native_value(environment) -> None:
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		LS0tLS1CRUdJTiBFTkNSWVBURUQgUFJJVkFURSBLRVktLS0tLQpNSUlGTFRCWEJna3Foa2lHOXcwQkJRMHdTakFwQmdrcWhraUc5dzBCQlF3d0hBUUlsL0ErZlkrSVFMVUNBZ2dBCk1Bd0dDQ3FHU0liM0RRSUpCUUF3SFFZSllJWklBV1VEQkFFcUJCQ2krUnBJVmpEN1hBYURHLzNoa3F5dEJJSUUKMEtnODJvbnhqZnp5VFpFVVFtU3ZsVkw4cncwY0hTWk92RS96TkFmTFhwRjcva213OVVWWkxMLy9jbHcybWlpNgo2ZnNSRlVLd1RTNVJYYWtRcDh2ZWllTHpDT3g2bzFLd0lrMGFvOUE3c0ptd1BMejBsTXZoWEsvdHlGanM5dUhVCm4yWWNTSEo0WGdzUGRnL2xKc1RjbjRwNFFMNTJLREFFSUUvY2pSVktiZEljbGFYN3pQZHBWSzhXQ2NQVWNSK2cKOWxJbTNIQnl4ZDd3NklDRXh0WXFabGFCbE56S3NTeVdZNmNOR0J5bG91WHR2YktidHFpNEhnaThpaDdlc2ZFTgpqZU0zYmhudVpQSmViQVBHYWFxQ2cyMHlKNDQyOGpEWlpVSUlKbEx1cnZhZ2JwME11S0tTRy83YURlQzVNaUlMCktMRHY2UjJZKzNoK3ZhQy8xOWlwVHFtZXlUZ3NDQXFpSVdhWUtpTzVYRnE2UWwyZXNkSVhZUjFPQ0h3MUxNOXQKMHh2VnZyZmVaNHllWXA5SGo4MTFJQTJZV3ZUckVMMjA0VUwyUHIwcDByV1B4a1QrakhPdTFjNHlZSDI3MXNPVwpIQzdqZ2VwRUM4ckNVejZnbW82OFRPL04yZGhTVk51QjI2ZnF2dnZpVG5iK3p2SXp3ak41STA0eHpJWGlMdVVQCkZZS1BsWVVNOFpPN2htcmxWY1hWOTgrSDFsUFBPZVpOZU81RDFJODZSSkJ4SUpvdmxzazlqSW5CL2JsZTN4d3gKL1UzcU91UUZRRXBWUXJ0VktwVlc4VmJGZ0VWUHRTU2Q4eklzY1NjbTM4STcyMU1XT3E1cnJFOG03TFA5Y3lFNQplZzQyTUhPa0ZOcFd2YUt4Q3BSbDA0ckp1bnZZd0hzRjluRnZJV2hON1RWbmx2dVF6SnNMK29pWFArMTJzQzRjCi9jTjloWDZJNjBmNTBaV2FXNDlhVVNQdWRRcVc4Znd4MzVpREdHVmdIb0RQMzhoTUE2Y2hCT21pb3gyTG8rR0oKVHdUSjlBbGU2dVVtaWplV0NSMUEyZE1xS3dSenlsQlpwL0hIMGljcHhRdEhPTm44SDd0Y01JNWg3VnFiZjgvSwp6Q0ZZeHNZampja3lETlI1NElzV25iR0ZlK3ZBVTRTYTJmc1hPMHR6STN5SHArVExSeC9hVlZ4NEhuTURnQ2NUCmMyUEx2TWJsKzJzdHlwekxMQzl5MitEYjM5Wmd0WjZzNGpadTdZTUkxTStXWnByL0FMQ2t2NWZHVGYyZHoycEkKVkdEaTNXTHI5dWZMcnA2eHMreTVQaTRkTER3aDdZbTUwekhNOVNrdnMzSTBONDFlNEtDOGNrMmZEUmZrRjNkOQpsdHBRNXNDaDIvb3g4cjdXSk1XNjZXaUVzdkU4cUZLbCs0dHIxWE9Md1JwQTVZNTFZcXZjMGxqeWkrVFNqdVhpCitZRjFaK3hoUFhKUElZMFBOVWtDUmptNmhxVDJlRWIyT3pNazZEcE9EVkorK2NBcXBoV240SHI0eGhVVjVBYXEKQTBIZEJCcTczVDlHNVAyaU5XdUJib0hTTzNLUncxT3BiRHRDc1ZvN1h3V2NBdU9oOTg2ck0zelhPcDdsd29ERQphMHNMay9Kd1pndEVPTndqeGlXWk4zWHZOYWl4aVJ2MEt3NnNsUFBDYytkWlRSN0VIdHVZZXlGTXBtK2VZQkoxClhXbTBGMFo4RERRSkExYkZqbTUyeDNhR3hnVWFHYkVyZHNFb0ZpcHFoVURaWXhjeVhucythRFhuMURkZFJnbDMKZjFzQ09FSkFyNlpQVStlNkszMDE1NEFLcHZQMlJYTTZFNjVjR2lxUU1NKzdSV3czOGdrRlZPL1NqdEZmcGtuVQp0dmZjZFBYcTltVkQ4SXFNU0lkQVNka3Y5bmN1L25pTXNZS1crMG40djF0ZzJQeC9WaUVEOVhkSEtTUDNEbjExCkpvWnR0bkhUenFKQWl5emtUd0pNYjhVTHlIb2JsN2pEZFBmaTJkam1ZcnVnTEZDaDBOemRaTnBXNFdoeE5LejAKellqTTVuTnAzWXFOajVpQnAxTDVCUnBiSHdwbGhvenFTZGo5Y2Z1MWJRdWFBRnZya3NCMDhXa2puSkdSWFdSbApPUHMrRStyNGU0SFY3ZUxET2Y3MTFxRWhOVTJzNVVCODVpY3Fzc3AxWU52TgotLS0tLUVORCBFTkNSWVBURUQgUFJJVkFURSBLRVktLS0tLQo=