diff --git a/esrally/track/params.py b/esrally/track/params.py index ca5b2d5bc..1953dad78 100644 --- a/esrally/track/params.py +++ b/esrally/track/params.py @@ -866,9 +866,12 @@ def __init__(self, track, params, **kwargs): params["index"] = params.get("source-index") self._source_index = get_target(track, params) self._target_index = params.get("target-index", f"{self._source_index}-{self._fixed_interval}") + self._sampling_method = params.get("sampling-method", None) def params(self): parsed_params = {"fixed-interval": self._fixed_interval, "source-index": self._source_index, "target-index": self._target_index} + if self._sampling_method: + parsed_params["sampling-method"] = self._sampling_method parsed_params.update(self._client_params()) return parsed_params diff --git a/tests/track/params_test.py b/tests/track/params_test.py index 096fab5d2..86701bb28 100644 --- a/tests/track/params_test.py +++ b/tests/track/params_test.py @@ -2958,6 +2958,7 @@ def test_downsample_all_params(self): "source-index": "test-source-index", "target-index": "test-target-index", "fixed-interval": "1m", + "sampling-method": "last_value", }, ) @@ -3010,3 +3011,4 @@ def test_downsample_empty_params(self): p = source.params() assert p["fixed-interval"] == "1h" assert p["target-index"] == f"{p['source-index']}-{p['fixed-interval']}" + assert p.get("sampling-method") is None