Skip to content

Commit 84e02eb

Browse files
authored
Merge pull request #149 from olehermanse/master
CFE-4565: Relaxed assertion related to cloud provider drivers
2 parents 03ef6e8 + fb5b190 commit 84e02eb

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

cf_remote/spawn.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,9 @@ def get_by_ip(cls, ip, driver=None, nodes=None):
7777
print("Don't know which driver to use: %s" % _DRIVERS.keys())
7878
return None
7979

80-
assert driver is not None, "TODO: How does this happen?"
81-
nodes = nodes or driver.list_nodes()
80+
if not nodes:
81+
assert driver is not None
82+
nodes = driver.list_nodes()
8283
for node in nodes:
8384
if node.state in (0, "running") and (
8485
ip in node.public_ips or ip in node.private_ips
@@ -96,8 +97,9 @@ def get_by_name(cls, name, driver=None, nodes=None):
9697
print("Don't know which driver to use: %s" % _DRIVERS.keys())
9798
return None
9899

99-
assert driver is not None, "TODO: How does this happen?"
100-
nodes = nodes or driver.list_nodes()
100+
if not nodes:
101+
assert driver is not None
102+
nodes = driver.list_nodes()
101103
for node in nodes:
102104
if node.state in (0, "running") and node.name == name:
103105
return cls(node.name, driver, node)
@@ -113,8 +115,9 @@ def get_by_uuid(cls, uuid, driver=None, nodes=None):
113115
print("Don't know which driver to use: %s" % _DRIVERS.keys())
114116
return None
115117

116-
assert driver is not None, "TODO: How does this happen?"
117-
nodes = nodes or driver.list_nodes()
118+
if not nodes:
119+
assert driver is not None
120+
nodes = driver.list_nodes()
118121
for node in nodes:
119122
if node.uuid == uuid:
120123
return cls(node.name, driver, node)

0 commit comments

Comments
 (0)