fix test_generate_public_url_for_kvs_record

Author: vdusek

tests/integration/test_actor_key_value_store.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import os
 from typing import TYPE_CHECKING
 
 from apify_shared.consts import ApifyEnvVars
@@ -205,7 +206,19 @@ async def main() -> None:
 
         async with Actor:
             public_api_url = Actor.config.api_public_base_url
-            default_store_id = Actor.config.default_key_value_store_id
+
+            default_kvs_id = (
+                os.environ.get(
+                    'ACTOR_DEFAULT_KEY_VALUE_STORE_ID',
+                    None,
+                )
+                or os.environ.get(
+                    'APIFY_DEFAULT_KEY_VALUE_STORE_ID',
+                    None,
+                )
+                or 'default'
+            )
+
             record_key = 'public-record-key'
 
             kvs = await Actor.open_key_value_store()
@@ -221,7 +234,7 @@ async def main() -> None:
             signature = create_hmac_signature(url_signing_secret_key, record_key)
             assert (
                 record_url
-                == f'{public_api_url}/v2/key-value-stores/{default_store_id}/records/{record_key}?signature={signature}'
+                == f'{public_api_url}/v2/key-value-stores/{default_kvs_id}/records/{record_key}?signature={signature}'
             )
 
     actor = await make_actor(label='kvs-get-public-url', main_func=main)