Skip to content

Commit 35cb52b

Browse files
committed
return pod names
1 parent bef0ffa commit 35cb52b

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/fault_tolerance/deploy/scenarios.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
import asyncio
1717
import logging
1818
import re
19-
from dataclasses import dataclass, field
2019
from abc import ABC, abstractmethod
20+
from dataclasses import dataclass, field
2121
from enum import Enum, auto
2222
from typing import TYPE_CHECKING, Dict, List, Optional, Pattern
2323

@@ -213,7 +213,7 @@ async def execute(
213213
) # have some requests processed after the rolling upgrade has completed
214214

215215
return await deployment.get_pod_names(self.service_names)
216-
216+
217217
def get_failure_key(self) -> str:
218218
"""Get the failure key for the rolling upgrade failure."""
219219
return f"rolling_upgrade:{','.join(self.service_names)}"
@@ -237,6 +237,8 @@ async def execute(
237237
pod.delete(force=True) # force means no graceful termination
238238
pod_names.append(pod.name)
239239

240+
return pod_names
241+
240242
def get_failure_key(self) -> str:
241243
"""Get the failure key for the delete pod failure."""
242244
return f"delete_pod:{','.join(self.service_names)}"
@@ -288,8 +290,9 @@ async def execute(
288290
)
289291
process.kill(self.signal)
290292
pod_names.append(pod.name)
293+
291294
return pod_names
292-
295+
293296
def get_failure_key(self) -> str:
294297
"""Get the failure key for the terminate process failure."""
295298
return f"terminate_process:{','.join(self.service_names)}:{self.process_name}:{self.signal}"

0 commit comments

Comments
 (0)