[NBS] ability to override the NBS's binary file path; tweak encryptio…

…n_at_rest

cloud/blockstore/tests/encryption_at_rest/test.py

@@ -177,7 +177,7 @@ def read_first_block():
     assert vol0.EncryptionDesc.Mode == ENCRYPTION_DEFAULT_AES_XTS
 
     assert vol0.EncryptionDesc.EncryptionKey.KekId == KEK_ID
-    assert len(vol0.EncryptionDesc.EncryptionKey.EncryptedDEK) == 512
+    assert len(vol0.EncryptionDesc.EncryptionKey.EncryptedDEK) >= 32
 
     expected_data = os.urandom(4096)
 

cloud/blockstore/tests/python/lib/daemon.py

@@ -1,7 +1,8 @@
 import json
+import os
 import requests
-import subprocess
 import socket
+import subprocess
 import time
 
 from .config import NbsConfigurator
@@ -22,6 +23,14 @@
     ensure_path_exists
 
 
+def __get_bin_path(name, default):
+    p = yatest_common.get_param(name)
+    if p is None:
+        p = os.environ.get(name, f"{default}|{p}|{name}")
+
+    return yatest_common.binary_path(p)
+
+
 def _match(labels, query):
     for name, value in query.items():
         v = labels.get(name)
@@ -145,7 +154,10 @@ def wait_for_registration(self, delay_sec=1, max_retry_number=None):
 
 
 def start_nbs(config: NbsConfigurator, name='nbs-server'):
-    exe_path = yatest_common.binary_path("cloud/blockstore/apps/server/nbsd")
+
+    exe_path = __get_bin_path(
+        "NBS_SERVER_PATH",
+        "cloud/blockstore/apps/server/nbsd")
 
     cwd = get_unique_path_for_current_test(
         output_path=yatest_common.output_path(),
@@ -176,7 +188,10 @@ def start_nbs(config: NbsConfigurator, name='nbs-server'):
 
 
 def start_disk_agent(config: NbsConfigurator, name='disk-agent'):
-    exe_path = yatest_common.binary_path("cloud/blockstore/apps/disk_agent/diskagentd")
+
+    exe_path = __get_bin_path(
+        "DISK_AGENT_PATH",
+        "cloud/blockstore/apps/disk_agent/diskagentd")
 
     cwd = get_unique_path_for_current_test(
         output_path=yatest_common.output_path(),