Skip to content

Commit 0461d27

Browse files
committed
task: cleanup setting "use_secondary_network", "resource_name" Jinja2 variables
Signed-off-by: Thomas Haller <thaller@redhat.com>
1 parent b6b74ec commit 0461d27

2 files changed

Lines changed: 18 additions & 16 deletions

File tree

manifests/sriov-pod.yaml.j2

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,11 @@ spec:
1717
image: {{ test_image }}
1818
command: {{ command }}
1919
args: {{ args }}
20-
imagePullPolicy: {{ image_pull_policy }} {% if use_secondary_network and resource_name %}
20+
imagePullPolicy: {{ image_pull_policy }}{% if use_secondary_network and has_resource_name %}
2121
resources:
2222
requests:
2323
{{ resource_name }}: '1'
2424
limits:
25-
{{ resource_name }}: '1' {% endif %}
25+
{{ resource_name }}: '1'{% endif %}
2626
securityContext:
2727
privileged: {{ privileged_pod }}

task.py

Lines changed: 16 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,16 @@ def _fetch_default_resource_name(
336336
logger.info(f"autodetected resource_name as {repr(resource_name)}")
337337
return resource_name
338338

339-
def get_template_args(self) -> dict[str, str | list[str]]:
339+
def get_template_args(self) -> dict[str, str | list[str] | bool]:
340+
resource_name = (
341+
self.ts.connection.resource_name
342+
or Task._fetch_default_resource_name(
343+
self.client,
344+
self.get_namespace(),
345+
self.ts.connection.secondary_network_nad,
346+
)
347+
or ""
348+
)
340349
return {
341350
"name_space": _j(self.get_namespace()),
342351
"test_image": _j(tftbase.get_tft_test_image()),
@@ -348,19 +357,12 @@ def get_template_args(self) -> dict[str, str | list[str]]:
348357
"pod_name": _j(self.pod_name),
349358
"privileged_pod": _j(self._get_template_args_privileged_pod()),
350359
"port": self._get_template_args_port(),
351-
"secondary_network_nad": _j(self.ts.connection.effective_secondary_network_nad),
352-
"use_secondary_network": (
353-
"1" if self.ts.connection.secondary_network_nad else ""
354-
),
355-
"resource_name": (
356-
self.ts.connection.resource_name
357-
or Task._fetch_default_resource_name(
358-
self.client,
359-
self.get_namespace(),
360-
self.ts.connection.secondary_network_nad,
361-
)
362-
or ""
360+
"secondary_network_nad": _j(
361+
self.ts.connection.effective_secondary_network_nad
363362
),
363+
"use_secondary_network": bool(self.ts.connection.secondary_network_nad),
364+
"has_resource_name": bool(resource_name),
365+
"resource_name": _j(resource_name),
364366
"default_network": _j(self.node.default_network),
365367
}
366368

@@ -394,7 +396,7 @@ def render_file(
394396
log_info: str,
395397
in_file_template: str,
396398
out_file_yaml: str,
397-
template_args: Optional[dict[str, str | list[str]]] = None,
399+
template_args: Optional[dict[str, str | list[str] | bool]] = None,
398400
) -> None:
399401
if template_args is None:
400402
template_args = self.get_template_args()

0 commit comments

Comments
 (0)