From 273f0d13a52c63f45eda5a4c92df01b8ce76a309 Mon Sep 17 00:00:00 2001
From: Andrey Berezin <a.berezin@yadro.com>
Date: Tue, 20 Feb 2024 13:27:45 +0300
Subject: [PATCH] [#184] Add streaming param

Signed-off-by: Andrey Berezin <a.berezin@yadro.com>
---
 src/frostfs_testlib/load/load_config.py |  2 ++
 tests/test_load_config.py               | 13 +++++++++++++
 2 files changed, 15 insertions(+)

diff --git a/src/frostfs_testlib/load/load_config.py b/src/frostfs_testlib/load/load_config.py
index 532be16..7bde399 100644
--- a/src/frostfs_testlib/load/load_config.py
+++ b/src/frostfs_testlib/load/load_config.py
@@ -221,6 +221,8 @@ class LoadParams:
     )
     # Percentage of filling of all data disks on all nodes
     fill_percent: Optional[float] = None
+    # if set, the payload is generated on the fly and is not read into memory fully.
+    streaming: Optional[int] = metadata_field(all_load_scenarios, None, "STREAMING", False)
 
     # ------- COMMON SCENARIO PARAMS -------
     # Load time is the maximum duration for k6 to give load. Default is the BACKGROUND_LOAD_DEFAULT_TIME value.
diff --git a/tests/test_load_config.py b/tests/test_load_config.py
index 8f28621..dc019b7 100644
--- a/tests/test_load_config.py
+++ b/tests/test_load_config.py
@@ -156,6 +156,7 @@ class TestLoadConfig:
             "READERS": 7,
             "DELETERS": 8,
             "READ_AGE": 8,
+            "STREAMING": 9,
             "PREGEN_JSON": "pregen_json",
             "PREPARE_LOCALLY": True,
         }
@@ -195,6 +196,7 @@ class TestLoadConfig:
             "READ_RATE": 9,
             "READ_AGE": 8,
             "DELETE_RATE": 11,
+            "STREAMING": 9,
             "PREPARE_LOCALLY": True,
         }
 
@@ -225,6 +227,7 @@ class TestLoadConfig:
             "READERS": 7,
             "DELETERS": 8,
             "READ_AGE": 8,
+            "STREAMING": 9,
             "NO_VERIFY_SSL": True,
             "PREGEN_JSON": "pregen_json",
         }
@@ -265,6 +268,7 @@ class TestLoadConfig:
             "WRITE_RATE": 10,
             "READ_RATE": 9,
             "READ_AGE": 8,
+            "STREAMING": 9,
             "DELETE_RATE": 11,
         }
 
@@ -303,6 +307,7 @@ class TestLoadConfig:
             "WRITE_RATE": 10,
             "READ_RATE": 9,
             "READ_AGE": 8,
+            "STREAMING": 9,
             "DELETE_RATE": 11,
         }
 
@@ -335,6 +340,7 @@ class TestLoadConfig:
             "READERS": 7,
             "DELETERS": 8,
             "READ_AGE": 8,
+            "STREAMING": 9,
             "PREGEN_JSON": "pregen_json",
         }
 
@@ -366,6 +372,7 @@ class TestLoadConfig:
             "READERS": 7,
             "DELETERS": 8,
             "READ_AGE": 8,
+            "STREAMING": 9,
             "PREGEN_JSON": "pregen_json",
         }
 
@@ -418,6 +425,7 @@ class TestLoadConfig:
             "READERS": 0,
             "DELETERS": 0,
             "READ_AGE": 0,
+            "STREAMING": 0,
             "PREGEN_JSON": "",
             "PREPARE_LOCALLY": False,
         }
@@ -455,6 +463,7 @@ class TestLoadConfig:
             "READ_RATE": 0,
             "DELETE_RATE": 0,
             "READ_AGE": 0,
+            "STREAMING": 0,
             "PREPARE_LOCALLY": False,
         }
 
@@ -483,6 +492,7 @@ class TestLoadConfig:
             "READERS": 0,
             "DELETERS": 0,
             "READ_AGE": 0,
+            "STREAMING": 0,
             "NO_VERIFY_SSL": False,
             "PREGEN_JSON": "",
         }
@@ -521,6 +531,7 @@ class TestLoadConfig:
             "READ_RATE": 0,
             "DELETE_RATE": 0,
             "READ_AGE": 0,
+            "STREAMING": 0,
         }
 
         self._check_preset_params(load_params, expected_preset_args)
@@ -549,6 +560,7 @@ class TestLoadConfig:
             "READERS": 0,
             "DELETERS": 0,
             "READ_AGE": 0,
+            "STREAMING": 0,
             "PREGEN_JSON": "",
         }
 
@@ -578,6 +590,7 @@ class TestLoadConfig:
             "READERS": 0,
             "DELETERS": 0,
             "READ_AGE": 0,
+            "STREAMING": 0,
             "PREGEN_JSON": "",
         }