Skip to content

Commit 15b8fbb

Browse files
committed
move extract_block() to resource_utils package
1 parent a4de5c6 commit 15b8fbb

File tree

2 files changed

+21
-18
lines changed

2 files changed

+21
-18
lines changed

tests/suite/test_otel.py

Lines changed: 8 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import pytest
22
from settings import TEST_DATA
33
from suite.utils.resources_utils import (
4+
extract_block,
45
get_nginx_template_conf,
56
replace_configmap_from_yaml,
67
wait_before_test,
@@ -25,17 +26,6 @@
2526

2627
@pytest.mark.otel
2728
class TestOtel:
28-
def extract_block(self, nginx_config, block_name):
29-
"""
30-
Extract a block of configuration from the nginx config file.
31-
32-
:param nginx_config: The nginx config file content as a string.
33-
:param block_name: The name of the block to extract.
34-
:return: The extracted block as a string.
35-
"""
36-
start = nginx_config.find(block_name)
37-
end = nginx_config.find("}", start) + 1
38-
return nginx_config[start:end]
3929

4030
def test_otel_not_enabled(
4131
self,
@@ -96,7 +86,7 @@ def test_otel_endpoint(
9686
print("Step 2: Ensure that the otel module is loaded")
9787
assert otel_module in (nginx_config)
9888

99-
exporter_block = self.extract_block(nginx_config, "otel_exporter")
89+
exporter_block = extract_block(nginx_config, "otel_exporter")
10090

10191
print("Step 3: Ensure that the otel_exporter is enabled")
10292
assert "otel_exporter" in (exporter_block)
@@ -147,7 +137,7 @@ def test_otel_header(
147137
print("Step 2: Ensure that the otel module is loaded")
148138
assert otel_module in (nginx_config)
149139

150-
exporter_block = self.extract_block(nginx_config, "otel_exporter")
140+
exporter_block = extract_block(nginx_config, "otel_exporter")
151141

152142
print("Step 3: Ensure that the otel_exporter is enabled")
153143
assert "otel_exporter" in (exporter_block)
@@ -199,7 +189,7 @@ def test_otel_header_name_only(
199189
print("Step 2: Ensure that the otel module is loaded")
200190
assert otel_module in (nginx_config)
201191

202-
exporter_block = self.extract_block(nginx_config, "otel_exporter")
192+
exporter_block = extract_block(nginx_config, "otel_exporter")
203193

204194
print("Step 3: Ensure that the otel_exporter is enabled")
205195
assert "otel_exporter" in (exporter_block)
@@ -251,7 +241,7 @@ def test_otel_header_value_only(
251241
print("Step 2: Ensure that the otel module is loaded")
252242
assert otel_module in (nginx_config)
253243

254-
exporter_block = self.extract_block(nginx_config, "otel_exporter")
244+
exporter_block = extract_block(nginx_config, "otel_exporter")
255245

256246
print("Step 3: Ensure that the otel_exporter is enabled")
257247
assert "otel_exporter" in (exporter_block)
@@ -303,7 +293,7 @@ def test_otel_service_name(
303293
print("Step 2: Ensure that the otel module is loaded")
304294
assert otel_module in (nginx_config)
305295

306-
exporter_block = self.extract_block(nginx_config, "otel_exporter")
296+
exporter_block = extract_block(nginx_config, "otel_exporter")
307297

308298
print("Step 3: Ensure that the otel_exporter is enabled")
309299
assert "otel_exporter" in (exporter_block)
@@ -354,7 +344,7 @@ def test_otel_trace(
354344
print("Step 2: Ensure that the otel module is loaded")
355345
assert otel_module in (nginx_config)
356346

357-
exporter_block = self.extract_block(nginx_config, "otel_exporter")
347+
exporter_block = extract_block(nginx_config, "otel_exporter")
358348

359349
print("Step 3: Ensure that the otel_exporter is enabled")
360350
assert "otel_exporter" in (exporter_block)
@@ -405,7 +395,7 @@ def test_otel_all(
405395
print("Step 2: Ensure that the otel module is loaded")
406396
assert otel_module in (nginx_config)
407397

408-
exporter_block = self.extract_block(nginx_config, "otel_exporter")
398+
exporter_block = extract_block(nginx_config, "otel_exporter")
409399

410400
print("Step 3: Ensure that the otel_exporter is enabled")
411401
assert "otel_exporter" in (exporter_block)

tests/suite/utils/resources_utils.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1070,6 +1070,19 @@ def get_ts_nginx_template_conf(v1: CoreV1Api, resource_namespace, resource_name,
10701070
return get_file_contents(v1, file_path, pod_name, pod_namespace)
10711071

10721072

1073+
def extract_block(nginx_config, block_name):
1074+
"""
1075+
Extract a block of configuration from the nginx config file.
1076+
1077+
:param nginx_config: The nginx config file content as a string.
1078+
:param block_name: The name of the block to extract.
1079+
:return: The extracted block as a string.
1080+
"""
1081+
start = nginx_config.find(block_name)
1082+
end = nginx_config.find("}", start) + 1
1083+
return nginx_config[start:end]
1084+
1085+
10731086
def create_example_app(kube_apis, app_type, namespace) -> None:
10741087
"""
10751088
Create a backend application.

0 commit comments

Comments
 (0)